Update app.py

app.py

@@ -1,92 +1,101 @@
 import gradio as gr
 import requests
+import json
 
-# ==============================
-#   CONFIGURATION
-# ==============================
+# === CONFIG ===
 API_BASE = "https://agents-course-unit4-scoring.hf.space"
 
-# ==============================
-#   BASIC AGENT CLASS
-# ==============================
-class BasicAgent:
-    def __init__(self):
-        pass
-
-    def __call__(self, question: str) -> str:
-        # Simple example agent — replace with your own logic
-        # You can later integrate a Hugging Face model or reasoning steps
-        if "capital" in question.lower():
-            return "Paris"
-        return "I need to research that."
-
-# ==============================
-#   API INTERACTION FUNCTIONS
-# ==============================
+# === CORE FUNCTIONS ===
+
 def fetch_questions():
+    """Fetch all available questions."""
     try:
-        res = requests.get(f"{API_BASE}/questions")
-        res.raise_for_status()
-        return res.json()
+        response = requests.get(f"{API_BASE}/questions")
+        response.raise_for_status()
+        questions = response.json()
+        return questions
     except Exception as e:
-        return {"error": str(e)}
-
-def submit_answers(username, agent_code_link, answers):
-    payload = {
-        "username": username,
-        "agent_code": agent_code_link,
-        "answers": answers
-    }
-    res = requests.post(f"{API_BASE}/submit", json=payload)
-    return res.json()
-
-# ==============================
-#   GRADIO UI LOGIC
-# ==============================
-def run_evaluation(username, agent_code_link):
-    agent = BasicAgent()
-    data = fetch_questions()
-
-    if "error" in data:
-        return f"❌ Error fetching questions: {data['error']}", None
-
-    results = []
-    for q in data:
-        task_id = q["task_id"]
-        question = q["question"]
-        answer = agent(question)
-        results.append({"task_id": task_id, "submitted_answer": answer})
+        return f"⚠️ Error fetching questions: {e}"
 
+def fetch_random_question():
+    """Fetch a random question."""
     try:
-        result = submit_answers(username, agent_code_link, results)
+        response = requests.get(f"{API_BASE}/random-question")
+        response.raise_for_status()
+        q = response.json()
+        return f"**Task ID:** {q['task_id']}\n\n**Question:** {q['question']}"
     except Exception as e:
-        result = {"error": str(e)}
-
-    return result, results
-
-# ==============================
-#   GRADIO INTERFACE
-# ==============================
-with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
-    gr.Markdown("# 🚀 Basic Agent Evaluation Runner")
-    gr.Markdown("""
-    1️⃣ **Clone this Space** on your Hugging Face profile.  
-    2️⃣ **Modify the `BasicAgent`** class with your logic.  
-    3️⃣ **Log in** and run evaluation to test your agent.  
-    """)
-
-    username = gr.Textbox(label="Your Hugging Face Username")
-    agent_code_link = gr.Textbox(label="Public Hugging Face Space Code Link", placeholder="https://huggingface.co/spaces/YourName/YourSpace/tree/main")
-    
-    run_btn = gr.Button("Run Evaluation & Submit All Answers 🚀")
-
-    output_status = gr.JSON(label="Run Status / Submission Result")
-    output_table = gr.JSON(label="Questions and Agent Answers")
-
-    run_btn.click(
-        fn=run_evaluation,
-        inputs=[username, agent_code_link],
-        outputs=[output_status, output_table]
-    )
-
-demo.launch()
+        return f"⚠️ Error fetching random question: {e}"
+
+def submit_answers(username, code_link, answers_json):
+    """Submit answers and return the score."""
+    try:
+        answers = json.loads(answers_json)
+        payload = {
+            "username": username,
+            "agent_code": code_link,
+            "answers": answers
+        }
+
+        response = requests.post(f"{API_BASE}/submit", json=payload)
+        response.raise_for_status()
+
+        result = response.json()
+        return json.dumps(result, indent=2)
+
+    except json.JSONDecodeError:
+        return "❌ Invalid JSON format for answers. Make sure it's a valid list of objects."
+    except Exception as e:
+        return f"⚠️ Submission failed: {e}"
+
+# === DEMO / UI ===
+
+def auto_agent():
+    """Simple rule-based agent for demo purposes."""
+    questions = fetch_questions()
+    if isinstance(questions, str):
+        return questions  # Error message
+
+    answers = []
+    for q in questions[:5]:  # just answer first 5 for example
+        answers.append({
+            "task_id": q["task_id"],
+            "submitted_answer": "42"  # dummy placeholder
+        })
+    return json.dumps(answers, indent=2)
+
+# === GRADIO UI ===
+
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
+    gr.Markdown("## 🤖 GAIA Agent Submission Portal")
+    gr.Markdown("This interface lets you test your agent, view questions, and submit your answers for leaderboard scoring.")
+
+    with gr.Tab("📋 Fetch Questions"):
+        fetch_all_btn = gr.Button("Fetch All Questions")
+        questions_output = gr.Textbox(label="All Questions", lines=10)
+        fetch_all_btn.click(fetch_questions, outputs=questions_output)
+
+        fetch_random_btn = gr.Button("Fetch Random Question")
+        random_output = gr.Markdown()
+        fetch_random_btn.click(fetch_random_question, outputs=random_output)
+
+    with gr.Tab("🧠 Agent Answers"):
+        gr.Markdown("Generate dummy answers to test the submission structure.")
+        gen_btn = gr.Button("Generate Sample Answers")
+        answers_box = gr.Textbox(label="Generated Answers (Editable)", lines=10)
+        gen_btn.click(auto_agent, outputs=answers_box)
+
+    with gr.Tab("🚀 Submit Answers"):
+        username = gr.Textbox(label="Your Hugging Face Username")
+        code_link = gr.Textbox(label="Public Space Code Link (e.g. https://huggingface.co/spaces/yourname/yourspace/tree/main)")
+        answers_json = gr.Textbox(label="Answers JSON", lines=10)
+        submit_btn = gr.Button("Submit to Leaderboard")
+        result_box = gr.Textbox(label="Result / Score", lines=10)
+        submit_btn.click(submit_answers, inputs=[username, code_link, answers_json], outputs=result_box)
+
+    gr.Markdown("---")
+    gr.Markdown("💡 **Tip:** Replace the dummy answers with your real agent’s output to get scored.")
+
+# === RUN ===
+if __name__ == "__main__":
+    demo.launch()