Update app.py

app.py

@@ -1,71 +1,92 @@
-import streamlit as st
+import gradio as gr
 import requests
 
-BASE_URL = "https://agents-course-unit4-scoring.hf.space"
+# ==============================
+#   CONFIGURATION
+# ==============================
+API_BASE = "https://agents-course-unit4-scoring.hf.space"
 
+# ==============================
+#   BASIC AGENT CLASS
+# ==============================
 class BasicAgent:
     def __init__(self):
-        # You can initialize any tools, memory, or logic here
         pass
 
-    def answer_question(self, question: str) -> str:
-        """
-        Implement your custom logic here.
-        For now, this just echoes the question.
-        """
-        return f"This is my answer to: {question}"
+    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."
 
-# Initialize Streamlit app
-st.set_page_config(page_title="Basic Agent Evaluation Runner", page_icon="🤖", layout="centered")
-st.title("🤖 Basic Agent Evaluation Runner")
+# ==============================
+#   API INTERACTION FUNCTIONS
+# ==============================
+def fetch_questions():
+    try:
+        res = requests.get(f"{API_BASE}/questions")
+        res.raise_for_status()
+        return res.json()
+    except Exception as e:
+        return {"error": str(e)}
 
-st.markdown("""
-### Instructions:
-1️⃣ Clone this space on your Hugging Face profile.  
-2️⃣ Modify the `BasicAgent` class with your logic.  
-3️⃣ Log in below and run evaluation.
-""")
+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()
 
-# --- UI for Hugging Face login ---
-with st.expander("🔐 Login"):
-    hf_token = st.text_input("Enter your Hugging Face token", type="password")
-    if hf_token:
-        st.success("✅ Token saved!")
+# ==============================
+#   GRADIO UI LOGIC
+# ==============================
+def run_evaluation(username, agent_code_link):
+    agent = BasicAgent()
+    data = fetch_questions()
 
-# --- Initialize agent ---
-agent = BasicAgent()
+    if "error" in data:
+        return f"❌ Error fetching questions: {data['error']}", None
 
-# --- Fetch questions ---
-st.subheader("📋 Questions and Answers")
+    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})
 
-if st.button("🚀 Run Evaluation & Submit All Answers"):
-    st.info("Fetching questions...")
     try:
-        response = requests.get(f"{BASE_URL}/questions")  # Correct endpoint
-        if response.status_code != 200:
-            st.error(f"Failed to fetch questions: {response.status_code} {response.reason}")
-        else:
-            data = response.json()
-            st.success(f"Fetched {len(data)} questions.")
-            answers = []
+        result = submit_answers(username, agent_code_link, results)
+    except Exception as e:
+        result = {"error": str(e)}
 
-            for q in data:
-                question = q.get("question", "")
-                task_id = q.get("task_id", "")
-                answer = agent.answer_question(question)
-                answers.append({
-                    "task_id": task_id,
-                    "answer": answer
-                })
-                st.write(f"**Q:** {question}")
-                st.write(f"**A:** {answer}")
+    return result, results
 
-            # Submit answers
-            st.info("Submitting answers...")
-            submit_res = requests.post(f"{BASE_URL}/submit", json={"answers": answers})
-            if submit_res.status_code == 200:
-                st.success("✅ Submission complete! Check leaderboard or logs.")
-            else:
-                st.error(f"Submission failed: {submit_res.status_code}")
-    except Exception as e:
-        st.error(f"Error: {str(e)}")
+# ==============================
+#   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()