Update routers/inference.py

routers/inference.py

@@ -25,9 +25,8 @@ class NewsResponse(BaseModel):
     title: str
     subhead: str
     content: str
-    sources: list[str]  # Lista de URLs/links utilizados
-    title_instagram: str  # Adicionado campo Instagram title
-    content_instagram: str  # Adicionado campo Instagram description
+    title_instagram: str  # Campo Instagram title
+    content_instagram: str  # Campo Instagram description
 
 # Referência ao diretório de arquivos temporários (deve ser o mesmo do outro módulo)
 TEMP_DIR = Path("/tmp")
@@ -260,33 +259,9 @@ def extract_text_from_response(response):
 
 def extract_sources_from_response(response):
     """
-    Extrai as fontes (URLs) do grounding metadata.
+    Função removida - sources não são mais necessárias.
     """
-    sources = []
-    
-    if not (hasattr(response, 'candidates') and response.candidates):
-        return sources
-    
-    for candidate in response.candidates:
-        if not (hasattr(candidate, 'grounding_metadata') and candidate.grounding_metadata):
-            continue
-        
-        grounding_metadata = candidate.grounding_metadata
-        
-        if hasattr(grounding_metadata, 'grounding_chunks') and grounding_metadata.grounding_chunks:
-            for chunk in grounding_metadata.grounding_chunks:
-                try:
-                    if (hasattr(chunk, 'web') and chunk.web and 
-                        hasattr(chunk.web, 'uri') and chunk.web.uri):
-                        
-                        uri = chunk.web.uri
-                        if uri and uri not in sources:
-                            sources.append(uri)
-                            
-                except Exception:
-                    continue
-    
-    return sources
+    return []
 
 @router.post("/rewrite-news", response_model=NewsResponse)
 async def rewrite_news(news: NewsRequest):
@@ -486,11 +461,15 @@ News base: Noah Centineo Attached to Play Rambo in Prequel Movie 'John Rambo'
 
         logger.info("Resposta do modelo recebida com sucesso")
 
-        # Extrair texto e fontes
+        # Extrair texto
         response_text = extract_text_from_response(response)
-        sources = extract_sources_from_response(response)
 
         logger.info(f"Texto extraído: {len(response_text) if response_text else 0} caracteres")
+        
+        # Log da resposta bruta completa para debug
+        logger.info("=== RESPOSTA BRUTA DA API ===")
+        logger.info(f"Resposta completa: {response_text}")
+        logger.info("=== FIM RESPOSTA BRUTA ===")
 
         # Verificar se o texto está vazio
         if not response_text or response_text.strip() == "":
@@ -523,20 +502,28 @@ News base: Noah Centineo Attached to Play Rambo in Prequel Movie 'John Rambo'
             else:
                 content = "Conteúdo não encontrado"
         
-        # Campos do Instagram (corrigindo os nomes das variáveis)
+        # Campos do Instagram com debug adicional
         insta_title_match = re.search(r"<instagram_title>(.*?)</instagram_title>", response_text, re.DOTALL)
         title_instagram = insta_title_match.group(1).strip() if insta_title_match else "Título Instagram não encontrado"
         
         insta_desc_match = re.search(r"<instagram_description>(.*?)</instagram_description>", response_text, re.DOTALL)
         content_instagram = insta_desc_match.group(1).strip() if insta_desc_match else "Descrição Instagram não encontrada"
         
+        # Debug específico para Instagram fields
+        logger.info(f"Instagram Title Match: {bool(insta_title_match)}")
+        logger.info(f"Instagram Description Match: {bool(insta_desc_match)}")
+        
+        if insta_title_match:
+            logger.info(f"Instagram Title encontrado: {title_instagram[:100]}...")
+        if insta_desc_match:
+            logger.info(f"Instagram Description encontrado: {content_instagram[:100]}...")
+        
         logger.info(f"Processamento concluído com sucesso - Título: {title[:50]}...")
         
         return NewsResponse(
             title=title,
             subhead=subhead,
             content=content,
-            sources=sources,
             title_instagram=title_instagram,
             content_instagram=content_instagram
         )