Update app.py

app.py

@@ -11,7 +11,6 @@ from wikipedia_searcher import WikipediaSearcher
 
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
-
 class GaiaAgent:
     def __init__(self):
         tools = [
@@ -26,13 +25,6 @@ class GaiaAgent:
             tools=tools
         )
 
-    #def __call__(self, query: str) -> str:
-        #return self.agent.run(query)
-
-    #def __call__(self, query: str) -> str:
-        #result = self.agent.run(query)
-        #return result.get("output", "No output returned")
-
     def __call__(self, question: str) -> str:
         prompt = f"""You are an agent solving the GAIA benchmark and you are required to provide exact answers.
         Rules to follow:
@@ -42,23 +34,19 @@ class GaiaAgent:
         4. For numbers, use the exact number, no other format.
         5. For names, use the exact name as found in sources.
         6. If the question has an associated file, download the file first using the task ID.
-
         Examples of good responses:
         - "42"
         - "Pinco Palla"
         - "Yes"
         - "October 5, 2001"
         - "Buenos Aires"
-
         Never include phrases like "the answer is..." or "Based on my research".
         Only return the exact answer.
-
         QUESTION:
         {question}
         """
         result = self.agent.run(prompt)
-        return result.get("output", "No output returned")
-
+        return result  # ✅ Fixed: removed .get() since result is a string
 
 
 def run_and_submit_all(profile: gr.OAuthProfile | None):
@@ -184,12 +172,12 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
 # Gradio UI
 with gr.Blocks() as demo:
     gr.Markdown("# Basic Agent Evaluation Runner")
-    gr.Markdown("""
-        **Instructions:**
-        1. Clone this space and define your agent and tools.
-        2. Log in to your Hugging Face account using the button below.
-        3. Click 'Run Evaluation & Submit All Answers' to test your agent and submit results.
-    """)
+    gr.Markdown("""\
+**Instructions:**
+1. Clone this space and define your agent and tools.
+2. Log in to your Hugging Face account using the button below.
+3. Click 'Run Evaluation & Submit All Answers' to test your agent and submit results.
+""")
 
     gr.LoginButton()
     run_button = gr.Button("Run Evaluation & Submit All Answers")