Update main.py

main.py

@@ -573,9 +573,63 @@ EXEMPLOS
                                 video_x = 0
                                 video_y = 0
                             print(f"   → video_y (crop offset): {video_y}px")
-                            
-                            # 5. Montar title_url
+
+                            # 5. Gerar legendas se necessário
+                            needs_legenda = result_data.get("legenda", False)
+                            srt_for_export = None
+                            if needs_legenda:
+                                print("🎙️ Gerando legendas (Groq + Gemini)...")
+                                try:
+                                    # Transcricao via Groq usando o arquivo local
+                                    import urllib.request as _urlreq
+                                    import uuid as _uuid
+
+                                    # Groq precisa de URL — usamos o video já upado (export_video_url)
+                                    srt_raw, _, _, _ = await get_groq_srt_base(
+                                        export_video_url, language="en", temperature=0.4,
+                                        has_bg_music=False
+                                    )
+
+                                    if srt_raw and srt_raw.strip():
+                                        # Tradução via Gemini (mesmo prompt do /subtitle)
+                                        translate_prompt = f"""IDIOMA: A legenda traduzida DEVE ser inteiramente em PORTUGUÊS DO BRASIL (pt-BR).
+
+Traduza essa legenda pro português do Brasil, corrija qualquer erro de formatação, pontuação e mantenha timestamps e os textos nos seus respectivos blocos de legenda.
+Você DEVE se basear estritamente na legenda original fornecida. NUNCA crie legendas novas e NUNCA adicione ou verifique diálogos no áudio que não estejam presentes na legenda original. Apenas traduza.
+Mande o SRT completo, sem textos adicionais na resposta, apenas o SRT traduzido.
+Não deve ser literal a tradução, deve se adaptar ao contexto e entonação. Se alguém estiver gritando, ESCREVA MAIÚSCULO!
+NUNCA legende músicas, apenas diálogos falados.
+PALAVRÕES: substitua com asteriscos (ex: "merda" → "merd*", "caralho" → "caral**").
+
+LEGENDA ORIGINAL:
+{srt_raw}"""
+                                        translate_model = get_gemini_model("flash")
+                                        translate_resp = await client.generate_content(translate_prompt, model=translate_model)
+                                        translated_srt = translate_resp.text.strip()
+
+                                        # Limpar bloco de código se o Gemini retornar com ```
+                                        if "```" in translated_srt:
+                                            parts = translated_srt.split("```")
+                                            for part in parts:
+                                                clean = part.strip()
+                                                if clean.startswith("srt"): clean = clean[3:].strip()
+                                                if re.match(r"\d+\s*\n\d{2}:", clean):
+                                                    translated_srt = clean
+                                                    break
+
+                                        if translated_srt:
+                                            srt_for_export = translated_srt
+                                            print(f"✅ Legendas geradas ({len(srt_for_export)} chars)")
+                                        else:
+                                            print("⚠️ Gemini retornou SRT vazio na tradução")
+                                    else:
+                                        print("⚠️ Groq não retornou transcrição válida")
+                                except Exception as sub_e:
+                                    print(f"⚠️ Erro ao gerar legendas: {sub_e}")
+
+                            # 6. Montar title_url
                             import urllib.parse
+
                             title_params = urllib.parse.urlencode({
                                 "text": title_text,
                                 "image_url": screenshot_url,
@@ -583,7 +637,7 @@ EXEMPLOS
                             })
                             title_url = f"https://habulaj-recurve-api-img.hf.space/cover/title?{title_params}"
                             
-                            # 6. Chamar API de video export
+                            # 7. Chamar API de video export
                             print(f"🎬 Chamando video export API...")
                             export_payload = {
                                 "video_url": export_video_url,
@@ -597,6 +651,10 @@ EXEMPLOS
                                 "video_height": crop_h,
                                 "video_template": video_template
                             }
+                            if srt_for_export:
+                                export_payload["subtitles"] = srt_for_export
+                                print(f"📝 Legendas incluídas no export")
+
                             export_resp = requests.post(
                                 "https://habulaj-recurve-videos-export.hf.space/video/export",
                                 json=export_payload,