Update main.py

main.py

@@ -210,6 +210,396 @@ async def chat_endpoint(request: ChatRequest):
         raise HTTPException(status_code=500, detail=str(e))
 
 
+class ProcessUrlRequest(BaseModel):
+    url: str
+    context: Optional[str] = ""
+    version: str  # "recurvepop" or "girlsmoodaily"
+    raw_url: Optional[bool] = True
+
+@app.post("/process-url")
+async def process_url_endpoint(request: ProcessUrlRequest):
+    """
+    Processa um post do Instagram a partir de uma URL, sem cadastrar no Supabase.
+    Chama o agente 'process' da conta especificada em `version`.
+    """
+    if not client:
+        raise HTTPException(status_code=500, detail="Gemini client is not initialized")
+
+    temp_file = None
+    cropped_file_path = None
+    cropped_video_path = None
+    screenshot_path = None
+
+    try:
+        from agent_config import AGENTS
+        account = request.version
+        if account not in AGENTS:
+            raise HTTPException(status_code=400, detail=f"Conta '{account}' não configurada.")
+        agent_conf = AGENTS[account]["process"]
+        agent_name = agent_conf["name"]
+
+        # 1. Chamar a API externa para obter informações do post
+        print(f"🔗 Buscando dados do post via API externa: {request.url}")
+        api_headers = {
+            "accept": "*/*",
+            "accept-language": "pt-BR,pt;q=0.9,en-US;q=0.8,en;q=0.7",
+            "content-type": "application/json",
+            "origin": "http://localhost:5173",
+            "referer": "http://localhost:5173/",
+            "user-agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/146.0.0.0 Safari/537.36",
+        }
+        api_payload = {"url": request.url, "videoQuality": "best"}
+        api_resp = requests.post(
+            "https://proud-paper-3751.fly.dev/api/json",
+            headers=api_headers,
+            json=api_payload,
+            timeout=60,
+        )
+        if not api_resp.ok:
+            raise HTTPException(status_code=502, detail=f"Erro na API externa: {api_resp.text}")
+
+        api_data = api_resp.json()
+        video_url = api_data.get("url")
+        context = request.context or api_data.get("caption", "")
+        comments = api_data.get("comments", [])
+
+        if not video_url:
+            raise HTTPException(status_code=502, detail="API externa não retornou URL de mídia.")
+
+        print(f"✅ Mídia obtida: {video_url}")
+
+        # 2. Baixar mídia
+        print(f"📥 Baixando mídia: {video_url}")
+        response = download_file_with_retry(video_url, timeout=600)
+        content_type = response.headers.get("content-type", "").lower()
+        if "image" in content_type:
+            if "png" in content_type: ext = ".png"
+            elif "webp" in content_type: ext = ".webp"
+            else: ext = ".jpg"
+        else:
+            ext = ".webm" if "webm" in content_type else ".mp4"
+
+        temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=ext)
+        for chunk in response.iter_content(chunk_size=1024 * 1024):
+            if chunk:
+                temp_file.write(chunk)
+        temp_file.close()
+
+        video_path_to_analyze = temp_file.name
+        files_to_send = [video_path_to_analyze]
+
+        # 3. Crop
+        if "image" in content_type:
+            print("✂️ Processando imagem: detectando e cortando...")
+            try:
+                cropped_file_path = detect_and_crop_image(video_path_to_analyze)
+                if cropped_file_path and os.path.exists(cropped_file_path):
+                    files_to_send.append(cropped_file_path)
+            except Exception as e:
+                print(f"⚠️ Erro ao cortar imagem: {e}")
+        else:
+            print("✂️ Processando vídeo: detectando e cortando bordas...")
+            try:
+                cropped_video_path = tempfile.NamedTemporaryFile(delete=False, suffix=".mp4").name
+                crop_success = detect_and_crop_video(video_path_to_analyze, cropped_video_path)
+                if crop_success and os.path.exists(cropped_video_path):
+                    files_to_send.append(cropped_video_path)
+                    print("✅ Vídeo cortado adicionado como segundo anexo")
+                else:
+                    cropped_video_path = None
+            except Exception as e:
+                cropped_video_path = None
+                print(f"⚠️ Erro ao cortar vídeo: {e}")
+
+        # 4. Montar contextos
+        contexto_add = f"\n{context}" if context else ""
+        comentarios_add = ""
+        if comments:
+            comentarios_add = "\nCOMENTÁRIOS DO POST (Use como forte inspiração para criar títulos mais reais e humanizados):\n"
+            for c in comments:
+                if isinstance(c, dict) and (text := c.get("text", "").strip()):
+                    comentarios_add += f"- {text} ({c.get('like_count', 0)} curtidas)\n"
+
+        if "image" in content_type:
+            tipo_conteudo_add = "\n\nCONTEXTO DO CONTEÚDO: Este post é uma IMAGEM. O título vai aparecer em cima da imagem."
+        else:
+            tipo_conteudo_add = ""
+
+        filter_message_add = ""
+
+        prompt = agent_conf["get_prompt"](
+            date_str=time.strftime("%d/%m/%Y"),
+            contexto_add=contexto_add,
+            comentarios_add=comentarios_add,
+            tipo_conteudo_add=tipo_conteudo_add,
+            filter_message_add=filter_message_add,
+        )
+
+        # 5. Gerar com Gemini
+        model_obj = get_gemini_model("flash")
+        print(f"🧠 Enviando para Gemini (flash) [{agent_name}]...")
+        response_gemini = await client.generate_content(prompt, files=files_to_send, model=model_obj)
+
+        titles_data = extract_json_from_text(response_gemini.text)
+        if not titles_data:
+            return JSONResponse(
+                content={"raw_content": response_gemini.text, "error": "Failed to parse JSON"},
+                status_code=200,
+            )
+
+        result_json = titles_data if isinstance(titles_data, list) else [titles_data]
+
+        # 6. Video export (mesma lógica do /process/{account})
+        final_content_url = None
+        srt_for_export = None
+        if result_json:
+            result_data = result_json[0] if isinstance(result_json[0], dict) else {}
+            title_text = result_data.get("title", "")
+
+            if "image" not in content_type and title_text:
+                title_text = title_text[0].upper() + title_text[1:] if title_text else title_text
+                try:
+                    video_for_export = (
+                        cropped_video_path
+                        if cropped_video_path and os.path.exists(cropped_video_path)
+                        else temp_file.name
+                    )
+
+                    screenshot_path = tempfile.NamedTemporaryFile(delete=False, suffix=".jpg").name
+                    duration_probe = subprocess.run(
+                        ["ffprobe", "-v", "quiet", "-show_entries", "format=duration", "-of", "csv=p=0", video_for_export],
+                        capture_output=True, text=True,
+                    )
+                    video_duration = (
+                        float(duration_probe.stdout.strip())
+                        if duration_probe.returncode == 0 and duration_probe.stdout.strip()
+                        else 10.0
+                    )
+                    middle_time = str(video_duration / 2)
+
+                    ffmpeg_ss = subprocess.run(
+                        ["ffmpeg", "-y", "-i", video_for_export, "-ss", middle_time, "-frames:v", "1", screenshot_path],
+                        capture_output=True, text=True,
+                    )
+                    if ffmpeg_ss.returncode == 0:
+                        print("📸 Enviando screenshot para recurve-save...")
+                        with open(screenshot_path, "rb") as ss_f:
+                            upload_resp = requests.post(
+                                "https://habulaj-recurve-save.hf.space/upload",
+                                files={"files": ("screenshot.jpg", ss_f, "image/jpeg")},
+                                timeout=60,
+                            )
+                            upload_resp.raise_for_status()
+                        screenshot_url = upload_resp.json().get("url", "")
+
+                        if screenshot_url:
+                            # Upload vídeo cortado
+                            cropped_video_url = None
+                            if video_for_export != temp_file.name:
+                                print("☁️ Enviando vídeo cortado para recurve-save...")
+                                with open(video_for_export, "rb") as vf:
+                                    vid_upload_resp = requests.post(
+                                        "https://habulaj-recurve-save.hf.space/upload",
+                                        files={"files": ("cropped_video.mp4", vf, "video/mp4")},
+                                        data={"long_duration": "yes"},
+                                        timeout=120,
+                                    )
+                                    vid_upload_resp.raise_for_status()
+                                cropped_video_url = vid_upload_resp.json().get("url", "")
+
+                            export_video_url = cropped_video_url if cropped_video_url else video_url
+
+                            import cv2
+                            cap = cv2.VideoCapture(video_for_export)
+                            crop_w = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH)) if cap.isOpened() else 1080
+                            crop_h = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT)) if cap.isOpened() else 1920
+                            cap.release()
+
+                            canvas_width = 1080
+                            image_height = 616
+                            video_dest_height = 1920 - image_height
+                            video_template = "horizontal" if crop_w > crop_h else "vertical"
+
+                            if video_template == "vertical":
+                                scaled_h = int(crop_h * canvas_width / crop_w) if crop_w > 0 else crop_h
+                                video_x, video_y = 0, max(0, (scaled_h - video_dest_height) // 2)
+                            else:
+                                video_x, video_y = 0, 0
+
+                            # Legendas
+                            needs_legenda = result_data.get("legenda", False)
+                            if needs_legenda:
+                                print("🎙️ Gerando legendas (Groq + Gemini)...")
+                                try:
+                                    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():
+                                        translate_prompt = f"""IDIOMA: A legenda traduzida DEVE ser inteiramente em PORTUGUÊS DO BRASIL (pt-BR). Independente do idioma original do vídeo.
+
+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.
+Deve traduzir exatamente o texto da legenda observando o contexto, não é pra migrar, por exemplo, textos de um bloco de legenda pra outro. Deve traduzir exatamente o texto de cada bloco de legenda, manter sempre as palavras, nunca retirar.
+Mande o SRT completo, sem textos adicionais na resposta, apenas o SRT traduzido. A legenda acima é uma base gerada pelo Whisper que precisa ser limpa e traduzida, não o resultado final.
+A legenda deve ser totalmente traduzida corretamente analisando o contexto e a entonação de falar. Se alguém estiver gritando, ESCREVA MAIÚSCULO! etc... Adapte gírias e qualquer coisa do tipo. Não deve ser literal a tradução, deve se adaptar.
+
+MÚSICA E LETRAS:
+- NUNCA LEGENDE MÚSICAS OU CANÇÕES.
+- Se houver música de fundo ou pessoas cantando uma música, IGNORE COMPLETAMENTE e não inclua na legenda.
+- VOCÊ DEVE LEGENDAR APENAS DIÁLOGOS E FALAS REAIS.
+
+PALAVRÕES E CENSURA:
+- Você DEVE censurar palavras de baixo calão e palavrões pesados utilizando asteriscos.
+- Substitua a maior parte da palavra censurada e mantenha apenas as primeiras letras.
+- Exemplo: "filha da puta" se torna "filha da pu**", "caralho" se torna "caral**", "merda" se torna "merd*", "foda" se torna "fod*".
+
+EXTREMAMENTE IMPORTANTE: NUNCA legende músicas (quando detectar que é uma música, não legende), apenas diálogos falados. Nunca altere o timing das legendas, deve ser exatamente igual ao original de referência. Nunca legende ações também, exemplo: [Música alta], [Música de encerramento], etc. Deve ser apenas, unicamente, diálogo humano.
+
+EXEMPLO:
+
+(Original): 1
+00:00:01,000 --> 00:00:04,000
+hey what are you doing here i thought you left already
+
+2
+00:00:04,500 --> 00:00:07,200
+yeah i was going to but then i realized i forgot my keys
+
+3
+00:00:07,900 --> 00:00:10,500
+you always forget something man this is crazy
+
+4
+00:00:11,000 --> 00:00:14,000
+relax it's not a big deal stop acting like that
+
+5
+00:00:14,500 --> 00:00:17,800
+i am not acting you said you would be on time
+
+6
+00:00:18,000 --> 00:00:21,500
+okay okay i'm sorry can we just go now
+
+7
+00:00:22,000 --> 00:00:25,000
+fine but if we are late again you are a son of a bitch
+
+(Traduzido, como você deveria traduzir): 1
+00:00:01,000 --> 00:00:04,000
+Ué, o que você tá fazendo aqui? Não era pra você já ter ido embora?
+
+2
+00:00:04,500 --> 00:00:07,200
+Eu ia, mas aí percebi que esqueci minhas chaves.
+
+3
+00:00:07,900 --> 00:00:10,500
+Cara, você SEMPRE esquece alguma coisa, isso é surreal!
+
+4
+00:00:11,000 --> 00:00:14,000
+Ah, relaxa! Não é o fim do mundo, para de drama.
+
+5
+00:00:14,500 --> 00:00:17,800
+Não é drama! Você falou que ia chegar no horário!
+
+6
+00:00:18,000 --> 00:00:21,500
+Tá, tá... foi mal. Bora logo?
+
+7
+00:00:22,000 --> 00:00:25,000
+Tá bom, mas se a gente se atrasar de novo, você é um filha da pu**!
+
+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()
+                                        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
+                                except Exception as sub_e:
+                                    print(f"⚠️ Erro ao gerar legendas: {sub_e}")
+
+                            import urllib.parse
+                            title_params = urllib.parse.urlencode({"text": title_text, "version": account})
+                            title_url = f"https://habulaj-recurve-api-img.hf.space/cover/title?{title_params}&image_url={screenshot_url}"
+                            result_json[0]["title_url"] = title_url
+
+                            export_payload = {
+                                "video_url": export_video_url,
+                                "title_url": title_url,
+                                "image_height": image_height,
+                                "cut_start": 0,
+                                "cut_end": video_duration,
+                                "video_x": video_x,
+                                "video_y": video_y,
+                                "video_width": crop_w,
+                                "video_height": crop_h,
+                                "video_template": video_template,
+                            }
+                            if srt_for_export:
+                                export_payload["subtitles"] = srt_for_export
+
+                            print("🎬 Chamando video export API...")
+                            export_resp = requests.post(
+                                "https://habulaj-recurve-videos-export.hf.space/video/export",
+                                json=export_payload,
+                                timeout=600,
+                            )
+                            if not export_resp.ok:
+                                raise Exception(f"API de video export falhou: {export_resp.status_code}: {export_resp.text}")
+                            export_data = export_resp.json()
+                            final_content_url = export_data.get("video_url")
+                            print(f"✅ Vídeo exportado: {final_content_url}")
+                except Exception as ve:
+                    print(f"⚠️ Erro no video export: {ve}")
+                    raise Exception(f"Falha no video export: {ve}")
+
+            elif "image" in content_type and title_text and result_data.get("result_type") == "meme":
+                try:
+                    img_for_meme = (
+                        cropped_file_path
+                        if cropped_file_path and os.path.exists(cropped_file_path)
+                        else temp_file.name
+                    )
+                    import urllib.parse
+                    meme_params = urllib.parse.urlencode({"text": title_text})
+                    final_content_url = f"https://habulaj-recurve-api-img.hf.space/meme?{meme_params}"
+                    result_json[0]["title_url"] = final_content_url
+                except Exception as e:
+                    print(f"⚠️ Erro ao gerar URL do meme: {e}")
+
+        if result_json and srt_for_export and isinstance(result_json[0], dict):
+            result_json[0]["subtitle_srt"] = srt_for_export
+
+        if final_content_url and isinstance(result_json[0], dict):
+            result_json[0]["final_content_url"] = final_content_url
+
+        return result_json
+
+    except HTTPException:
+        raise
+    except Exception as e:
+        print(f"⚠️ Erro em /process-url: {e}")
+        raise HTTPException(status_code=500, detail=str(e))
+    finally:
+        if temp_file and os.path.exists(temp_file.name): os.unlink(temp_file.name)
+        if cropped_file_path and os.path.exists(cropped_file_path): os.unlink(cropped_file_path)
+        if cropped_video_path and os.path.exists(cropped_video_path): os.unlink(cropped_video_path)
+        if screenshot_path and os.path.exists(screenshot_path): os.unlink(screenshot_path)
+
+
 @app.api_route("/process/{account}", methods=["GET", "POST"])
 async def process_account_endpoint(account: str):
     if not client:
@@ -475,14 +865,83 @@ async def process_account_endpoint(account: str):
 
                                     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).
+                                        translate_prompt = f"""IDIOMA: A legenda traduzida DEVE ser inteiramente em PORTUGUÊS DO BRASIL (pt-BR). Independente do idioma original do vídeo.
 
 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**").
+Deve traduzir exatamente o texto da legenda observando o contexto, não é pra migrar, por exemplo, textos de um bloco de legenda pra outro. Deve traduzir exatamente o texto de cada bloco de legenda, manter sempre as palavras, nunca retirar.
+Mande o SRT completo, sem textos adicionais na resposta, apenas o SRT traduzido. A legenda acima é uma base gerada pelo Whisper que precisa ser limpa e traduzida, não o resultado final.
+A legenda deve ser totalmente traduzida corretamente analisando o contexto e a entonação de falar. Se alguém estiver gritando, ESCREVA MAIÚSCULO! etc... Adapte gírias e qualquer coisa do tipo. Não deve ser literal a tradução, deve se adaptar.
+
+MÚSICA E LETRAS:
+- NUNCA LEGENDE MÚSICAS OU CANÇÕES.
+- Se houver música de fundo ou pessoas cantando uma música, IGNORE COMPLETAMENTE e não inclua na legenda.
+- VOCÊ DEVE LEGENDAR APENAS DIÁLOGOS E FALAS REAIS.
+
+PALAVRÕES E CENSURA:
+- Você DEVE censurar palavras de baixo calão e palavrões pesados utilizando asteriscos.
+- Substitua a maior parte da palavra censurada e mantenha apenas as primeiras letras.
+- Exemplo: "filha da puta" se torna "filha da pu**", "caralho" se torna "caral**", "merda" se torna "merd*", "foda" se torna "fod*".
+
+EXTREMAMENTE IMPORTANTE: NUNCA legende músicas (quando detectar que é uma música, não legende), apenas diálogos falados. Nunca altere o timing das legendas, deve ser exatamente igual ao original de referência. Nunca legende ações também, exemplo: [Música alta], [Música de encerramento], etc. Deve ser apenas, unicamente, diálogo humano.
+
+EXEMPLO:
+
+(Original): 1
+00:00:01,000 --> 00:00:04,000
+hey what are you doing here i thought you left already
+
+2
+00:00:04,500 --> 00:00:07,200
+yeah i was going to but then i realized i forgot my keys
+
+3
+00:00:07,900 --> 00:00:10,500
+you always forget something man this is crazy
+
+4
+00:00:11,000 --> 00:00:14,000
+relax it's not a big deal stop acting like that
+
+5
+00:00:14,500 --> 00:00:17,800
+i am not acting you said you would be on time
+
+6
+00:00:18,000 --> 00:00:21,500
+okay okay i'm sorry can we just go now
+
+7
+00:00:22,000 --> 00:00:25,000
+fine but if we are late again you are a son of a bitch
+
+(Traduzido, como você deveria traduzir): 1
+00:00:01,000 --> 00:00:04,000
+Ué, o que você tá fazendo aqui? Não era pra você já ter ido embora?
+
+2
+00:00:04,500 --> 00:00:07,200
+Eu ia, mas aí percebi que esqueci minhas chaves.
+
+3
+00:00:07,900 --> 00:00:10,500
+Cara, você SEMPRE esquece alguma coisa, isso é surreal!
+
+4
+00:00:11,000 --> 00:00:14,000
+Ah, relaxa! Não é o fim do mundo, para de drama.
+
+5
+00:00:14,500 --> 00:00:17,800
+Não é drama! Você falou que ia chegar no horário!
+
+6
+00:00:18,000 --> 00:00:21,500
+Tá, tá... foi mal. Bora logo?
+
+7
+00:00:22,000 --> 00:00:25,000
+Tá bom, mas se a gente se atrasar de novo, você é um filha da pu**!
 
 LEGENDA ORIGINAL:
 {srt_raw}"""