Update app.py

app.py

@@ -3,9 +3,10 @@ import json
 import os
 import requests
 
-from mistral_hf_wrapper import query_mistral
 from agent import solve_task, load_tasks
+from mistral_hf_wrapper import MistralInference  # Correct class import
 
+# Load from environment variables
 API_URL = os.getenv("HF_MISTRAL_ENDPOINT")
 API_TOKEN = os.getenv("HF_TOKEN")
 USERNAME = os.getenv("HF_USERNAME")
@@ -18,19 +19,34 @@ def run_and_submit_all():
 
     for task in tasks:
         try:
-            answer = solve_task(task, model)
-            answers.append({"task_id": task["task_id"], "submitted_answer": answer.strip()})
+            result = solve_task(task, model)
+            answers.append(result)
         except Exception as e:
-            answers.append({"task_id": task["task_id"], "submitted_answer": f"ERROR: {str(e)}"})
+            answers.append({
+                "task_id": task["question_id"],
+                "submitted_answer": f"ERROR: {str(e)}"
+            })
 
+    # Submit answers to GAIA scoring API
     res = requests.post(
         "https://agents-course-unit4-scoring.hf.space/submit",
         headers={"Content-Type": "application/json"},
-        json={"username": USERNAME, "agent_code": CODE_LINK, "answers": answers},
+        json={
+            "username": USERNAME,
+            "agent_code": CODE_LINK,
+            "answers": answers
+        },
     )
 
     if res.ok:
         return json.dumps(res.json(), indent=2)
-    return f"Error submitting: {res.status_code} - {res.text}"
+    else:
+        return f"Error submitting: {res.status_code} - {res.text}"
 
-gr.Interface(run_and_submit_all, inputs=[], outputs="textbox", title="GAIA Benchmark Agent Submission").launch()
+# Launch Gradio interface
+gr.Interface(
+    fn=run_and_submit_all,
+    inputs=[],
+    outputs="textbox",
+    title="GAIA Benchmark Agent Submission"
+).launch()