Update main.py

main.py

@@ -387,26 +387,43 @@ Se o contexto enviado pelo usuário não for verdadeiro ou estiver impreciso, ig
         if cut_file and os.path.exists(cut_file.name): os.unlink(cut_file.name)
 
 
-class FilterRequest(BaseModel):
-    media_url: Optional[str] = None
-    video_url: Optional[str] = None
-    image_url: Optional[str] = None
-    context: Optional[str] = None
-    discord_id: Optional[int] = 1
-
-@app.post("/filter")
-async def filter_endpoint(request: FilterRequest):
+@app.api_route("/filter-girlsmoodaily", methods=["GET", "POST"])
+async def filter_endpoint():
     if not client:
         raise HTTPException(status_code=500, detail="Gemini client is not initialized")
     temp_file = None
     try:
-        url_to_download = request.media_url or request.video_url or request.image_url
+        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"
+        }
+        
+        # Buscar 1 post pendente para filtro
+        select_url = f"{supabase_url}/rest/v1/posts?select=*&filter_message=is.null&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}")
+            
+        records = res_get.json()
+        if not records:
+            return {"status": "ok", "message": "Nenhuma postagem pendente para ser filtrada."}
+            
+        record = records[0]
+        record_id = record.get("id")
+        url_to_download = record.get("ig_post_url")
+        context_text = record.get("ig_caption", "")
+        
         if not url_to_download:
-            raise HTTPException(status_code=400, detail="URL da mídia (media_url, video_url ou image_url) é obrigatória")
+            raise HTTPException(status_code=400, detail=f"Registro ID {record_id} falhou: ig_post_url inválida.")
 
-        bot_names = {1: "Diana", 2: "Vicky", 3: "Amanda"}
-        discord_id = 1 if request.discord_id is None else request.discord_id
-        agent_name = bot_names.get(discord_id, "Sistema")
+        discord_id = 1
+        agent_name = "Diana"
         
         try:
             import urllib.parse
@@ -441,7 +458,7 @@ async def filter_endpoint(request: FilterRequest):
 
         media_path_to_analyze = temp_file.name
 
-        contexto_add = f"\n\nContexto Adicional / Legenda Original:\n{request.context}" if request.context else ""
+        contexto_add = f"\n\nContexto Adicional / Legenda Original:\n{context_text}" if context_text else ""
 
         prompt = f"""Você é DIANA, a Curadora de Conteúdo da @girlsmoodaily no Instagram.
 
@@ -556,9 +573,23 @@ Reprovando (exemplo 2):
                 timeout=5
             )
         except Exception as e:
-            print(f"⚠️ Erro ao enviar mensagem para o Discord: {e}")
-
-        return filter_data
+            print(f"⚠️ Erro ao enviar log para o Discord: {e}")
+
+        # Atualizar no Supabase
+        update_url = f"{supabase_url}/rest/v1/posts?id=eq.{record_id}"
+        patch_payload = {
+            "filter_message": filter_data.get("filter_message"),
+            "approved_filter": filter_data.get("approved_filter")
+        }
+        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}")
+
+        return {
+            "success": True, 
+            "record_id": record_id, 
+            "filter_data": filter_data
+        }
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Erro interno: {str(e)}")
     finally: