Update app.py

app.py

@@ -4,17 +4,17 @@ import gradio as gr
 import requests
 import pandas as pd
 
-#from smolagents.agent import CodeAgent
+from smolagents import Tool, CodeAgent, HfApiModel
 #from smolagents.models import HfApiModel
-from smolagents import Tool
-from smolagents import  CodeAgent, HfApiModel
 
 from audio_transcriber import AudioTranscriptionTool
 from image_analyzer import ImageAnalysisTool
 from wikipedia_searcher import WikipediaSearcher
 
+# GAIA scoring endpoint
+DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
-# System prompt
+# Static system prompt for GAIA
 SYSTEM_PROMPT = """You are an agent solving the GAIA benchmark and you are required to provide exact answers.
 Rules to follow:
 1. Return only the exact requested answer: no explanation and no reasoning.
@@ -31,11 +31,9 @@ Examples of good responses:
 Never include phrases like "the answer is..." or "Based on my research".
 Only return the exact answer."""
 
-
-# Tool definitions
+# Define agent tools
 audio_tool = AudioTranscriptionTool()
 image_tool = ImageAnalysisTool()
-
 wiki_tool = Tool.from_function(
     name="wikipedia_search",
     description="Search for facts using Wikipedia.",
@@ -46,20 +44,16 @@ wiki_tool = Tool.from_function(
 
 tools = [audio_tool, image_tool, wiki_tool]
 
-
-# Agent factory
-def MyAgent():
-    return CodeAgent(
-        tools=tools,
-        system_prompt=SYSTEM_PROMPT,
-        model=HfApiModel(
-            api_url="https://api-inference.huggingface.com/models/HuggingFaceH4/zephyr-7b-beta",
+# Define the custom agent using Mixtral on Hugging Face
+class MyAgent(CodeAgent):
+    def __init__(self):
+        model = HfApiModel(
+            api_url="https://api-inference.huggingface.com/models/mistralai/Mixtral-8x7B-Instruct-v0.1",
             api_key=os.getenv("HF_API_TOKEN")
         )
-    )
-
+        super().__init__(model=model, tools=tools, system_prompt=SYSTEM_PROMPT)
 
-# Main run and submission logic
+# Evaluation + Submission function
 def run_and_submit_all(profile: gr.OAuthProfile | None):
     space_id = os.getenv("SPACE_ID")
 
@@ -70,7 +64,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
         print("User not logged in.")
         return "Please Login to Hugging Face with the button.", None
 
-    api_url = os.getenv("GAIA_API_URL", "https://gaia-benchmark.com/api")
+    api_url = DEFAULT_API_URL
     questions_url = f"{api_url}/questions"
     submit_url = f"{api_url}/submit"
 
@@ -97,7 +91,6 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     results_log = []
     answers_payload = []
     print(f"Running agent on {len(questions_data)} questions...")
-
     for item in questions_data:
         task_id = item.get("task_id")
         if not task_id:
@@ -152,16 +145,14 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     except Exception as e:
         return f"An unexpected error occurred during submission: {e}", pd.DataFrame(results_log)
 
-
-# Gradio UI setup
+# Gradio UI
 with gr.Blocks() as demo:
     gr.Markdown("# Basic Agent Evaluation Runner")
     gr.Markdown("""
         **Instructions:**
-        1. Clone this space, modify code to define your agent's logic, tools, and packages.
+        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 fetch questions, run your agent, submit answers, and see your score.
-        **Note:** Submitting can take some time.
+        3. Click 'Run Evaluation & Submit All Answers' to test your agent and submit results.
     """)
 
     gr.LoginButton()
@@ -172,9 +163,8 @@ with gr.Blocks() as demo:
 
     run_button.click(fn=run_and_submit_all, outputs=[status_output, results_table])
 
-# App startup logs
 if __name__ == "__main__":
-    print("\n" + "-" * 30 + " App Starting " + "-" * 30)
+    print("\n" + "-"*30 + " App Starting " + "-"*30)
     space_host = os.getenv("SPACE_HOST")
     space_id = os.getenv("SPACE_ID")
 
@@ -182,17 +172,16 @@ if __name__ == "__main__":
         print(f"✅ SPACE_HOST found: {space_host}")
         print(f"   Runtime URL should be: https://{space_host}.hf.space")
     else:
-        print("ℹ️  SPACE_HOST environment variable not found (running locally?).")
+        print("ℹ️  SPACE_HOST not found.")
 
     if space_id:
         print(f"✅ SPACE_ID found: {space_id}")
         print(f"   Repo URL: https://huggingface.co/spaces/{space_id}")
-        print(f"   Repo Tree URL: https://huggingface.co/spaces/{space_id}/tree/main")
     else:
-        print("ℹ️  SPACE_ID environment variable not found (running locally?).")
+        print("ℹ️  SPACE_ID not found.")
 
-    print("-" * (60 + len(" App Starting ")) + "\n")
-    print("Launching Gradio Interface for Basic Agent Evaluation...")
+    print("-"*(60 + len(" App Starting ")) + "\n")
     demo.launch(debug=True, share=False)
 
 
+