Update app.py

app.py

@@ -1,14 +1,12 @@
 import os
+import re
 import gradio as gr
 import requests
-import inspect
 import pandas as pd
-import re
 
-# 🔹 NOVO: imports do smolagents
+from huggingface_hub import InferenceClient
 from smolagents import CodeAgent, DuckDuckGoSearchTool, InferenceClientModel
 
-
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
@@ -18,20 +16,18 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # =========================================================
 def clean_answer(text: str) -> str:
     """
-    Cleans the answer returned by the model:
-    - removes newlines
-    - removes phrases like 'final answer', 'answer:'
-    - removes quotes
-    - trims spaces
-    But DOES NOT delete useful content.
+    Limpa a resposta retornada pelo modelo:
+    - remove quebras de linha
+    - remove 'final answer', 'answer:', etc
+    - remove aspas externas
+    - normaliza espaços
+    NÃO apaga o conteúdo útil.
     """
-
     if not text:
         return ""
 
     text = str(text).strip()
 
-    # Remover frases proibidas
     patterns_to_remove = [
         r"(?i)final answer[:\- ]*",
         r"(?i)answer[:\- ]*",
@@ -41,25 +37,20 @@ def clean_answer(text: str) -> str:
     for p in patterns_to_remove:
         text = re.sub(p, "", text).strip()
 
-    # Remover quebras de linha
     text = text.replace("\n", " ").strip()
 
-    # Remover aspas externas (se existirem)
-    if len(text) >= 2 and text.startswith('"') and text.endswith('"'):
-        text = text[1:-1].strip()
-    if len(text) >= 2 and text.startswith("'") and text.endswith("'"):
+    # aspas externas
+    if len(text) >= 2 and text[0] == text[-1] and text[0] in ['"', "'"]:
         text = text[1:-1].strip()
 
-    # Remover espaços múltiplos
     text = re.sub(r"\s+", " ", text)
 
     return text.strip()
 
 
 # =========================================================
-#  Basic Agent Definition – AGORA usando smolagents
+#  Prompt base para o agente
 # =========================================================
-
 SYSTEM_PROMPT = (
     "You are an AI agent solving GAIA-style questions.\n"
     "You have access to a web search tool (DuckDuckGoSearchTool).\n"
@@ -74,6 +65,9 @@ SYSTEM_PROMPT = (
 )
 
 
+# =========================================================
+#  Basic Agent Definition – usando smolagents
+# =========================================================
 class BasicAgent:
     """
     Agente smolagents com DuckDuckGoSearchTool,
@@ -81,60 +75,51 @@ class BasicAgent:
     """
 
     def __init__(self):
-        print("Initializing improved GAIA agent with search...")
+        print("Initializing GAIA agent with web search...")
 
-        self.search_tool = DuckDuckGoSearchTool()
+        # ---- Token HF vindo do secret HF_TOKEN ----
+        hf_token = os.getenv("HF_TOKEN")
+        if not hf_token:
+            print("⚠️  HF_TOKEN not found in environment! InferenceClient may fail.")
 
-        # Prompt especializado
-        self.model = InferenceClientModel(
-            system_prompt=(
-                "You are a GAIA evaluation agent that must answer EXACTLY the final answer.\n"
-                "You MUST use the search tool to retrieve verified information.\n"
-                "When you use search, read the results and extract ONLY the exact required answer.\n"
-                "RULES:\n"
-                " - Output MUST be a SINGLE short string.\n"
-                " - NO explanations.\n"
-                " - NO reasoning.\n"
-                " - NO multi-sentence output.\n"
-                " - NO citations or URLs.\n"
-                " - NO extra words.\n"
-                "Examples of valid outputs:\n"
-                "   '2002'\n"
-                "   '7'\n"
-                "   'egalitarian'\n"
-                "   'Mercedes Sosa'\n"
-                "INVALID OUTPUTS:\n"
-                "   'The final answer is 7.'\n"
-                "   'According to Wikipedia, the answer is 2002.'\n"
-                "   'After checking, I think it is 3.'\n"
-                "Your response MUST be only the exact final answer.\n"
-            )
-        )
+        # Cliente de inferência com token
+        client = InferenceClient(token=hf_token) if hf_token else InferenceClient()
+
+        # Modelo remoto via Inference API
+        self.model = InferenceClientModel(client=client)
 
-        # CodeAgent com raciocínio guiado
+        # Ferramenta de busca
+        self.search_tool = DuckDuckGoSearchTool()
+
+        # CodeAgent com ferramenta de busca
         self.agent = CodeAgent(
             model=self.model,
             tools=[self.search_tool],
-            max_steps=8,    # permite buscar e refinar
+            max_steps=8,  # permite alguns passos (buscar + refinar)
         )
 
     def __call__(self, question: str) -> str:
-        print(f"Processing: {question[:80]}...")
+        print(f"\nProcessing question: {question[:80]}...")
         try:
-            raw = self.agent.run(
-                f"Answer this question using search:\n{question}\n"
-                "Return ONLY the exact final answer."
+            # Prompt completo passado para o agente
+            full_prompt = (
+                f"{SYSTEM_PROMPT}\n\n"
+                f"Question: {question}\n\n"
+                "Follow the rules above and return ONLY the exact final answer."
             )
+
+            raw = self.agent.run(full_prompt)
             final = clean_answer(raw)
+            print(f"Raw answer: {raw}")
             print(f"Final cleaned answer: {final}")
             return final
         except Exception as e:
-            print(f"Error: {e}")
+            print(f"Error inside BasicAgent.__call__: {e}")
             return ""
 
 
 # =========================================================
-#  Runner + submit (mantido do template, só usando BasicAgent novo)
+#  Runner + submit (mantido do template, usando BasicAgent novo)
 # =========================================================
 def run_and_submit_all(profile: gr.OAuthProfile | None):
     """
@@ -155,22 +140,21 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     questions_url = f"{api_url}/questions"
     submit_url = f"{api_url}/submit"
 
-    # 1. Instantiate Agent (agora nosso agente smolagents)
+    # 1. Instantiate Agent
     try:
         agent = BasicAgent()
     except Exception as e:
         print(f"Error instantiating agent: {e}")
         return f"Error initializing agent: {e}", None
 
-    # In the case of an app running as a hugging Face space, this link points toward your codebase
-    # (useful for others so please keep it public)
+    # Link para o código do agente (Space público)
     agent_code = f"https://huggingface.co/spaces/{space_id}/tree/main"
     print(f"Agent code URL: {agent_code}")
 
     # 2. Fetch Questions
     print(f"Fetching questions from: {questions_url}")
     try:
-        response = requests.get(questions_url, timeout=15)
+        response = requests.get(questions_url, timeout=60)  # timeout maior
         response.raise_for_status()
         questions_data = response.json()
         if not questions_data:
@@ -317,9 +301,8 @@ with gr.Blocks() as demo:
 
 if __name__ == "__main__":
     print("\n" + "-" * 30 + " App Starting " + "-" * 30)
-    # Check for SPACE_HOST and SPACE_ID at startup for information
     space_host_startup = os.getenv("SPACE_HOST")
-    space_id_startup = os.getenv("SPACE_ID")  # Get SPACE_ID at startup
+    space_id_startup = os.getenv("SPACE_ID")
 
     if space_host_startup:
         print(f"✅ SPACE_HOST found: {space_host_startup}")
@@ -327,7 +310,7 @@ if __name__ == "__main__":
     else:
         print("ℹ️  SPACE_HOST environment variable not found (running locally?).")
 
-    if space_id_startup:  # Print repo URLs if SPACE_ID is found
+    if space_id_startup:
         print(f"✅ SPACE_ID found: {space_id_startup}")
         print(f"   Repo URL: https://huggingface.co/spaces/{space_id_startup}")
         print(
@@ -337,6 +320,5 @@ if __name__ == "__main__":
         print("ℹ️  SPACE_ID environment variable not found (running locally?). Repo URL cannot be determined.")
 
     print("-" * (60 + len(" App Starting ")) + "\n")
-
     print("Launching Gradio Interface for Basic Agent Evaluation...")
-    demo.launch(debug=True, share=False)
+    demo.launch(debug=True, share=False)