Update app.py

app.py

@@ -68,43 +68,29 @@ SYSTEM_PROMPT = (
 # =========================================================
 #  Basic Agent Definition – usando smolagents
 # =========================================================
-class BasicAgent:
-    """
-    GAIA Agent powered by Mixtral + DuckDuckGo Search
-    """
 
+class BasicAgent:
     def __init__(self):
-        print("Initializing improved GAIA agent...")
+        print("Initializing GAIA Agent...")
 
         HF_TOKEN = os.getenv("HF_TOKEN")
+        print("HF_TOKEN exists?", HF_TOKEN is not None)
+
         if not HF_TOKEN:
-            print("❌ ERROR: HF_TOKEN secret not found!")
-            raise ValueError("HF_TOKEN missing")
+            raise ValueError("❌ HF_TOKEN not found — create a Secret in your Space!")
 
-        # search tool
         self.search_tool = DuckDuckGoSearchTool()
 
-        # LLM client (Mixtral)
-        client = InferenceClient(
+        self.client = InferenceClient(
             model="mistralai/Mixtral-8x7B-Instruct-v0.1",
             token=HF_TOKEN,
         )
 
-        # Model wrapper
         self.model = InferenceClientModel(
-            client=client,
-            system_prompt="""
-You are a GAIA evaluation agent.
-
-Rules:
-- Use web search when needed.
-- Output ONLY the final answer.
-- No explanations, no reasoning, no URLs, no extra words.
-- EXACT MATCH evaluation.
-""",
+            client=self.client,
+            system_prompt="Answer ONLY with the final answer. No steps, no explanation."
         )
 
-        # The real agent
         self.agent = CodeAgent(
             model=self.model,
             tools=[self.search_tool],
@@ -112,13 +98,22 @@ Rules:
         )
 
     def __call__(self, question: str) -> str:
+        print(f"\n=== NEW QUESTION ===\n{question}\n")
+
         try:
-            result = self.agent.run(
-                f"Answer this GAIA question. Return ONLY the final answer:\n{question}"
+            raw = self.agent.run(
+                f"Answer this question exactly. Only answer:\n{question}"
             )
-            return clean_answer(result)
+            print("RAW MODEL RESPONSE:", raw)
+
+            if not raw or raw.strip() == "":
+                print("⚠️ MODEL RETURNED EMPTY STRING ⚠️")
+                return ""
+
+            return clean_answer(raw)
+
         except Exception as e:
-            print(f"Agent error: {e}")
+            print("ERROR:", e)
             return ""