Update app.py

app.py

@@ -10,30 +10,15 @@ from Gradio_UI import GradioUI
 # --- FERRAMENTAS PERSONALIZADAS ---
 
 @tool
-def get_football_team_info(team_name: str, info_type: str) -> str:
-    """A tool that fetches football team information (results or upcoming matches) from ZeroZero.
+@tool
+def get_football_info(query: str) -> str:
+    """Pesquisa informações atualizadas sobre futebol (jogos, resultados, classificações).
     Args:
-        team_name: The name of the football team (e.g., 'Benfica', 'Sporting').
-        info_type: Type of information to fetch: 'resultados' (past matches) or 'proximos' (future matches).
+        query: O que queres saber (ex: 'próximo jogo benfica', 'resultado sporting ontem').
     """
-    try:
-        search_query = team_name.replace(" ", "+")
-        base_url = "https://www.zerozero.pt"
-        search_url = f"{base_url}/search.php?input_search={search_query}"
-        
-        headers = {
-            "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/119.0.0.0 Safari/537.36"
-        }
-        
-        response = requests.get(search_url, headers=headers, timeout=10)
-        
-        if response.status_code == 200:
-            return f"Encontrei informações para '{team_name}'. Podes consultar os {info_type} aqui: {response.url}"
-        else:
-            return f"O ZeroZero não permitiu o acesso direto (Status: {response.status_code}). Tenta usar a ferramenta de busca geral."
-            
-    except Exception as e:
-        return f"Erro ao procurar a equipa: {str(e)}"
+    # Esta tool agora serve apenas para dar contexto, 
+    # o agente usará a search_tool para encontrar os dados reais.
+    return f"Vou pesquisar na web por: {query}. Por favor, analisa os resultados dos primeiros sites (Record, A Bola, ZeroZero ou Google)."
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -70,16 +55,13 @@ agent = CodeAgent(
     model=model,
     tools=[
         final_answer, 
-        get_football_team_info, 
-        get_current_time_in_timezone, 
-        search_tool, # Agora já está definido!
-        image_generation_tool
+        search_tool,      # A ferramenta MAIS importante aqui
+        get_football_info, 
+        get_current_time_in_timezone
     ], 
-    max_steps=6,
+    max_steps=10,         # Mais fôlego para o agente
     verbosity_level=1,
-    name="FootballTimeAgent",
-    description="Um agente que sabe horas e resultados de futebol.",
-    prompt_templates=prompt_templates
+    # prompt_templates=prompt_templates  <-- Comenta esta linha para evitar o KeyError
 )
 
 if __name__ == "__main__":