mj$dZddlmZddlZddlZddlmZmZddlm Z ddl m Z ddl Z ddZ e ZdZd Zd Zd Zdd Zdd d Zd!dZ d"dZ d# d$dZedf d%dZedk(r~ddlZddlZddlZeej<dkDrej<dnddddgZeeej@jBdde ejDdZ#e$de#yy)&u Geração de B-roll via IA usando Seedance-1-Pro (ByteDance) no Replicate. API: https://replicate.com/bytedance/seedance-1-pro Custo aproximado: US$0.05/segundo @ 1080p, ~US$0.025/segundo @ 720p Uso típico (substitui Pexels): clips = gerar_videos_seedance( termos=[{"query": "...", "inicio": 0.0, "fim": 5.0}, ...], job_id="abc", temp_dir=Path("/tmp/temp"), resolution="720p", ) # clips => [{"path": "/tmp/...mp4", "inicio": 0.0, "fim": 5.0}, ...] ) annotationsN)ThreadPoolExecutor as_completed)Path)IterablecNtjddj}|r|Std}|j r |j dj D]}|j}|r|jdr'd|vr,|jdd\}}|jdk(sU|jjd jd }|tjd<|cS yy#t$r}td |Yd}~yd}~wwxYw) NREPLICATE_API_TOKENz /opt/mia/.envzutf-8)encoding#="'z%[seedance] erro lendo /opt/mia/.env: ) osgetenvstriprexists read_text splitlines startswithsplitenviron Exceptionprint)tokenv_pathlinekves 9/opt/mia/workspace/video_editor/backend/seedance_utils.py_load_replicate_tokenr# s  )))2 . 4 4 6C  O$H ? **G*<GGI zz|ts3d?zz#q)1779 55 ,2237A89BJJ45H   ? 9!= > >  ?s&A4D:AD?D D$ DD$zbytedance/seedance-1-pro cX|xsdjjd}|sy|dS)z Recebe a query curta vinda do Claude (ex: "person typing on laptop") e devolve um prompt rico otimizado para Seedance, vertical 9:16. r .z, cinematic vertical 9:16 composition, soft natural lighting, shallow depth of field, smooth slow camera movement, ultra-realistic, high detail photographic quality, 4k, professional color grading)rrstrip)queryqs r"_enriquecer_promptr,As? "$$S)A  #K Lc tj|d|5}|j|jj ddt |d5}|j dD]}|s|j| dddddd|jxr|jjdkDS#1swY@xYw#1swYDxYw#t$r}td |d |Yd}~y d}~wwxYw) z7Baixa um MP4 pra o caminho destino. Retorna True/False.T)streamtimeoutparentsexist_okwbi) chunk_sizeNrz[seedance] erro baixando z: F) requestsgetraise_for_statusparentmkdiropen iter_contentwriterstatst_sizerr)urldestinor0rfchunkr!s r"_baixar_arquivorERs  \\#dG < '   NN  =gt$ '^^y^A'E' ' '~~>GLLN$:$:Q$>>  ' ' ' '  )#b45sLC:C C-CC  6CC C  CC C<!C77C<cp|yt|tr|St|ttfr|rt |dSdD]8}t ||d}t |r |cSt|ts6|cS t|}|jdr|S y#t$rYgwxYw#t$rYywxYw)u O Replicate Python client retorna: - str (URL) na maioria dos modelos novos - objeto FileOutput (com método .url) em alguns - list[str] em modelos multi-output Normaliza pra URL única. Nr)r@http) isinstancestrlisttuple_resolver_output_urlgetattrcallablerr)outputattrr ss r"rLrLbs~&# &4-(V#F1I.. FD$ ' A; s 3 H K << H        s$B:B) B&%B&) B54B5cddl}|jdxsdj}|syt|} t t t ttt|jddt|jddz xstt} tj} td|d| d |d | |jt| | |||d d  } t!| }|std|d| y||d|ddz }t#||std|d|ytj| z }td|d|dd|t%|t|dt|d|| t|ddS#t$r} td|d|d| Yd} ~ yd} ~ wwxYw)zk Gera UM clip via Seedance, baixa o MP4 e devolve dict do clip. Retorna None em caso de falha. rNr*r fiminicioz[seedance] gerando clip z (zs, z): F)promptduration resolution aspect_ratiofps camera_fixed)inputz&[seedance] erro chamando modelo (clip z, 'z'): z[seedance] clip z sem URL no output: _seedance_02dz.mp4z [seedance] falha ao baixar clip z de z OK em .1fzs -> r)pathrTrSr*duracao_geradatempo_geracao_s) replicater7rr,maxSEEDANCE_DUR_MINminintroundfloatSEEDANCE_DUR_DEFAULTtimerrun MODEL_SLUGrrLrErI)termoidxjob_idtemp_dirrWrXrYrbr*rU duracao_alvot0rOr!r@destdurs r"_gerar_um_cliprus YYw  %2 , , .E   &F' E% %+,uUYYx5K/LL MNfRf*L B $SEL>ZLE9 UV  (( , %    v &C   %9&DE  3s)48 8D 3 % 0T#?@ ))+ C SES tf =>D h(U5\"& a=  #  6se3ugT!MNsF F=F88F=720pc|Dcgc](}|jdxsdjs'|*}}|s tdgSts tdgSttj d<t |}|jddtj}tdt|d |d |g} t| 5} t|D cic]"\} } | jt| | |||||| $} } } t| D]+}| |} |j}|r| j!|- ddd| j%dtj|z }t| }t||z }td|d|d|dd| Scc}wcc} } w#t"$r}td | d |Yd}~d}~wwxYw#1swYxYw)u; Gera N clips em paralelo via Seedance-1-Pro. Args: termos: lista de {"query": str, "inicio": float, "fim": float} job_id: id do job (pra nomear arquivos) temp_dir: pasta destino dos MP4s resolution: "720p" (mais barato) ou "1080p" aspect_ratio: "9:16" pra Reels (default) fps: 24 ou 30 max_workers: paralelismo (default 3 — Replicate aceita bem) Returns: list[dict] = [{"path", "inicio", "fim", "query", "duracao_gerada", "tempo_geracao_s"}, ...] Lista vazia se TODAS falharem. r*r u)[seedance] nenhum termo válido recebido.u;[seedance] REPLICATE_API_TOKEN não configurado. Abortando.r Tr1u"[seedance] iniciando geração de z clip(s), workers=z , resolution=) max_workersz[seedance] worker z explodiu: Nc |dS)NrT)xs r"z'gerar_videos_seedance..s !H+r-)keyz[seedance] terminou: z OK, z falha(s) em r^zs total)r7rrREPLICATE_TOKENrrrr:rjlenr enumeratesubmitrurresultappendrsort)termosrorprWrXrYrxtt_start resultadospoolirmfuturesfutrBr!totalsucessofalhass r"gerar_videos_seedancers2 BAAEE'N$8b#?#?#Aa BF B  9:  KL )8BJJ$%H~H NN4$N/iikG .s6{m< Mzl <=J  4> &f-  5 KKq&(L#     ( >C A >JJL%%a(  >>$OO-O. IIK' !E*oG [7 "F !'%x}U3Kw WX [C*  >*1#[<== >>>sR(F F  G'F%G#F+:G%G+ G4G G GGG1080pc8|dk(rdnd}t||z|zdS)zZ Estimativa simples de custo do Replicate. 1080p ~= $0.05/s, 720p ~= $0.025/s rg?g?)rg) qtd_clips duracao_srWpreco_ss r"estimar_custo_usdrs) !G+dG Y&0! 44r-__main__rz$a photographer adjusting camera lensgg@)r*rTrS)rrorprWz RESULTADO:)returnrI)r*rIrrI)x)r@rIrArr0rfrbool)rz str | None)rmdictrnrfrorIrprrWrIrXrIrYrfrz dict | None)rvz9:16r$)rzIterable[dict]rorIrprrWrIrXrIrYrfrxrfrrJ)rrfrrfrWrIrrh)%__doc__ __future__rrrjconcurrent.futuresrrpathlibrtypingrr6r#r~rlrdSEEDANCE_DUR_MAXrir,rErLrurr__name__systempfileuuidrargvruuid4hex gettempdiroutrrzr-r"rs # ? .() ' "  D@ @ @ @ @  @  @ @@NF F FF F  F FF FR8L'.5!$5385 z #CHH  1!7]F tzz|#)h))+,  C  ,r-