Update main.py

main.py

@@ -446,12 +446,14 @@ EXEMPLOS
 
         result_json = titles_data if isinstance(titles_data, list) else [titles_data]
         
-        # Se for vídeo, chamar a API de video export
-        final_video_url = None
-        if 'image' not in content_type and result_json:
-            try:
-                title_text = result_json[0].get("title", "") if isinstance(result_json[0], dict) else ""
-                if title_text:
+        final_content_url = None
+        if result_json:
+            result_data = result_json[0] if isinstance(result_json[0], dict) else {}
+            title_text = result_data.get("title", "")
+            
+            # Se for vídeo, chamar a API de video export
+            if 'image' not in content_type and title_text:
+                try:
                     # Usar o vídeo cortado se disponível, senão o original
                     video_for_export = cropped_video_path if cropped_video_path and os.path.exists(cropped_video_path) else temp_file.name
                     
@@ -545,20 +547,45 @@ EXEMPLOS
                             )
                             export_resp.raise_for_status()
                             export_data = export_resp.json()
-                            final_video_url = export_data.get("video_url")
-                            print(f"✅ Vídeo exportado: {final_video_url}")
+                            final_content_url = export_data.get("video_url")
+                            print(f"✅ Vídeo exportado: {final_content_url}")
                         else:
                             print("⚠️ Falha ao obter URL do screenshot enviado.")
-            except Exception as ve:
-                print(f"⚠️ Erro no video export: {ve}")
+                except Exception as ve:
+                    print(f"⚠️ Erro no video export: {ve}")
+
+            # Se for imagem e result_type == meme
+            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
+                    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
+                        })
+                        final_content_url = f"https://habulaj-recurve-api-img.hf.space/meme?{meme_params}"
+                        print(f"✅ Meme API URL gerada: {final_content_url}")
+                except Exception as e:
+                    print(f"⚠️ Erro ao gerar URL do meme: {e}")
         
         # Atualizar no Supabase
         update_url = f"{supabase_url}/rest/v1/posts?id=eq.{record_id}"
         patch_payload = {
             "result": result_json
         }
-        if final_video_url:
-            patch_payload["final_video_url"] = final_video_url
+        if final_content_url:
+            patch_payload["final_content_url"] = final_content_url
         res_patch = requests.patch(update_url, headers=headers, json=patch_payload, timeout=10)
         if not res_patch.ok:
             print(f"⚠️ Erro ao atualizar Supabase: {res_patch.text}")
@@ -569,8 +596,8 @@ EXEMPLOS
             "record_id": record_id,
             "result": result_json
         }
-        if final_video_url:
-            response_data["final_video_url"] = final_video_url
+        if final_content_url:
+            response_data["final_content_url"] = final_content_url
         return response_data
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Erro interno: {str(e)}")