Update routers/searchterm.py

routers/searchterm.py

@@ -15,6 +15,8 @@ from fastapi import APIRouter, HTTPException, Body
 from fastapi.responses import FileResponse
 from newspaper import Article
 from threading import Timer
+from google import genai
+from google.genai import types
 
 router = APIRouter()
 
@@ -22,6 +24,10 @@ BRAVE_API_KEY = os.getenv("BRAVE_API_KEY")
 if not BRAVE_API_KEY:
     raise ValueError("BRAVE_API_KEY não está definido!")
 
+GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
+if not GEMINI_API_KEY:
+    raise ValueError("GEMINI_API_KEY não está definido!")
+
 BRAVE_SEARCH_URL = "https://api.search.brave.com/res/v1/web/search"
 BRAVE_HEADERS = {
     "Accept": "application/json",
@@ -111,6 +117,86 @@ def create_temp_file(data: Dict[str, Any]) -> Dict[str, str]:
     }
 
 
+async def generate_gemini_response(input_text: str) -> str:
+    """Gera resposta usando o modelo Gemini"""
+    try:
+        client = genai.Client(api_key=GEMINI_API_KEY)
+        model = "gemini-2.5-flash-lite"
+        
+        contents = [
+            types.Content(
+                role="user",
+                parts=[
+                    types.Part.from_text(text="Retorne um json de exemplo"),
+                ],
+            ),
+            types.Content(
+                role="model",
+                parts=[
+                    types.Part.from_text(text="""Com certeza! Aqui está um exemplo de JSON simples:
+```json
+{
+  \"nome\": \"Maria Silva\",
+  \"idade\": 30,
+  \"profissao\": \"Engenheira de Software\",
+  \"habilidades\": [
+    \"Python\",
+    \"JavaScript\",
+    \"SQL\",
+    \"Docker\"
+  ],
+  \"contato\": {
+    \"email\": \"[email protected]\",
+    \"telefone\": \"+55 11 98765-4321\"
+  },
+  \"ativo\": true,
+  \"projetos\": null
+}
+```
+**Explicação dos elementos:**
+*   **`{}`**: Representa um objeto JSON.
+*   **`\"chave\": valor`**: Um objeto é composto por pares de chave-valor.
+    *   As **chaves** são sempre strings, delimitadas por aspas duplas.
+    *   Os **valores** podem ser de diversos tipos:
+        *   **String:** ` \"Maria Silva\" ` (delimitada por aspas duplas)
+        *   **Número:** ` 30 ` (inteiro ou decimal)
+        *   **Booleano:** ` true ` ou ` false `
+        *   **Array (Lista):** ` [\"Python\", \"JavaScript\", \"SQL\", \"Docker\"] ` (uma lista de valores, delimitada por colchetes `[]`, onde os elementos são separados por vírgulas)
+        *   **Objeto:** ` {\"email\": \"[email protected]\", \"telefone\": \"+55 11 98765-4321\"} ` (outro objeto JSON aninhado)
+        *   **`null`**: Representa a ausência de valor.
+Este é um exemplo bastante comum e demonstra a estrutura básica do JSON."""),
+                ],
+            ),
+            types.Content(
+                role="user",
+                parts=[
+                    types.Part.from_text(text=input_text),
+                ],
+            ),
+        ]
+        
+        generate_content_config = types.GenerateContentConfig(
+            thinking_config=types.ThinkingConfig(
+                thinking_budget=0,
+            ),
+        )
+        
+        # Coletamos toda a resposta em stream
+        full_response = ""
+        for chunk in client.models.generate_content_stream(
+            model=model,
+            contents=contents,
+            config=generate_content_config,
+        ):
+            if chunk.text:
+                full_response += chunk.text
+        
+        return full_response
+        
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Erro ao gerar resposta do Gemini: {str(e)}")
+
+
 async def search_brave_term(client: httpx.AsyncClient, term: str) -> List[Dict[str, str]]:
     params = {"q": term, "count": 10, "safesearch": "off", "summary": "false"}
     
@@ -244,6 +330,51 @@ async def search_terms(payload: Dict[str, List[str]] = Body(...)) -> Dict[str, A
     }
 
 
+@router.post("/inference/terms")
+async def inference_terms(payload: Dict[str, str] = Body(...)) -> Dict[str, Any]:
+    """
+    Endpoint para fazer inferência com o modelo Gemini
+    
+    Body:
+    {
+        "input": "Sua pergunta ou texto aqui"
+    }
+    """
+    input_text = payload.get("input")
+    
+    if not input_text or not isinstance(input_text, str):
+        raise HTTPException(
+            status_code=400, 
+            detail="Campo 'input' é obrigatório e deve ser uma string."
+        )
+    
+    if len(input_text.strip()) == 0:
+        raise HTTPException(
+            status_code=400, 
+            detail="Campo 'input' não pode estar vazio."
+        )
+    
+    try:
+        # Gera resposta usando o Gemini
+        response = await generate_gemini_response(input_text)
+        
+        return {
+            "input": input_text,
+            "response": response,
+            "model": "gemini-2.5-flash-lite",
+            "timestamp": time.time()
+        }
+        
+    except HTTPException:
+        # Re-raise HTTPExceptions para manter o status code correto
+        raise
+    except Exception as e:
+        raise HTTPException(
+            status_code=500, 
+            detail=f"Erro interno do servidor: {str(e)}"
+        )
+
+
 @router.get("/download-temp/{file_id}")
 async def download_temp_file(file_id: str):
     """Endpoint para download do arquivo temporário"""