Update app.py

app.py

@@ -1,103 +1,93 @@
 # app.py
 
+import os
 import gradio as gr
-from postpartum_agent import PostpartumResearchAgent
 import requests
-import transformers
+import pandas as pd
+
+from postpartum_agent import PostpartumResearchAgent
 
 # Instantiate your agent
 agent = PostpartumResearchAgent()
 
-# GAIA API base URL
-GAIA_API_URL = "https://huggingface.co/api/gaia"
+# Correct GAIA Unit 4 scoring API
+DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
+# ✅ Single question: interactive
 def answer_question(question):
     return agent.kb_search(question)
-# def run(self, question):
-#     return self.kb_search(question)
-
-
-def submit_to_gaia(username, space_link):
-    GAIA_API_URL = "https://huggingface.co/api/gaia"
-
-    try:
-        # Step 1: Fetch questions
-        response = requests.get(f"{GAIA_API_URL}/questions")
-        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}"
-
-        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.strip(),
-            "agent_code": space_link.strip(),
-            "answers": answers
-        }
-
-        print("🚀 Submitting answers to GAIA...")
-        submit_response = requests.post(f"{GAIA_API_URL}/submit", json=payload)
-
-        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: {str(e)}"
-
-
 
+# ✅ Full submission flow: official Unit 4 style
+def run_and_submit_all(profile: gr.OAuthProfile | None):
+    if not profile:
+        return "❌ Please log in with your Hugging Face account.", None
+
+    username = profile.username
+    space_id = os.getenv("SPACE_ID")
+    agent_code = f"https://huggingface.co/spaces/{space_id}/tree/main"
+
+    questions_url = f"{DEFAULT_API_URL}/questions"
+    submit_url = f"{DEFAULT_API_URL}/submit"
+
+    # Fetch GAIA questions
+    response = requests.get(questions_url)
+    questions_data = response.json()
+
+    answers_payload = []
+    results_log = []
+
+    for item in questions_data:
+        task_id = item["task_id"]
+        question_text = item["question"]
+        submitted_answer = agent.kb_search(question_text)
+        answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})
+        results_log.append({
+            "Task ID": task_id,
+            "Question": question_text,
+            "Submitted Answer": submitted_answer
+        })
+
+    payload = {
+        "username": username,
+        "agent_code": agent_code,
+        "answers": answers_payload
+    }
+
+    submit_response = requests.post(submit_url, json=payload)
+    result = submit_response.json()
+
+    final_status = (
+        f"✅ Submission Complete!\n"
+        f"User: {result.get('username')}\n"
+        f"Score: {result.get('score', 'N/A')}%\n"
+        f"Correct: {result.get('correct_count', '?')}/{result.get('total_attempted', '?')}\n"
+        f"Message: {result.get('message', 'No message')}"
+    )
+
+    return final_status, pd.DataFrame(results_log)
+
+# ✅ ✅ ✅ Gradio app
 with gr.Blocks() as demo:
     gr.Markdown("# 🤱 Postpartum Research Agent")
-    gr.Markdown("Ask any postpartum question and see the AI agent help you!")
+    gr.Markdown(
+        "Ask any postpartum question and see the AI agent help you. "
+        "When ready, submit all answers to the GAIA leaderboard!"
+    )
 
     with gr.Row():
         with gr.Column():
-            question = gr.Textbox(label="Your Question", placeholder="How do I handle postpartum fatigue?")
-            output = gr.Textbox(label="Answer")
+            question = gr.Textbox(label="Your Question")
+            output = gr.Textbox(label="Agent Answer")
+            ask_btn = gr.Button("Get Answer")
 
         with gr.Column():
-            username = gr.Textbox(label="Your HF Username", placeholder="Your Hugging Face username")
-            space_link = gr.Textbox(label="Your Space Link", placeholder="https://huggingface.co/spaces/YOU/YOUR_SPACE/tree/main")
-            submit_btn = gr.Button("Submit to GAIA")
-            submit_output = gr.Textbox(label="Submission Result")
-
-    ask_btn = gr.Button("Get Answer")
+            gr.Markdown("## Submit to GAIA Benchmark")
+            gr.LoginButton()
+            run_button = gr.Button("Run & Submit All")
+            status_output = gr.Textbox(label="Submission Status", lines=5)
+            results_table = gr.DataFrame(label="Questions & Answers")
 
     ask_btn.click(fn=answer_question, inputs=question, outputs=output)
-    submit_btn.click(fn=submit_to_gaia, inputs=[username, space_link], outputs=submit_output)
+    run_button.click(fn=run_and_submit_all, outputs=[status_output, results_table])
 
 demo.launch()