Update app.py

app.py

@@ -19,36 +19,65 @@ def answer_question(question):
 
 def submit_to_gaia(username, space_link):
     GAIA_API_URL = "https://huggingface.co/api/gaia"
-    
+
     try:
-        # 1. Fetch tasks from GAIA
+        # Step 1: Fetch questions
         response = requests.get(f"{GAIA_API_URL}/questions")
-        questions = response.json()  # Now properly parsed into a list of dicts
+        if response.status_code != 200:
+            return f"❌ Failed to fetch tasks. HTTP {response.status_code}: {response.text}"
+        
+        questions = response.json()
+        if not isinstance(questions, list):
+            return f"❌ Unexpected response format: {questions}"
 
-        # 2. Generate answers using your agent
         answers = []
-        for q in questions:
-            task_id = q["task_id"]
-            question_text = q["question"]
-            agent_answer = agent.search_advice(question_text)  # Use your semantic search
-            answers.append({
-                "task_id": task_id,
-                "submitted_answer": agent_answer.strip()
-            })
-
-        # 3. Submit answers
+        for idx, q in enumerate(questions):
+            try:
+                task_id = q["task_id"]
+                question_text = q["question"]
+                print(f"➡️ [{idx+1}/{len(questions)}] Answering: {question_text}")
+
+                agent_answer = agent.search_advice(question_text)
+                cleaned_answer = (agent_answer or "").strip()
+
+                if not cleaned_answer:
+                    print(f"⚠️ Warning: Empty answer for task {task_id}")
+
+                answers.append({
+                    "task_id": task_id,
+                    "submitted_answer": cleaned_answer
+                })
+
+            except Exception as inner_e:
+                print(f"❌ Error on question {idx+1}: {inner_e}")
+                continue
+
+        if not answers:
+            return "❌ No answers were collected. All tasks failed or returned empty."
+
         payload = {
-            "username": username,
-            "agent_code": space_link,
+            "username": username.strip(),
+            "agent_code": space_link.strip(),
             "answers": answers
         }
+
+        print("🚀 Submitting answers to GAIA...")
         submit_response = requests.post(f"{GAIA_API_URL}/submit", json=payload)
-        result = submit_response.json()
 
-        return f"✅ Submission complete!\n\nScore: {result.get('score', 'N/A')}\nStatus: {result.get('status', 'N/A')}\nMessage: {result.get('message', 'No message')}"
-    
+        try:
+            result = submit_response.json()
+        except Exception:
+            return f"❌ Failed to parse submission response:\n{submit_response.text}"
+
+        return f"""✅ Submission complete!
+Score: {result.get('score', 'N/A')}
+Status: {result.get('status', 'N/A')}
+Message: {result.get('message', 'No message')}
+"""
+
     except Exception as e:
-        return f"❌ Submission failed due to: {str(e)}"
+        return f"❌ Submission failed: {str(e)}"
+
 
 
 with gr.Blocks() as demo: