Update routers/inference.py

routers/inference.py

@@ -2,6 +2,7 @@ import os
 import logging
 import json
 import requests
+import httpx
 from fastapi import APIRouter, HTTPException
 from pydantic import BaseModel
 from google import genai
@@ -19,7 +20,7 @@ router = APIRouter()
 
 class NewsRequest(BaseModel):
     content: str
-    file_id: str  # ID do arquivo temporário ao invés da URL
+    file_id: str = None  # Agora é opcional
 
 class NewsResponse(BaseModel):
     title: str
@@ -108,6 +109,58 @@ def get_brazilian_date_string():
         date_string = now.strftime("%d de %B de %Y")
         return date_string
 
+async def generate_sources_from_content(content: str) -> str:
+    """
+    Chama o endpoint de busca de termos para gerar fontes baseadas no conteúdo.
+    Retorna o file_id do arquivo gerado.
+    """
+    try:
+        # Configurar a URL base - ajuste conforme sua configuração
+        base_url = os.getenv("BASE_URL", "http://localhost:8000")
+        search_url = f"{base_url}/search-terms"
+        
+        # Fazer chamada HTTP para o endpoint de busca
+        payload = {"context": content}
+        
+        async with httpx.AsyncClient(timeout=120.0) as client:
+            response = await client.post(search_url, json=payload)
+            
+            if response.status_code != 200:
+                logger.error(f"Erro na busca de termos: {response.status_code} - {response.text}")
+                raise HTTPException(
+                    status_code=500, 
+                    detail=f"Erro ao gerar fontes: {response.status_code}"
+                )
+            
+            result = response.json()
+            file_info = result.get("file_info", {})
+            file_id = file_info.get("file_id")
+            
+            if not file_id:
+                logger.error("File ID não encontrado na resposta da busca")
+                raise HTTPException(
+                    status_code=500, 
+                    detail="Erro ao obter ID do arquivo de fontes"
+                )
+            
+            logger.info(f"Fontes geradas com sucesso. File ID: {file_id}")
+            logger.info(f"Total de resultados encontrados: {result.get('total_results', 0)}")
+            
+            return file_id
+            
+    except httpx.RequestError as e:
+        logger.error(f"Erro de conexão ao gerar fontes: {str(e)}")
+        raise HTTPException(
+            status_code=503, 
+            detail="Serviço de busca indisponível"
+        )
+    except Exception as e:
+        logger.error(f"Erro inesperado ao gerar fontes: {str(e)}")
+        raise HTTPException(
+            status_code=500, 
+            detail=f"Erro interno ao gerar fontes: {str(e)}"
+        )
+
 def load_sources_file(file_id: str) -> str:
     """
     Carrega o arquivo de fontes pelo ID do arquivo temporário.
@@ -267,6 +320,7 @@ def extract_sources_from_response(response):
 async def rewrite_news(news: NewsRequest):
     """
     Endpoint para reescrever notícias usando o modelo Gemini com arquivo de fontes.
+    Se file_id não for fornecido, gera automaticamente as fontes baseadas no conteúdo.
     """
     try:
         # Verificar API key
@@ -274,8 +328,17 @@ async def rewrite_news(news: NewsRequest):
         if not api_key:
             raise HTTPException(status_code=500, detail="API key não configurada")
         
+        # Se file_id não foi fornecido, gera as fontes automaticamente
+        if not news.file_id:
+            logger.info("File ID não fornecido. Gerando fontes automaticamente...")
+            file_id = await generate_sources_from_content(news.content)
+            logger.info(f"Fontes geradas automaticamente. File ID: {file_id}")
+        else:
+            file_id = news.file_id
+            logger.info(f"Usando file_id fornecido: {file_id}")
+        
         # Carregar arquivo de fontes pelo ID
-        sources_content = load_sources_file(news.file_id)
+        sources_content = load_sources_file(file_id)
         
         client = genai.Client(api_key=api_key)
         model = "gemini-2.5-pro"
@@ -286,39 +349,29 @@ async def rewrite_news(news: NewsRequest):
         # Instruções do sistema
         SYSTEM_INSTRUCTIONS = f"""
 Você é um jornalista brasileiro, escrevendo para portais digitais. Sua missão é transformar notícias internacionais em matérias originais, detalhadas e atualizadas para o público brasileiro. Sempre use a notícia-base como ponto de partida, mas consulte o arquivo fontes.txt para extrair todas as informações relevantes, complementando fatos, contexto, dados e antecedentes. Não invente informações; na dúvida, não insira.
-
 Seu estilo de escrita deve ser direto, claro e conversacional, sem jargões ou floreios desnecessários. Frases curtas e bem estruturadas, parágrafos segmentados para leitura digital e SEO. Evite repetições, clichês e generalizações.
-
 Evite frases redundantes ou genéricas como:
 - "Destacando como a experiência pode ser um divisor de águas profissional"
 - "Reafirma a força criativa do país no cenário global"
 - "A revelação contextualizou não apenas sua performance na dança, mas também"
-
 O conteúdo deve priorizar clareza, contexto e completude:
-
 - Comece com a informação mais relevante e específica.
 - Contextualize causas, consequências e conexões com outros acontecimentos.
 - Inclua dados, datas, lançamentos e fontes confiáveis.
 - Use citações, títulos de obras e nomes próprios quando pertinentes.
 - Finalize sempre com fatos concretos, nunca com opinião genérica.
-
 Formato da matéria:
-
 <headline>título aqui</headline>
 <subhead>subtítulo aqui</subhead>
 <body>conteúdo aqui</body>
 <instagram_title>título curto e chamativo para Instagram</instagram_title>
 <instagram_description>resumo atrativo com as principais informações, terminando com chamada para ler mais na bio</instagram_description>
-
 INSTRUÇÕES PARA CONTEÚDO INSTAGRAM:
 - Instagram Title:  Direto, chamativo e informativo. Deve funcionar bem como texto sobreposto na imagem do post.
 - Instagram Description: Resumo de 2-3 parágrafos com os principais pontos da notícia, mas sem revelar tudo. Termine SEMPRE com uma chamada como "Leia mais sobre [tema] no link da nossa bio" ou variação similar. Adicione no máximo 5 hashtags no final.
-
 Use <strong> para destaques e <em> para títulos de obras, etc.
 Padrão de títulos: apenas a primeira palavra em maiúscula (exceto nomes próprios ou títulos de obras).
-
 O resultado deve soar como uma reportagem profissional, completa, confiável e envolvente, no estilo da Fox News: informativa, detalhada, mantendo linguagem humana e acessível. Sempre busque complementar a notícia com informações adicionais e contextos do arquivo fontes.txt, evitando opiniões pessoais. Mantenha tom informativo e predominantemente conservador (o jornal tem um público maior da ala da Direita/Republicana, sem juízos de valor ou opiniões pessoais. Não use linguagem woke, de vitimização ou que force interpretações ideológicas.
-
 A data de hoje é {date_string}.
 """
 
@@ -341,19 +394,12 @@ Observação: O Jornal mantém um público majoritário de maioria Republicana a
 
         EXAMPLE_INPUT_2 = """
 Observação: O Jornal mantém um público majoritário de maioria Republicana americana/Direita brasileira, então é um jornal parcialmente conservador. Escreva a notícia ciente disso. News base: The Office spinoff series 'The Paper' has set a September premiere date at Peacock.
-
 The new mockumentary series from Greg Daniels and Michael Koman will debut Sept. 4 on Peacock, the streamer announced Thursday. The first four episodes of 'The Paper' will premiere on Sept. 4, with two new episodes dropping every Thursday through Sept. 25.
-
 'The Paper' follows the documentary crew that immortalized Dunder Mifflin's Scranton branch in 'The Office' as they find a new subject when they discover a historic Midwestern newspaper and the publisher trying to revive it, according to the official logline.
-
 'The Office' fan-favorite Oscar Nuñez returns to the franchise in 'The Paper,' joining series regulars Domhnall Gleeson, Sabrina Impacciatore, Chelsea Frei, Melvin Gregg, Gbemisola Ikumelo, Alex Edelman, Ramona Young and Tim Key.
-
 Guest stars for the show include Eric Rahill, Tracy Letts, Molly Ephraim, Mo Welch, Allan Havey, Duane Shepard Sr., Nate Jackson and Nancy Lenehan.
-
 'The Paper' was created by Daniels, who created 'The Office,' under his banner Deedle-Dee Productions, and Koman, who has written on 'Nathan for You' and 'SNL.' Produced by Universal Television, a division of Universal Studio Group, 'The Paper' is executive produced by Ricky Gervais, Stephen Merchant, Howard Klein, Ben Silverman and Banijay Americas (formerly Reveille).
-
 Daniels serves as a director on the show alongside Ken Kwapis, Yana Gorskaya, Paul Lieberstein, Tazbah Chavez, Jason Woliner, Jennifer Celotta, Matt Sohn, Dave Rogers and Jeff Blitz.
-
 'The Office' launched in 2005 on NBC and ran for nine seasons leading up to the series finale in 2013. The cast of the beloved sitcom included Steve Carell, Rainn Wilson, John Krasinski, Jenna Fischer, Mindy Kaling and B.J. Novak, among others. The article is out of date, more information is needed.
 """
 
@@ -393,11 +439,8 @@ Observação: O Jornal mantém um público majoritário de maioria Republicana a
 
         EXAMPLE_INPUT_4 = """
 Observação: O Jornal mantém um público majoritário de maioria Republicana americana/Direita brasileira, então é um jornal parcialmente conservador. Escreva a notícia ciente disso. News base: Sylvester Stallone, Gloria Gaynor, Kiss Set for Kennedy Center Honors Amid Trump Overhaul
-
 The first honorees for the revamped Kennedy Center Honors have been unveiled by U.S. President Donald Trump, who is also the new chairman of the John F. Kennedy Center for the Performing Arts.
-
 This year’s honorees include Rocky star and filmmaker Sylvester Stallone; disco-era singer Gloria Gaynor; the rock band Kiss; Michael Crawford, the British star of Phantom of the Opera; and country crooner and songwriter George Strait.
-
 The 48th annual Kennedy Center Honors, set to air on the CBS network and stream on Paramount+, will be hosted by the U.S. President. Stallone was earlier named by Trump as one of his ambassadors to Hollywood. “He’s a very special guy. A real talent, never been given credit for the talent,” Trump added about the Hollywood actor during an hourlong press conference Wednesday
 """