Update main.py

main.py

@@ -217,29 +217,74 @@ class ProcessUrlRequest(BaseModel):
     raw_url: Optional[bool] = True
 
 @app.post("/process-url")
-async def process_url_endpoint(request: ProcessUrlRequest):
+async def process_url_endpoint(request: ProcessUrlRequest, background_tasks: BackgroundTasks):
     """
-    Processa um post do Instagram a partir de uma URL, sem cadastrar no Supabase.
-    Chama o agente 'process' da conta especificada em `version`.
+    Cria um registro inicial no Supabase e envia para processamento em background.
+    """
+    supabase_url = os.getenv("SUPABASE_URL", "").rstrip("/")
+    supabase_key = os.getenv("SUPABASE_KEY", "")
+    if not supabase_url or not supabase_key:
+        raise HTTPException(status_code=500, detail="Credenciais do Supabase não configuradas no ambiente.")
+        
+    headers = {
+        "apikey": supabase_key,
+        "Authorization": f"Bearer {supabase_key}",
+        "Content-Type": "application/json",
+        "Prefer": "return=representation"
+    }
+
+    # Inserir no Supabase com user_created=True
+    post_payload = {
+        "ig_post_url": request.url,
+        "ig_caption": request.context,
+        "account_target": request.version,
+        "user_created": True,
+        "published": False
+    }
+
+    res_post = requests.post(f"{supabase_url}/rest/v1/posts", headers=headers, json=post_payload, timeout=10)
+    if not res_post.ok:
+        raise HTTPException(status_code=500, detail=f"Erro ao criar registro no Supabase: {res_post.text}")
+
+    records = res_post.json()
+    record_id = records[0].get("id") if records else None
+
+    if not record_id:
+        raise HTTPException(status_code=500, detail="Registro criado, mas ID não retornado pelo Supabase.")
+
+    background_tasks.add_task(run_process_url, request, record_id)
+    return {"status": "ok", "message": "Solicitação em processamento no background.", "record_id": record_id}
+
+async def run_process_url(request: ProcessUrlRequest, record_id: int):
+    """
+    Processa um post do Instagram a partir de uma URL em background, salvando o resultado.
     """
     if not client:
-        raise HTTPException(status_code=500, detail="Gemini client is not initialized")
+        print("Gemini client is not initialized")
+        return
 
     temp_file = None
     cropped_file_path = None
     cropped_video_path = None
     screenshot_path = None
+    supabase_url = os.getenv("SUPABASE_URL", "").rstrip("/")
+    supabase_key = os.getenv("SUPABASE_KEY", "")
+    headers = {
+        "apikey": supabase_key,
+        "Authorization": f"Bearer {supabase_key}",
+        "Content-Type": "application/json"
+    }
 
     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.")
+            raise Exception(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}")
+        print(f"🔗 Buscando dados do post via API externa (process-url): {request.url}")
         api_headers = {
             "accept": "*/*",
             "accept-language": "pt-BR,pt;q=0.9,en-US;q=0.8,en;q=0.7",
@@ -256,7 +301,7 @@ async def process_url_endpoint(request: ProcessUrlRequest):
             timeout=60,
         )
         if not api_resp.ok:
-            raise HTTPException(status_code=502, detail=f"Erro na API externa: {api_resp.text}")
+            raise Exception(f"Erro na API externa: {api_resp.text}")
 
         api_data = api_resp.json()
         video_url = api_data.get("url")
@@ -264,7 +309,7 @@ async def process_url_endpoint(request: ProcessUrlRequest):
         comments = api_data.get("comments", [])
 
         if not video_url:
-            raise HTTPException(status_code=502, detail="API externa não retornou URL de mídia.")
+            raise Exception("API externa não retornou URL de mídia.")
 
         print(f"✅ Mídia obtida: {video_url}")
 
@@ -272,7 +317,19 @@ async def process_url_endpoint(request: ProcessUrlRequest):
         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:
+        is_image = "image" in content_type
+        
+        post_type = "image" if is_image else "video"
+        
+        # Atualizar type e ig_post_url com o que baixamos
+        patch_initial = {
+            "type": post_type,
+            "ig_post_url": video_url,
+            "ig_caption": context
+        }
+        requests.patch(f"{supabase_url}/rest/v1/posts?id=eq.{record_id}", headers=headers, json=patch_initial)
+
+        if is_image:
             if "png" in content_type: ext = ".png"
             elif "webp" in content_type: ext = ".webp"
             else: ext = ".jpg"
@@ -289,7 +346,7 @@ async def process_url_endpoint(request: ProcessUrlRequest):
         files_to_send = [video_path_to_analyze]
 
         # 3. Crop
-        if "image" in content_type:
+        if is_image:
             print("✂️ Processando imagem: detectando e cortando...")
             try:
                 cropped_file_path = detect_and_crop_image(video_path_to_analyze)
@@ -320,7 +377,7 @@ async def process_url_endpoint(request: ProcessUrlRequest):
                 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:
+        if is_image:
             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 = ""
@@ -337,26 +394,26 @@ async def process_url_endpoint(request: ProcessUrlRequest):
 
         # 5. Gerar com Gemini
         model_obj = get_gemini_model("flash")
-        print(f"🧠 Enviando para Gemini (flash) [{agent_name}]...")
+        print(f"🧠 Enviando para Gemini (flash) [{agent_name}] process-url...")
         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,
-            )
+            print("Failed to parse JSON")
+            return
 
         result_json = titles_data if isinstance(titles_data, list) else [titles_data]
 
-        # 6. Video export (mesma lógica do /process/{account})
+        # 6. Video export
         final_content_url = None
         srt_for_export = None
+        exported_cropped_video_url = 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:
+            if not is_image and title_text:
                 title_text = title_text[0].upper() + title_text[1:] if title_text else title_text
                 try:
                     video_for_export = (
@@ -394,7 +451,6 @@ async def process_url_endpoint(request: ProcessUrlRequest):
 
                         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:
@@ -405,9 +461,9 @@ async def process_url_endpoint(request: ProcessUrlRequest):
                                         timeout=120,
                                     )
                                     vid_upload_resp.raise_for_status()
-                                cropped_video_url = vid_upload_resp.json().get("url", "")
+                                exported_cropped_video_url = vid_upload_resp.json().get("url", "")
 
-                            export_video_url = cropped_video_url if cropped_video_url else video_url
+                            export_video_url = exported_cropped_video_url if exported_cropped_video_url else video_url
 
                             import cv2
                             cap = cv2.VideoCapture(video_for_export)
@@ -566,7 +622,7 @@ LEGENDA ORIGINAL:
                     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":
+            elif is_image and title_text and result_data.get("result_type") == "meme":
                 try:
                     img_for_meme = (
                         cropped_file_path
@@ -586,13 +642,20 @@ LEGENDA ORIGINAL:
         if final_content_url and isinstance(result_json[0], dict):
             result_json[0]["final_content_url"] = final_content_url
 
-        return result_json
+        # Salvar o resultado no banco
+        patch_final = {
+            "result": result_json
+        }
+        if final_content_url:
+            patch_final["final_content_url"] = final_content_url
+        if exported_cropped_video_url:
+            patch_final["crop_content_url"] = exported_cropped_video_url
+            
+        requests.patch(f"{supabase_url}/rest/v1/posts?id=eq.{record_id}", headers=headers, json=patch_final)
+        print(f"✅ /process-url via Background task concluído! Record: {record_id}")
 
-    except HTTPException:
-        raise
     except Exception as e:
-        print(f"⚠️ Erro em /process-url: {e}")
-        raise HTTPException(status_code=500, detail=str(e))
+        print(f"⚠️ Erro em run_process_url background: {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)
@@ -630,7 +693,7 @@ async def process_account_endpoint(account: str):
         system_discord_id = AGENTS[account].get("system_discord_id", 0)
 
         # Buscar 1 post aprovado pelo filtro mas ainda não processado
-        select_url = f"{supabase_url}/rest/v1/posts?select=*&account_target=eq.{account}&approved_filter=eq.true&result=is.null&limit=1"
+        select_url = f"{supabase_url}/rest/v1/posts?select=*&account_target=eq.{account}&approved_filter=eq.true&result=is.null&user_created=eq.false&limit=1"
         res_get = requests.get(select_url, headers=headers, timeout=10)
         if not res_get.ok:
             raise HTTPException(status_code=500, detail=f"Erro ao ler posts: {res_get.text}")
@@ -1223,7 +1286,7 @@ async def run_filter_account(account: str):
         system_discord_id = AGENTS[account].get("system_discord_id", 0)
 
         # Buscar 1 post pendente para filtro para essa conta
-        select_url = f"{supabase_url}/rest/v1/posts?select=*&account_target=eq.{account}&filter_message=is.null&limit=1"
+        select_url = f"{supabase_url}/rest/v1/posts?select=*&account_target=eq.{account}&filter_message=is.null&user_created=eq.false&limit=1"
         res_get = requests.get(select_url, headers=headers, timeout=10)
         if not res_get.ok:
             raise HTTPException(status_code=500, detail=f"Erro ao ler posts: {res_get.text}")
@@ -1429,7 +1492,7 @@ async def publish_account_endpoint(account: str):
         system_discord_id = AGENTS[account].get("system_discord_id", 0)
 
         # Buscar 1 post pronto para publicação
-        select_url = f"{supabase_url}/rest/v1/posts?select=*&account_target=eq.{account}&result=not.is.null&final_content_url=not.is.null&published=eq.false&or=(superior_needs_verification.is.null,superior_needs_verification.eq.false)&limit=1"
+        select_url = f"{supabase_url}/rest/v1/posts?select=*&account_target=eq.{account}&result=not.is.null&final_content_url=not.is.null&published=eq.false&user_created=eq.false&or=(superior_needs_verification.is.null,superior_needs_verification.eq.false)&limit=1"
         res_get = requests.get(select_url, headers=headers, timeout=10)
         if not res_get.ok:
             raise HTTPException(status_code=500, detail=f"Erro ao ler posts: {res_get.text}")