Update main.py

main.py

@@ -39,7 +39,7 @@ class ProcessLogger:
 
 sys.path.append(str(Path(__file__).parent.parent))
 from detect_crop_image import detect_and_crop_image
-from detect_crop_video import detect_and_crop_video
+from detect_crop_video import detect_and_crop_video, detect_and_crop_text
 
 from gemini_webapi import GeminiClient
 from gemini_webapi.constants import Model
@@ -399,48 +399,84 @@ async def run_process_url(request: ProcessUrlRequest, record_id: int):
         else:
             logger.log("✂️ Processando vídeo: detectando movimento e bordas...")
             try:
-                cropped_video_path = tempfile.NamedTemporaryFile(delete=False, suffix=".mp4").name
-                crop_status = detect_and_crop_video(video_path_to_analyze, cropped_video_path, text_cut=request.text_cut)
+                # Criar arquivo temporário para o vídeo cortado (Passo 1: Motion Crop)
+                cropped_video_tmp = tempfile.NamedTemporaryFile(delete=False, suffix=".mp4").name
+                cropped_video_path = cropped_video_tmp # Track for finally cleanup
+                # Sempre fazemos o motion crop primeiro sem o text_cut interno
+                crop_status = detect_and_crop_video(video_path_to_analyze, cropped_video_tmp, text_cut=False)
                 
-                if crop_status == "aborted_area_too_small":
-                    logger.log("🚫 Crop abortado: área de texto no centro (segurança).")
-                    
-                    # Notificação do Agente no Discord (Octavio/Diana)
-                    try:
-                        import urllib.parse as _up
-                        reject_msg = f"Analisando aqui, notei que o texto desse vídeo está bem no centro. Se eu cortasse agora, ia acabar destruindo o conteúdo principal. Como a gente preza pela qualidade da página, preferi reprovar esse aqui automaticamente. ❌"
-                        _reject_url = "https://discordmsg.arthurmribeiro51.workers.dev/?" + _up.urlencode({"mensagem": reject_msg, "id": discord_id})
-                        requests.get("https://proxy.onrecurve.com/", params={"quest": _reject_url}, timeout=5)
-                    except Exception as _e_dc:
-                        logger.log(f"⚠️ Erro ao enviar Discord de rejeição automática: {_e_dc}")
-
-                    requests.patch(f"{supabase_url}/rest/v1/posts?id=eq.{record_id}", headers=headers, json={
-                        "approved_filter": False,
-                        "filter_message": "Rejeitado automaticamente: Região útil de texto insuficiente para crop seguro (texto centralizado).",
-                        "result": {"error": "Rejeitado automaticamente: Região útil de texto insuficiente para crop seguro (texto centralizado). Tente outro vídeo."},
-                        "contains_image": False
-                    })
-                    return # Para o processamento aqui
-
-                if crop_status == "success" and os.path.exists(cropped_video_path):
-                    files_to_send.append(cropped_video_path)
-                    logger.log(f"✅ Vídeo cortado com sucesso: {cropped_video_path}")
+                if crop_status == "success" and os.path.exists(cropped_video_tmp):
+                    # Passo 2: Verificação via Gemini (Apenas se o usuário solicitou text_cut)
+                    if request.text_cut:
+                        logger.log("🧠 Verificando presença de textos externos via Gemini (process-url)...")
+                        text_check_model = get_gemini_model("flash")
+                        text_check_prompt = """Analise o vídeo anexado e determine se há textos de edição externa (overlays de terceiros) adicionados após a produção original ou profissional do conteúdo.
+
+Retorne um JSON no formato: {"has_texts": true/false}
+
+Regras de Classificação:
+- Considere como FALSE (Não marcar como texto de edição):
+   - Gráficos de Transmissão: Nomes de indicados, categorias de premiação, placares esportivos ou "lower thirds" que fazem parte da transmissão oficial da TV/Evento.
+   - Créditos Artísticos/Branding Profissional: Logotipos de marcas (ex: "Dior") ou créditos integrados que fazem parte da peça audiovisual original ou de um "lookbook" profissional.
+   - Textos de Ambiente: Placas, telões no fundo do palco ou escritos em roupas.
+   - Estética Profissional: Qualquer texto que tenha sido claramente renderizado como parte da master original do vídeo.
+- Considere como TRUE (Marcar como texto de edição):
+   - Legendas de Redes Sociais: Captions explicativas ou legendas automáticas adicionadas para visualização em redes sociais.
+   - Textos de Engajamento: Frases como "Assista até o final", "O que você achou?", emojis flutuantes ou textos genéricos de meme.
+   - Marcas d'água de editores externos: Textos que identificam perfis de usuários ou editores que não são os produtores originais do evento.
+
+Objetivo: O foco é detectar se o vídeo foi "re-editado" com textos informativos/estáticos típicos de redes sociais. Se o texto parece vir da produção original (Oscar, marcas de luxo, TV), retorne false. Responda apenas com o JSON, sem explicações."""
+                        
+                        try:
+                            text_resp = await client.generate_content(text_check_prompt, files=[cropped_video_tmp], model=text_check_model)
+                            text_data = extract_json_from_text(text_resp.text)
+                            has_texts = text_data.get("has_texts", False) if isinstance(text_data, dict) else False
+                            
+                            if has_texts:
+                                logger.log("🔍 Gemini detectou textos de edição externa. Iniciando OCR...")
+                                ocr_video_tmp = tempfile.NamedTemporaryFile(delete=False, suffix=".mp4").name
+                                text_crop_status = detect_and_crop_text(cropped_video_tmp, ocr_video_tmp)
+                                
+                                if text_crop_status == "aborted_area_too_small":
+                                    logger.log("🚫 Crop abortado: área de texto no centro (segurança).")
+                                    # Notificação do Agente no Discord (Octavio/Diana)
+                                    try:
+                                        import urllib.parse as _up
+                                        reject_msg = f"Analisando aqui, notei que o texto desse vídeo está bem no centro. Se eu cortasse agora, ia acabar destruindo o conteúdo principal. Como a gente preza pela qualidade da página, preferi reprovar esse aqui automaticamente. ❌"
+                                        _reject_url = "https://discordmsg.arthurmribeiro51.workers.dev/?" + _up.urlencode({"mensagem": reject_msg, "id": discord_id})
+                                        requests.get("https://proxy.onrecurve.com/", params={"quest": _reject_url}, timeout=5)
+                                    except Exception as _e_dc:
+                                        logger.log(f"⚠️ Erro ao enviar Discord de rejeição automática: {_e_dc}")
+
+                                    requests.patch(f"{supabase_url}/rest/v1/posts?id=eq.{record_id}", headers=headers, json={
+                                        "approved_filter": False,
+                                        "filter_message": "Rejeitado automaticamente: Região útil de texto insuficiente para crop seguro (texto centralizado).",
+                                        "result": {"error": "Rejeitado automaticamente: Região útil de texto insuficiente para crop seguro (texto centralizado). Tente outro vídeo."},
+                                        "contains_image": False
+                                    })
+                                    if os.path.exists(cropped_video_tmp): os.unlink(cropped_video_tmp)
+                                    if os.path.exists(ocr_video_tmp): os.unlink(ocr_video_tmp)
+                                    return 
+                                elif text_crop_status == "success" and os.path.exists(ocr_video_tmp):
+                                    if os.path.exists(cropped_video_tmp): os.unlink(cropped_video_tmp)
+                                    cropped_video_tmp = ocr_video_tmp
+                                    cropped_video_path = ocr_video_tmp # Update tracking
+                                    logger.log(f"✅ Vídeo cortado via OCR com sucesso em: {cropped_video_tmp}")
+                                else:
+                                    if os.path.exists(ocr_video_tmp): os.unlink(ocr_video_tmp)
+                                    logger.log("⚠️ Crop de OCR ignorado ou falhou, seguindo com motion crop.")
+                            else:
+                                logger.log("✅ Gemini NÃO detectou textos de social media. Pulando OCR.")
+                        except Exception as ge:
+                            logger.log(f"⚠️ Erro na análise Gemini de textos: {ge}. Seguindo apenas com motion crop.")
                     
-                    # Upload antecipado para recurve-save (Performance)
-                    logger.log("☁️ Enviando vídeo cortado antecipadamente para recurve-save...")
-                    with open(cropped_video_path, "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()
-                    exported_cropped_video_url = vid_upload_resp.json().get("url", "")
-                    logger.log(f"✅ URL do vídeo cortado: {exported_cropped_video_url}")
+                    # Final path assignment
+                    cropped_video_path = cropped_video_tmp
+                    logger.log(f"✅ Vídeo (motion/ocr) consolidado em: {cropped_video_path}")
                 else:
                     cropped_video_path = None
                     logger.log("⚠️ Crop de vídeo ignorado ou falhou.")
+                    if os.path.exists(cropped_video_tmp): os.unlink(cropped_video_tmp)
             except Exception as e:
                 cropped_video_path = None
                 logger.log(f"⚠️ Erro no processo de crop: {e}")
@@ -1693,33 +1729,78 @@ async def run_filter_account(account: str, background_tasks: BackgroundTasks = N
         else:
             logger.log("✂️ Processando vídeo: detectando movimento e bordas...")
             try:
-                # Criar arquivo temporário para o vídeo cortado
+                # Criar arquivo temporário para o vídeo cortado (Passo 1: Motion Crop)
                 cropped_video_tmp = tempfile.NamedTemporaryFile(delete=False, suffix='.mp4').name
-                crop_status = detect_and_crop_video(media_path_to_analyze, cropped_video_tmp)
+                cropped_file_path = cropped_video_tmp # Track for finally cleanup
+                # Desativamos o text_cut inicial para fazer a análise do Gemini antes
+                crop_status = detect_and_crop_video(media_path_to_analyze, cropped_video_tmp, text_cut=False)
                 
-                if crop_status == "aborted_area_too_small":
-                    logger.log("🚫 Crop abortado: área de texto no centro (segurança). Reprovando filtro.")
+                if crop_status == "success" and os.path.exists(cropped_video_tmp):
+                    # Passo 2: Verificação via Gemini para saber se deve rodar OCR
+                    logger.log("🧠 Verificando presença de textos externos via Gemini...")
+                    text_check_model = get_gemini_model("flash")
+                    text_check_prompt = """Analise o vídeo anexado e determine se há textos de edição externa (overlays de terceiros) adicionados após a produção original ou profissional do conteúdo.
+
+Retorne um JSON no formato: {"has_texts": true/false}
+
+Regras de Classificação:
+- Considere como FALSE (Não marcar como texto de edição):
+   - Gráficos de Transmissão: Nomes de indicados, categorias de premiação, placares esportivos ou "lower thirds" que fazem parte da transmissão oficial da TV/Evento.
+   - Créditos Artísticos/Branding Profissional: Logotipos de marcas (ex: "Dior") ou créditos integrados que fazem parte da peça audiovisual original ou de um "lookbook" profissional.
+   - Textos de Ambiente: Placas, telões no fundo do palco ou escritos em roupas.
+   - Estética Profissional: Qualquer texto que tenha sido claramente renderizado como parte da master original do vídeo.
+- Considere como TRUE (Marcar como texto de edição):
+   - Legendas de Redes Sociais: Captions explicativas ou legendas automáticas adicionadas para visualização em redes sociais.
+   - Textos de Engajamento: Frases como "Assista até o final", "O que você achou?", emojis flutuantes ou textos genéricos de meme.
+   - Marcas d'água de editores externos: Textos que identificam perfis de usuários ou editores que não são os produtores originais do evento.
+
+Objetivo: O foco é detectar se o vídeo foi "re-editado" com textos informativos/estáticos típicos de redes sociais. Se o texto parece vir da produção original (Oscar, marcas de luxo, TV), retorne false. Responda apenas com o JSON, sem explicações."""
                     
-                    # Notificação do Agente no Discord
                     try:
-                        import urllib.parse as _up
-                        reject_msg = f"Analisei esse vídeo aqui e notei que o texto dele está bem no centro. Se eu fizesse o corte agora, ia acabar cortando partes importantes da mensagem. Por isso, decidi reprovar ele automaticamente para manter a qualidade! ❌"
-                        _reject_url = "https://discordmsg.arthurmribeiro51.workers.dev/?" + _up.urlencode({"mensagem": reject_msg, "id": discord_id})
-                        requests.get("https://proxy.onrecurve.com/", params={"quest": _reject_url}, timeout=5)
-                    except Exception as _e_dc:
-                        logger.log(f"⚠️ Erro ao enviar Discord de rejeição automática (crop): {_e_dc}")
+                        text_resp = await client.generate_content(text_check_prompt, files=[cropped_video_tmp], model=text_check_model)
+                        text_data = extract_json_from_text(text_resp.text)
+                        has_texts = text_data.get("has_texts", False) if isinstance(text_data, dict) else False
+                        
+                        if has_texts:
+                            logger.log("🔍 Gemini detectou textos de edição externa. Iniciando OCR...")
+                            # Passo 3: OCR Crop (Apenas se o Gemini confirmou)
+                            ocr_video_tmp = tempfile.NamedTemporaryFile(delete=False, suffix='.mp4').name
+                            text_crop_status = detect_and_crop_text(cropped_video_tmp, ocr_video_tmp)
+                            
+                            if text_crop_status == "aborted_area_too_small":
+                                logger.log("🚫 Crop abortado: área de texto no centro (segurança). Reprovando filtro.")
+                                # Notificação do Agente no Discord
+                                try:
+                                    import urllib.parse as _up
+                                    reject_msg = f"Analisei esse vídeo aqui e notei que o texto dele está bem no centro. Se eu fizesse o corte agora, ia acabar cortando partes importantes da mensagem. Por isso, decidi reprovar ele automaticamente para manter a qualidade! ❌"
+                                    _reject_url = "https://discordmsg.arthurmribeiro51.workers.dev/?" + _up.urlencode({"mensagem": reject_msg, "id": discord_id})
+                                    requests.get("https://proxy.onrecurve.com/", params={"quest": _reject_url}, timeout=5)
+                                except Exception as _e_dc:
+                                    logger.log(f"⚠️ Erro ao enviar Discord de rejeição automática (crop): {_e_dc}")
+
+                                requests.patch(f"{supabase_url}/rest/v1/posts?id=eq.{record_id}", headers=headers, json={
+                                    "approved_filter": False,
+                                    "filter_message": "Rejeitado automaticamente: Região útil de texto insuficiente para crop seguro (texto centralizado no vídeo).",
+                                    "result": {"error": "Rejeitado automaticamente: Região útil de texto insuficiente para crop seguro (texto centralizado). Tente outro vídeo."},
+                                    "contains_image": False
+                                })
+                                return {"status": "ok", "message": "Postagem rejeitada por crop insuficiente.", "next_steps": {}}
+                            elif text_crop_status == "success" and os.path.exists(ocr_video_tmp):
+                                # Reemplazamos el motion crop original por el OCR crop
+                                if os.path.exists(cropped_video_tmp): os.unlink(cropped_video_tmp)
+                                cropped_video_tmp = ocr_video_tmp
+                                cropped_file_path = ocr_video_tmp # Update tracking
+                                logger.log(f"✅ Vídeo cortado via OCR com sucesso em: {cropped_video_tmp}")
+                            else:
+                                if os.path.exists(ocr_video_tmp): os.unlink(ocr_video_tmp)
+                                logger.log("⚠️ Crop de OCR ignorado ou falhou, mantendo apenas motion crop.")
+                        else:
+                            logger.log("✅ Gemini NÃO detectou textos de social media. Pulando OCR.")
+                    except Exception as ge:
+                        logger.log(f"⚠️ Erro na análise Gemini de textos: {ge}. Seguindo apenas com motion crop.")
 
-                    requests.patch(f"{supabase_url}/rest/v1/posts?id=eq.{record_id}", headers=headers, json={
-                        "approved_filter": False,
-                        "filter_message": "Rejeitado automaticamente: Região útil de texto insuficiente para crop seguro (texto centralizado no vídeo).",
-                        "result": {"error": "Rejeitado automaticamente: Região útil de texto insuficiente para crop seguro (texto centralizado). Tente outro vídeo."},
-                        "contains_image": False
-                    })
-                    return # Para o processamento aqui
-                
-                if crop_status == "success" and os.path.exists(cropped_video_tmp):
                     cropped_file_path = cropped_video_tmp
-                    logger.log(f"✅ Vídeo cortado com sucesso em: {cropped_file_path}")
+                    logger.log(f"✅ Vídeo (motion) consolidado em: {cropped_file_path}")
                 else:
                     logger.log("⚠️ Crop de vídeo não gerou novo arquivo ou foi ignorado.")
                     if os.path.exists(cropped_video_tmp): os.unlink(cropped_video_tmp)