Update routers/analyze.py

routers/analyze.py

@@ -135,11 +135,13 @@ async def rewrite_article_direct(content: str) -> Optional[Dict[str, Any]]:
             "title": result.title,
             "subhead": result.subhead,
             "content": result.content,
-            "sources": getattr(result, "sources", [])
+            "title_instagram": getattr(result, "title_instagram", ""),
+            "content_instagram": getattr(result, "content_instagram", "")
         }
         
         # Validação básica da resposta
-        if all(key in rewritten_data and rewritten_data[key].strip() for key in ["title", "subhead", "content"]) and isinstance(rewritten_data["sources"], list):
+        required_keys = ["title", "subhead", "content", "title_instagram", "content_instagram"]
+        if all(key in rewritten_data and rewritten_data[key].strip() for key in required_keys):
             log.info("Artigo reescrito com sucesso (chamada direta)")
             return {
                 "success": True,
@@ -157,7 +159,7 @@ async def rewrite_article_direct(content: str) -> Optional[Dict[str, Any]]:
                 "raw_response": str(rewritten_data),
                 "status_code": 200,
                 "method": "direct_call",
-                "missing_keys": [key for key in ["title", "subhead", "content"] if not rewritten_data.get(key, "").strip()]
+                "missing_keys": [key for key in required_keys if not rewritten_data.get(key, "").strip()]
             }
             
     except Exception as e:
@@ -197,7 +199,8 @@ async def rewrite_article_http(content: str) -> Optional[Dict[str, Any]]:
                     rewritten_data = json.loads(response_text)
                     
                     # Validação básica da resposta
-                    if all(key in rewritten_data for key in ["title", "subhead", "content"]):
+                    required_keys = ["title", "subhead", "content", "title_instagram", "content_instagram"]
+                    if all(key in rewritten_data for key in required_keys):
                         log.info("Artigo reescrito com sucesso (HTTP)")
                         return {
                             "success": True,
@@ -215,7 +218,7 @@ async def rewrite_article_http(content: str) -> Optional[Dict[str, Any]]:
                             "raw_response": response_text,
                             "status_code": response.status,
                             "method": "http_call",
-                            "missing_keys": [key for key in ["title", "subhead", "content"] if key not in rewritten_data]
+                            "missing_keys": [key for key in required_keys if key not in rewritten_data]
                         }
                         
                 except json.JSONDecodeError as e:
@@ -294,25 +297,18 @@ async def fetch_brazil_interest_news():
         raise HTTPException(status_code=500, detail=f"Erro Supabase: {str(e)}")
 
 async def update_news_rewrite(news_id: int, rewritten_data: Dict[str, str]):
-    """Atualiza a notícia com os dados reescritos e conta as fontes"""
+    """Atualiza a notícia com os dados reescritos incluindo campos do Instagram"""
     try:
         session = await get_http_session()
         url = f"{SUPABASE_URL}/rest/v1/news"
         params = {"id": f"eq.{news_id}"}
         
-        # Conta as fontes - se vazio ou None, usa 1 como padrão
-        sources_list = rewritten_data.get("sources", [])
-        sources_count = len(sources_list) if sources_list else 1
-        
-        # Log para debug
-        log.info(f"Fontes encontradas: {sources_list}")
-        log.info(f"Contagem de fontes: {sources_count}")
-        
         payload = {
             "title_pt": rewritten_data.get("title", ""),
             "text_pt": rewritten_data.get("content", ""),
             "subhead_pt": rewritten_data.get("subhead", ""),
-            "sources": sources_count  # Adiciona a contagem de fontes
+            "title_instagram": rewritten_data.get("title_instagram", ""),
+            "content_instagram": rewritten_data.get("content_instagram", "")
         }
         
         async with session.patch(url, headers=SUPABASE_ROLE_HEADERS, json=payload, params=params) as response:
@@ -321,7 +317,7 @@ async def update_news_rewrite(news_id: int, rewritten_data: Dict[str, str]):
                 log.error(f"Erro ao atualizar notícia - Status: {response.status}, Response: {response_text}")
                 raise HTTPException(status_code=500, detail=f"Erro ao atualizar notícia - Status: {response.status}")
             
-            log.info(f"Notícia {news_id} atualizada com sucesso - Status: {response.status}, Fontes: {sources_count}")
+            log.info(f"Notícia {news_id} atualizada com sucesso - Status: {response.status}")
             
     except Exception as e:
         log.error(f"Erro ao atualizar notícia {news_id}: {str(e)}")