Update app.py

app.py

@@ -4,33 +4,34 @@ import requests
 import pandas as pd
 from smolagents import CodeAgent, DuckDuckGoSearchTool, LiteLLMModel
 
+# --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 class BasicAgent:
     def __init__(self):
-
+        # Using LiteLLM to connect to Qwen
         self.model = LiteLLMModel(model_id="huggingface/Qwen/Qwen2.5-72B-Instruct")
         
+        # Initialize search tool
+        search_tool = DuckDuckGoSearchTool()
+        
         self.agent = CodeAgent(
-            tools=[DuckDuckGoSearchTool()], 
+            tools=[search_tool], 
             model=self.model,
             add_base_tools=True 
         )
         print("Advanced smolagent initialized with LiteLLM.")
 
     def __call__(self, question: str) -> str:
-        # Prompting for Exact Match scoring
         clean_prompt = (
             f"Question: {question}\n\n"
             "Instructions: Solve the question above. Provide ONLY the final answer "
             "value without any explanation, units, or extra text. "
-            "Do not include 'FINAL ANSWER' in your response."
+            "Do not include the phrase 'FINAL ANSWER' in your response."
         )
         
         try:
-            # The agent will reason and use tools if necessary
             result = self.agent.run(clean_prompt)
-            # Ensure we return a clean string
             return str(result).strip()
         except Exception as e:
             print(f"Error during agent execution: {e}")
@@ -46,14 +47,15 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     api_url = DEFAULT_API_URL
     questions_url = f"{api_url}/questions"
     submit_url = f"{api_url}/submit"
+    
+    # Define agent_code link correctly
+    agent_code = f"https://huggingface.co/spaces/{space_id}/tree/main" if space_id else "https://huggingface.co/spaces"
 
     # 1. Instantiate Agent
     try:
         agent = BasicAgent()
     except Exception as e:
         return f"Error initializing agent: {e}", None
-    
-    agent_code = f"https://huggingface.co/spaces/{space_id}/tree/main"
 
     # 2. Fetch Questions
     try:
@@ -67,6 +69,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     results_log = []
     answers_payload = []
     
+    # LIMITING TO FIRST 5 FOR TESTING - Remove [:5] to do all
     for item in questions_data:
         task_id = item.get("task_id")
         question_text = item.get("question")
@@ -74,9 +77,6 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
         if not task_id or question_text is None:
             continue
 
-        # Optional: Handle file downloads for specific tasks
-        # You could add logic here to download from f"{api_url}/files/{task_id}"
-
         try:
             submitted_answer = agent(question_text)
             answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})
@@ -99,8 +99,8 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
         final_status = (
             f"Submission Successful!\n"
             f"User: {result_data.get('username')}\n"
-            f"Score: {result_data.get('score')}% "
-            f"({result_data.get('correct_count')}/{result_data.get('total_attempted')} correct)"
+            f"Score: {result_data.get('score', 0)}% "
+            f"({result_data.get('correct_count', 0)}/{result_data.get('total_attempted', 0)} correct)"
         )
         return final_status, pd.DataFrame(results_log)
     except Exception as e:
@@ -109,11 +109,8 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
 # --- Gradio Interface ---
 with gr.Blocks() as demo:
     gr.Markdown("# GAIA Agent Evaluation Runner")
-    gr.Markdown("This agent uses `smolagents` with a Python interpreter and Web Search to solve tasks.")
-
     gr.LoginButton()
     run_button = gr.Button("Run Evaluation & Submit All Answers", variant="primary")
-
     status_output = gr.Textbox(label="Status", lines=4)
     results_table = gr.DataFrame(label="Detailed Results", wrap=True)