Update app.py

app.py

@@ -3,40 +3,40 @@ import os
 import requests
 import pandas as pd
 import gradio as gr
-from smolagents import ToolCallingAgent
-from smolagents.models import OpenAIServerModel
 
+from smolagents import ToolCallingAgent, OpenAIServerModel
 from audio_transcriber import AudioTranscriptionTool
 from image_analyzer import ImageAnalysisTool
 from wikipedia_searcher import WikipediaSearcher
 
-DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
+DEFAULT_API_URL = "https://gaia-benchmark.com/api"
 
 class GaiaAgent:
     def __init__(self):
-        api_key = os.getenv("OPENAI_API_KEY")
-        if not api_key:
-            raise EnvironmentError("OPENAI_API_KEY not found in environment variables.")
-
-        model = OpenAIServerModel(
-            model_id="gpt-3.5-turbo",
-            api_key=api_key
-        )
         tools = [
             AudioTranscriptionTool(),
             ImageAnalysisTool(),
             WikipediaSearcher()
         ]
-        self.agent = ToolCallingAgent(model=model, tools=tools)
 
-    def __call__(self, prompt: str) -> str:
-        return self.agent.run([{"role": "user", "content": prompt}])
+        model_id = os.getenv("OPENAI_MODEL_ID", "gpt-3.5-turbo")
+        self.agent = ToolCallingAgent(
+            model=OpenAIServerModel(model_id=model_id),
+            tools=tools
+        )
+
+    def __call__(self, query: str) -> str:
+        result = self.agent.run(query)
+        return result.get("output", "No output returned")
 
 def run_and_submit_all(profile: gr.OAuthProfile | None):
     space_id = os.getenv("SPACE_ID")
 
     if profile:
         username = profile.username
+        if isinstance(username, list):
+            username = username[0]
+        username = username.strip()
         print(f"User logged in: {username}")
     else:
         print("User not logged in.")
@@ -118,7 +118,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
         return "Agent did not produce any answers to submit.", pd.DataFrame(results_log)
 
     submission_data = {
-        "username": username.strip(),
+        "username": username,
         "agent_code": agent_code,
         "answers": answers_payload
     }
@@ -149,6 +149,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
         return f"An unexpected error occurred during submission: {e}", pd.DataFrame(results_log)
 
 
+# Gradio UI
 with gr.Blocks() as demo:
     gr.Markdown("# Basic Agent Evaluation Runner")
     gr.Markdown("""
@@ -172,16 +173,16 @@ if __name__ == "__main__":
     space_id = os.getenv("SPACE_ID")
 
     if space_host:
-        print(f"\u2705 SPACE_HOST found: {space_host}")
+        print(f"✅ SPACE_HOST found: {space_host}")
         print(f"   Runtime URL should be: https://{space_host}.hf.space")
     else:
-        print("\u2139\ufe0f  SPACE_HOST not found.")
+        print("ℹ️  SPACE_HOST not found.")
 
     if space_id:
-        print(f"\u2705 SPACE_ID found: {space_id}")
+        print(f"✅ SPACE_ID found: {space_id}")
         print(f"   Repo URL: https://huggingface.co/spaces/{space_id}")
     else:
-        print("\u2139\ufe0f  SPACE_ID not found.")
+        print("ℹ️  SPACE_ID not found.")
 
     print("-"*(60 + len(" App Starting ")) + "\n")
     demo.launch(debug=True, share=False)