Update app.py

app.py

@@ -2,10 +2,9 @@ import gradio as gr
 import json
 import os
 import requests
-from .metadata import load_metadata
 
-from agent import solve_task, load_tasks
-from mistral_hf_wrapper import MistralInference  # Correct class import
+from .agent import solve_task, load_tasks
+from .mistral_hf_wrapper import MistralInference
 
 # Load from environment variables
 API_URL = os.getenv("HF_MISTRAL_ENDPOINT")
@@ -13,6 +12,7 @@ API_TOKEN = os.getenv("HF_TOKEN")
 USERNAME = os.getenv("HF_USERNAME")
 CODE_LINK = os.getenv("HF_CODE_LINK")
 
+# Launches agent to solve all tasks and submits to GAIA
 def run_and_submit_all():
     model = MistralInference(api_url=API_URL, api_token=API_TOKEN)
     tasks = load_tasks()
@@ -24,11 +24,11 @@ def run_and_submit_all():
             answers.append(result)
         except Exception as e:
             answers.append({
-                "task_id": task["question_id"],
+                "task_id": task.get("question_id", "unknown"),
                 "submitted_answer": f"ERROR: {str(e)}"
             })
 
-    # Submit answers to GAIA scoring API
+    # Submit results to GAIA benchmark API
     res = requests.post(
         "https://agents-course-unit4-scoring.hf.space/submit",
         headers={"Content-Type": "application/json"},
@@ -44,7 +44,7 @@ def run_and_submit_all():
     else:
         return f"Error submitting: {res.status_code} - {res.text}"
 
-# Launch Gradio interface
+# Gradio interface
 gr.Interface(
     fn=run_and_submit_all,
     inputs=[],