Update app.py

app.py

@@ -1,18 +1,18 @@
-# app.py – przywrócony layout benchmarku z poprawionym wywołaniem agenta
-
 import os
 import requests
 import pandas as pd
 import gradio as gr
-import asyncio
 
-from agent import answer_question
+from agent import GaiaAgent
 
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 class GAIALlamaAgent:
-    def __call__(self, question: str) -> str:
-        return asyncio.run(answer_question(question))
+    def __init__(self):
+        self.agent = GaiaAgent()
+    def __call__(self, question: str, task_id: str) -> str:
+        # Pass both question and task_id to the agent
+        return self.agent(question, task_id)
 
 def run_and_submit_all(profile: gr.OAuthProfile | None):
     space_id = os.getenv("SPACE_ID")
@@ -41,7 +41,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
         if not task_id or question_text is None:
             continue
         try:
-            submitted_answer = agent(question_text)
+            submitted_answer = agent(question_text, task_id)
         except Exception as e:
             submitted_answer = f"[ERROR] {e}"
         answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})