Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 import json
 import os
 import requests
-
+import traceback
 from agent import solve_task, load_tasks
 from mistral_hf_wrapper import MistralInference
 
@@ -22,47 +22,43 @@ def run_and_submit_all():
 
     answers = []
     for i, task in enumerate(tasks):
-        print(f"[INFO] Solving task {i+1}/{len(tasks)}: {task.get('question_id', 'N/A')}")
-
+        print(f"[INFO] Solving task {i+1}/{len(tasks)}: {task.get('question_id', 'UNKNOWN')}")
         try:
             result = solve_task(task, model)
-
-            if not result.get("submitted_answer"):  # Check empty responses
-                print("[WARN] Empty model response detected")
+            print(f"[INFO]Answer: {result['submitted_answer'][:100]}...")
+            if not result.get("submitted_answer"):
                 result["submitted_answer"] = "ERROR: Empty model response"
-            else:
-                print(f"Answer: {result['submitted_answer'][:100]}...")
-
             answers.append(result)
-
         except Exception as e:
-            print(f"[ERROR] Task failed: {e}")
+            tb = traceback.format_exc()
+            print(f"[ERROR] Task failed: {e}\n{tb}")
             answers.append({
                 "question_id": task.get("question_id", "UNKNOWN"),
                 "submitted_answer": f"ERROR: {str(e)}"
             })
 
     if not answers:
-        return "No answers generated. Check model response."
-
-    print("[INFO] Submitting answers to GAIA benchmark 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
-        },
-    )
+        return "No answers generated. Check model response or network."
 
-    if res.ok:
-        print(" Submission successful.")
-        return json.dumps(res.json(), indent=2)
+    try:
+        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
+            },
+            timeout=60  # increase timeout here too
+        )
 
-    print(f"[ERROR] Submission failed: {res.status_code} - {res.text}")
-    return f"Error submitting: {res.status_code} - {res.text}"
+        if res.ok:
+            return json.dumps(res.json(), indent=2)
+        else:
+            return f"Error submitting: {res.status_code} - {res.text}"
 
+    except Exception as e:
+        return f"[SUBMISSION ERROR] {str(e)}"
 # Gradio interface
 gr.Interface(
     fn=run_and_submit_all,