Update main.py

main.py

@@ -558,10 +558,12 @@ EXEMPLOS
             # Se for imagem e result_type == meme
             elif 'image' in content_type and title_text and result_data.get("result_type") == "meme":
                 try:
+                    contains_image = record.get("contains_image", False)
                     img_for_meme = cropped_file_path if cropped_file_path and os.path.exists(cropped_file_path) else temp_file.name
                     
+                    # Nano Banana Pro: só roda se tiver imagem de apoio E precisar de correção
                     image_needs_correction = record.get("image_needs_correction", False)
-                    if image_needs_correction:
+                    if contains_image and image_needs_correction:
                         print("🍌 Chamando Nano Banana Pro (Gemini) para limpar a imagem...")
                         clean_prompt = "Anexei uma imagem que pode ser um post de rede social, um meme, um screenshot ou qualquer composição visual que mistura texto com imagem. Sua única tarefa é limpar essa imagem, removendo tudo que não faz parte do conteúdo visual original. Remova completamente qualquer texto sobreposto, seja título, legenda, frase, username, arroba, logo ou qualquer marcação de outras páginas ou plataformas. Onde o texto estiver sobreposto diretamente na imagem, remova-o e reconstrua o fundo de forma coerente com o estilo visual ao redor. Preserve absolutamente tudo da imagem original: cores, iluminação, estilo artístico, traços, texturas, proporções, enquadramento, clima e contexto visual. Não altere, não melhore, não filtre e não modifique nada além do que for necessário para a remoção dos textos e marcas. Retorne apenas a imagem limpa, sem nenhum texto, sem nenhuma explicação, sem nenhum comentário. Só a imagem, mantendo a mesma proporção e tamanho."
                         nano_model = get_gemini_model("pro")
@@ -589,24 +591,31 @@ EXEMPLOS
                         else:
                             print("⚠️ Gemini não retornou imagem limpa. Usando a original.")
                     
-                    print("📸 Enviando imagem base do meme para recurve-save...")
-                    with open(img_for_meme, 'rb') as img_f:
-                        upload_resp = requests.post(
-                            "https://habulaj-recurve-save.hf.space/upload",
-                            files={'files': ('meme_base.jpg', img_f, 'image/jpeg')},
-                            timeout=60
-                        )
-                        upload_resp.raise_for_status()
-                    uploaded_img_url = upload_resp.json().get("url", "")
-                    
-                    if uploaded_img_url:
-                        import urllib.parse
-                        meme_params = urllib.parse.urlencode({
-                            "text": title_text,
-                            "image_url": uploaded_img_url
-                        })
+                    import urllib.parse
+                    if contains_image:
+                        # Tem imagem de apoio: faz upload e inclui image_url no meme
+                        print("📸 Enviando imagem base do meme para recurve-save...")
+                        with open(img_for_meme, 'rb') as img_f:
+                            upload_resp = requests.post(
+                                "https://habulaj-recurve-save.hf.space/upload",
+                                files={'files': ('meme_base.jpg', img_f, 'image/jpeg')},
+                                timeout=60
+                            )
+                            upload_resp.raise_for_status()
+                        uploaded_img_url = upload_resp.json().get("url", "")
+                        
+                        if uploaded_img_url:
+                            meme_params = urllib.parse.urlencode({
+                                "text": title_text,
+                                "image_url": uploaded_img_url
+                            })
+                            final_content_url = f"https://habulaj-recurve-api-img.hf.space/meme?{meme_params}"
+                            print(f"✅ Meme API URL gerada (com imagem): {final_content_url}")
+                    else:
+                        # Só texto, sem imagem de apoio: manda apenas o text
+                        meme_params = urllib.parse.urlencode({"text": title_text})
                         final_content_url = f"https://habulaj-recurve-api-img.hf.space/meme?{meme_params}"
-                        print(f"✅ Meme API URL gerada: {final_content_url}")
+                        print(f"✅ Meme API URL gerada (só texto): {final_content_url}")
                 except Exception as e:
                     print(f"⚠️ Erro ao gerar URL do meme: {e}")