o rXj@sRddlmZddlZddlmZddlmZddlmZm Z m Z GdddZ dS)) annotationsN)Path)OpenAI)retrystop_after_attemptwait_exponentialc@sreZdZdZ d2ddddd3ddZddd4ddZd5d6d d!Zeed"e d#d$d%d&d'dddd(d7d0d1Z dS)8SoraVideoGeneratoru OpenAI Sora (Videos API) — original short-form marketing clips. Generates fully new motion footage from a storyboard prompt. Does NOT animate static images. Audio is intentionally omitted (see src/generators/audio.py for a future modular add-on). sora-212g@g @)default_seconds poll_intervaltimeoutapi_keystrmodelr r floatr cCs,t|d|_||_t||_||_||_dS)N)r)rclientrrr r r )selfrrr r r rF/home/esfera/Documents/content_generation/src/generators/sora_video.py__init__s   zSoraVideoGenerator.__init__N)beatshooktopicwhy_it_matterskey_factcreative_direction video_style news_categoryrlist[dict] | Nonereturnc Csg} |r0|D])} | d| ddd| ddd| dp$| d p$dd | d d q| r7d | nd|d|d|d|d } d|pJ|d|d|d| d ddS)z8Compress cinematic reel storyboard into one Sora prompt.z- beatscenez: on-screen feel ' on_screenz '; visual visual voiceoverz; camera/motion motionzcinematic move z- hook: z - news: z - why it matters: z - key fact: z1 - ending CTA energy: follow for daily tech dropsuPremium vertical short-form marketing reel for an AI/tech media brand (Instagram Reels / TikTok / YouTube Shorts quality). Cinematic, original motion footage — NOT a slideshow, NOT animating a still poster, NOT PowerPoint. Multiple dynamic scenes with camera moves, parallax, light streaks, kinetic energy, product/UI showcases, futuristic AI environments. NO readable text overlays, NO real company logos, NO watermarks, NO voiceover requirement. Creative direction: z. Visual genre: z . Style: z. Narrative arc: z Pacing: punchy hook first 2 seconds, then explanatory visuals, then impactful ending. Photoreal + high-end motion graphics hybrid, rich lighting, depth, professional color grade.N)appendgetjoin) rrrrrrrrr beat_linesb beats_blockrrr build_prompt!s<    zSoraVideoGenerator.build_promptportraitaspectcCs|dkrdSdS)N landscape1280x720720x1280r)rr2rrrsize_for_aspectLsz"SoraVideoGenerator.size_for_aspect)minmaxT)stopwaitreraise)secondssizer2promptout_pathrr? str | Noner@dictcCsD|jjddd|p |j}|dvrd}|p||}|jjj|j|dd||d}|j} t |dd} t } | d vrgt | |j krPt d | d |j d t |j|jj| }t |dd } | d vs;| dkrzt |dd} td| pv| |jj| } t| dr| nt| }|d}|||| |j||| dS)NT)parentsexist_ok>48r r r))rrAr?r@statusqueued>failed completedz Sora job z timed out after sunknownrLerrorzSora generation failed: readz.mp4)pathvideo_idrr?r@rI)parentmkdirr r6rvideoscreateridgetattrtimer TimeoutErrorsleepr retrieve RuntimeErrordownload_contenthasattrrPbytes with_suffix write_bytes)rrArBr?r@r2secsszvideorRrIstartederrcontentdatarrrgenerateRsF        zSoraVideoGenerator.generate)r ) rrrrr rr rr r)rrrrrrrrrrrrrrrrr r)r1)r2rr r) rArrBrr?rCr@rCr2rr rD) __name__ __module__ __qualname____doc__rr0r6rrrrjrrrrr s   +r) __future__rrYpathlibropenairtenacityrrrrrrrrs