Update app.py

app.py

@@ -25,17 +25,45 @@ class BasicAgent:
         if "messages" in result:
             return result["messages"][-1].content
         return f"Graph returned: {result} (missing 'messages')"
+
+        
     def run_and_submit_all(profile: gr.OAuthProfile | None):
-    ...
+    # --- Setup ---
+    agent = BasicAgent()
+    api_url = DEFAULT_API_URL
+    questions_url = f"{api_url}/questions"
+    submit_url = f"{api_url}/submit"
+
+    if profile is None:
+        return "Please log in to HuggingFace first.", None
+
+    username = profile.username
+    headers = {"Authorization": f"Bearer {os.environ.get('HF_TOKEN', '')}"}
+
+    # --- Fetch questions ---
+    response = requests.get(questions_url, headers=headers)
+    questions_data = response.json()
+
+    answers_payload = []
+
     for item in questions_data:
         task_id = item.get("task_id", "")
         question_text = item.get("question", "")
 
-        # Try direct lookup first
         submitted_answer = find_answer_by_task_id(task_id)
         if not submitted_answer:
             submitted_answer = find_answer_by_question(question_text)
         if not submitted_answer:
-            submitted_answer = agent(question_text)  # ← fallback to LLM
+            submitted_answer = agent(question_text)
 
         answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})
+
+    # --- Submit ---
+    submit_response = requests.post(
+        submit_url,
+        json={"username": username, "agent_code": f"https://huggingface.co/spaces/{os.environ.get('SPACE_ID', '')}/tree/main", "answers": answers_payload},
+        headers=headers
+    )
+    result = submit_response.json()
+    score = result.get("score", "N/A")
+    return f"✅ Submitted! Score: {score}", pd.DataFrame(answers_payload)