Bi9dZddlZddlZddlmZmZddlmZddlmZm Z m Z ddl m Z m Z ddlmZmZmZdd lmZmZdd lmZej.eZGd d Z dd ede ede ede ede eeeff dZy)z Core visual generation logic for Napkin AI API. Provides high-level interface for generating visuals through the API, handling the complete workflow from request to downloaded files. N)datetimetimezone)Path)ListOptionalTuple)NapkinAPIClientProcessingError) VisualRequestStatusResponse OutputFormat)Settings get_settings)get_style_by_namec(.eZdZdZd"deefdZdZdZ d#de dee d ee d ee d ee d ee d ee de de dee dee dee dee dee dee de f dZ d$de de d e de de f dZ d%de deedee d ee d ee d ee d ee d ee de de dee dee de dee dee dee dee deeeeff$dZ d&dee deedee d ee d ee deee eeeff d!Zy)'VisualGeneratorz+High-level interface for visual generation.Nsettingsc8|xs t|_d|_y)z Initialize visual generator. Args: settings: Configuration settings. If None, loads from environment. N)rrclient)selfrs 1/var/www/html/Napkin-AI-API/src/core/generator.py__init__zVisualGenerator.__init__s!2LN 15 cKt|j|_|jjd{|S7w)zAsync context manager entry.N)r rr __aenter__)rs rrzVisualGenerator.__aenter__$s6%dmm4 kk$$&&&  's8AAActK|jr&|jj|||d{yy7w)zAsync context manager exit.N)r __aexit__)rexc_typeexc_valexc_tbs rrzVisualGenerator.__aexit__*s1 ;;++'''6B B B  Bs -868contentstyleformatlanguage variationscontext_before context_after transparentinvertedwidthheight visual_id visual_querytransparent_backgroundinverted_colorreturncN|xs|jj}|xs|jj}|xs|jj} t |j }d}|r t|}|j}n, t|jj}|j}t|||||||||n|||n| | | | |  S#t $rtj}YwxYw#t $r|}YLwxYw#t $r|jj}YqwxYw)az Prepare visual request with defaults. Args: content: Text content to visualize. style: Style name or ID. format: Output format (svg/png). language: Language code. variations: Number of variations (1-4). context_before: Context before content. context_after: Context after content. transparent: Enable transparent background. inverted: Invert colors. width: Width in pixels (PNG only). height: Height in pixels (PNG only). Returns: Prepared VisualRequest. N) r"r$style_idr%number_of_visualsr'r(r/r0r+r,r-r.) rdefault_formatdefault_languagedefault_variationsrlower ValueErrorSVGrid default_styler )rr"r#r$r%r&r'r(r)r*r+r,r-r.r/r0 format_str format_enumr3 style_objs r_prepare_requestz VisualGenerator._prepare_request/s5L;t}};; =t}}==C4==#C#C  +&z'7'7'9:K   !-e4 $<<  7-dmm.I.IJ $<< ()'%1$:-;-G>X%  - +&**K + !  ! 7==66 7s6C.C1+DC.-C.1 C?>C?D$#D$ request_idfile_idindexctjtjj d}|dkDrd|d|ddd|dzd |Sd|d|ddd |S) a Generate filename for saved visual. Args: request_id: Request ID. file_id: File ID. format: File format. index: File index for multiple variations. Returns: Generated filename. z %Y%m%d_%H%M%Srnapkin__N_v.)rnowrutcstrftime)rrArBr$rC timestamps r_generate_filenamez"VisualGenerator._generate_filenamesr(LL.77H 19YKqBQ(8519+QvhO O 1Z^$4AfX>>r output_dir save_filesc!K|js td|j|||||||| | | | ||||!tj d!j d!j d|jj!d{}|j}tj d||jj|d{}|jjdk7rtd |jxsd tj d |jd g}| rO|jd kDr?|xs|jj }t#|}|j%ddt'|dgxsg}dt(t*dt(t*dt*f!fd }|r1t-|D]\}}|j/dd|dz}|j/d}||j/d|}|j/dxs|j1||||}||z } |r\|jj3|d{}|j5||j7|tj d|nL|jj9|||d{|j7|tj d|!||fSt=|jD]}d|dz}|j1||!j j|}||z } |jj9|||d{|j7|tj d|||fS77777#t:$r"} tjd|| Yd} ~ d} ~ wwxYw7j#t:$r!} tjd|| Yd} ~ d} ~ wwxYww)a\ Generate visuals from text content. Args: content: Text content to visualize. output_dir: Directory to save files (uses config default if None). style: Style name or ID. format: Output format (svg/png). language: Language code. variations: Number of variations (1-4). context_before: Context before content. context_after: Context after content. transparent: Enable transparent background. inverted: Invert colors. width: Width in pixels (PNG only). height: Height in pixels (PNG only). save_files: Whether to save files to disk. Returns: Tuple of (final status, list of saved file paths). Raises: ProcessingError: If generation fails. z/Generator must be used as async context manager)r"r#r$r%r&r'r(r)r*r+r,r-r.r/r0z Generating z visual(s) in z formatNzRequest created: completedzGeneration failed: z Unknown errorzGeneration completed: z file(s) readyrT)parentsexist_okfilesfmturlr1c|dvr|S|r8d|jvsd|jvryd|jvrytdrjjSdS)N>pngsvgz.svgz image/svgr[z.pngrZr$)r8hasattrr$value)rWrXrequests r _infer_extz,VisualGenerator.generate.._infer_exts`.(J, syy{0J$,$/6w/Iw~~++TuTrr;file_rIr$filenamez Saved: %szFailed to download file %s: %s)r RuntimeErrorr@loggerinfor4r$ create_visualrAwait_for_completionstatusr]r error files_readyr storage_pathrmkdirgetattrrstr enumerategetrOdownload_file_by_url write_bytesappend download_file Exceptionrange)"rr"rPr#r$r%r&r'r(r)r*r+r,rQr-r.r/r0responserA final_status saved_pathsrVr_i file_inforBfile_urlfile_fmtra file_path file_contenter^s" @rgeneratezVisualGenerator.generatesX{{PQ Q''!)'#%#9)( &  '334N7>>BRRY Z 227;;((  ' |45"[[<+O?O-O- NN N;N60O-6N;;O->O O* O% O-%O**O-contentsconcurrent_limitc K|xsjj}tj| dtdt f fd }t |D cgc]\}} || |} }} tj| d{} td| D} tjd| dt|d| Scc} }w7Cw) a Generate visuals for multiple contents in batch. Args: contents: List of text contents. output_dir: Directory to save files. style: Style name or ID. format: Output format. concurrent_limit: Max concurrent requests. **kwargs: Additional generation parameters. Returns: List of tuples (content, status, file_paths) for each generation. r"rCc K 4d{ d}r d|dzdz } jd|| dd{\}}|||fcdddd{S7M77 #t$r?}tjd|dzd||dgfcYd}~cdddd{7Sd}~wwxYw#1d{7swYyxYww) z%Generate visual for one content item.Nbatch_rI03dr"rPr#r$z#Failed to generate visual for item z: )rrtrcrh) r"rCitem_dirrgpathsrr$kwargsrPr semaphorer#s r generate_onez4VisualGenerator.generate_batch..generate_one[s y-#H!#-&30H#H*7$--+ '#+#% + ! +%MFE#FE1!yy%!!-LL#FuqykQSTUSV!WX"D",,#!yy-!yysB>AB>B)*AA A B>AB>AB> B&'"B! B& B) B>BB>!B&&B))B;/B2 0B;7B>Nc3.K|] \}}}|s dyw)rINr).0rFrgs r z1VisualGenerator.generate_batch..vsAw|q&!&ws zBatch generation completed: /z successful) rbatch_concurrent_limitasyncio Semaphorermintrngathersumrcrdlen)rrrPr#r$rrrryr"tasksresults successfulrs` ``` ` @rgenerate_batchzVisualGenerator.generate_batch?s.,St}}/S/S%%&67  - -C - -,=Fhrs'((9DD1/   8 $ddR "&     C=  SM   >4: %&  r