Update app.py

app.py

@@ -18,30 +18,38 @@ def answer_question(question):
 
 
 def submit_to_gaia(username, space_link):
-    # Get questions
-    response = requests.get(f"{GAIA_API_URL}/questions")
-    questions = response.json()
-
-    answers = []
-    for q in questions:
-        task_id = q['task_id']
-        question_text = q['question']
-        agent_answer = agent.run(question_text)
-        answers.append({
-            "task_id": task_id,
-            "submitted_answer": agent_answer
-        })
-
-    payload = {
-        "username": username,
-        "agent_code": space_link,
-        "answers": answers
-    }
-
-    submit_response = requests.post(f"{GAIA_API_URL}/submit", json=payload)
-    result = submit_response.json()
-
-    return f"Submitted! Result: {result}"
+    GAIA_API_URL = "https://huggingface.co/api/gaia"
+    
+    try:
+        # 1. Fetch tasks from GAIA
+        response = requests.get(f"{GAIA_API_URL}/questions")
+        questions = response.json()  # Now properly parsed into a list of dicts
+
+        # 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
+        payload = {
+            "username": username,
+            "agent_code": space_link,
+            "answers": answers
+        }
+        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')}"
+    
+    except Exception as e:
+        return f"❌ Submission failed due to: {str(e)}"
+
 
 with gr.Blocks() as demo:
     gr.Markdown("# 🤱 Postpartum Research Agent")