agent

agent.py

@@ -0,0 +1,43 @@
+import os
+from dotenv import load_dotenv
+
+from pydantic_ai import Agent, RunContext
+# from pydantic_ai.messages import (ModelMessage, ModelRequest, ModelResponse, UserPromptPart, 
+#                                   SystemPromptPart, TextPart, ModelMessagesTypeAdapter,
+#                                   ToolReturnPart)
+from pydantic_ai.usage import UsageLimits
+from pydantic_core import to_jsonable_python
+from pydantic_ai.models.groq import GroqModel    
+
+from dataclasses import dataclass
+
+from dotenv import load_dotenv
+load_dotenv()
+
+# Defina um prompt de sistema padrão
+DEFAULT_SYSTEM_PROMPT = f"""
+Você é um agente assistente em um programa de detecção de conteúdos falsos (fake news).
+Intereja com o usuário sugerindo as ferramentas que você tem disponível. Se o usuário perguntar do que se trata,
+responda 'Sou uma IA disponível para te ajudar a detectar se um conteúdo que circula na internet é verdadeiro ou não,
+sou capaz de pesquisar na web se o conteúdo é verídico ou não.' Use as ferramenta sempre que possível. 
+A data de hoje é .
+Anexe sempre a fonte da pesquisa na resposta (G1, BBC, New York Times, etc), se houver fonte na pesquisa utilizada.
+Não invente fontes, não invente dados. Se não houver nada relacionado à pergunta do usuário nas pesquisas, diga que nada foi encontrado.
+Responda sempre em português do Brasil.
+Limite-se a 600 caracteres.
+"""
+
+api_key = os.getenv("GROQ_API_KEY")
+model = GroqModel(
+    model_name="openai/gpt-oss-120b"
+)
+
+
+def start_convo(user_input, messages_history):
+    search_agent = Agent(
+        model,
+        system_prompt=DEFAULT_SYSTEM_PROMPT
+    )
+    
+        
+    return search_agent

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 from ocr_script import ocr_tesseract_only
 import uuid
-
+from agent import start_convo
 # def ocr_tesseract_only(img):  # img pode ser PIL/np/str conforme 'type'
 #     # chamar seu OCR aqui e retornar texto
 #     return "texto extraído"
@@ -38,4 +38,4 @@ with gr.Blocks() as demo:
                 save_history=True,
             )
             chat.render()
-#demo.launch()
+demo.launch()

requirements.txt

@@ -3,4 +3,5 @@ pillow>=11.3.0
 pytesseract>=0.3.13
 regex>=2025.9.18
 matplotlib>=3.10.7
-pandas>=2.3.3
+pandas>=2.3.3
+pydantic-ai>=1.0.10