Update app.py

app.py

@@ -2,78 +2,76 @@ import os
 import gradio as gr
 import requests
 import pandas as pd
+from openai import OpenAI
+
+# =============================
+# CONSTANTS (DO NOT CHANGE)
+# =============================
+DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
+
+SYSTEM_PROMPT = """
+You are a general AI assistant.
+Answer the question and finish your response with:
+
+FINAL ANSWER: <answer>
+
+Rules:
+- If number: no commas, no units unless specified
+- If string: no articles, no abbreviations
+- If list: comma-separated, minimal words
+"""
+
+# =============================
+# MODEL
+# =============================
+client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
+
+def llm_answer(question: str) -> str:
+    response = client.chat.completions.create(
+        model="gpt-4o-mini",
+        messages=[
+            {"role": "system", "content": SYSTEM_PROMPT},
+            {"role": "user", "content": question}
+        ],
+        temperature=0
+    )
 
-API_URL = "https://agents-course-unit4-scoring.hf.space"
-
-
-# --------------------------
-# SMART RULE-BASED AGENT
-# --------------------------
-def agent_answer(question: str) -> str:
-    q = question.lower()
-
-    # 1️⃣ Mercedes Sosa
-    if "mercedes sosa" in q and "studio albums" in q:
-        return "4"
-
-    # 2️⃣ 1928 Olympics – least athletes
-    if "1928 summer olympics" in q and "least number of athletes" in q:
-        return "AFG"
-
-    # 3️⃣ Opposite of left
-    if "opposite of left" in q:
-        return "right"
-
-    # 4️⃣ Malko Competition
-    if "malko competition" in q and "first name" in q:
-        return "Erik"
-
-    # 5️⃣ Bird species in video
-    if "bird species" in q:
-        return "4"
-
-    # 6️⃣ Chess move fallback
-    if "chess" in q:
-        return "Qh5"
-
-    # 7️⃣ Excel sales question (safe numeric format)
-    if "excel file" in q and "total sales" in q:
-        return "1234.56"
-
-    # 8️⃣ Pitcher question (safe format)
-    if "pitcher" in q and "taishō tamai" in q:
-        return "Suzuki, Tanaka"
+    text = response.choices[0].message.content
 
-    # ---- DEFAULT FALLBACK ----
-    return "I don't know"
+    # Extract FINAL ANSWER only
+    if "FINAL ANSWER:" in text:
+        return text.split("FINAL ANSWER:")[-1].strip()
 
+    return text.strip()
 
-# --------------------------
-# RUN + SUBMIT
-# --------------------------
-def run_and_submit(profile: gr.OAuthProfile | None):
+# =============================
+# GAIA PIPELINE
+# =============================
+def run_and_submit_all(profile: gr.OAuthProfile | None):
     if not profile:
-        return "❌ Please login to Hugging Face.", None
+        return "❌ Please login to Hugging Face", None
 
     username = profile.username.strip()
-    space_id = os.getenv("SPACE_ID", "UNKNOWN")
-    agent_code = f"https://huggingface.co/spaces/{space_id}"
+    space_id = os.getenv("SPACE_ID")
+    agent_code = f"https://huggingface.co/spaces/{space_id}/tree/main"
 
     # Fetch questions
-    try:
-        questions = requests.get(f"{API_URL}/questions", timeout=20).json()
-    except Exception as e:
-        return f"❌ Error fetching questions: {e}", None
+    questions = requests.get(f"{DEFAULT_API_URL}/questions").json()
 
     answers = []
     logs = []
 
     for q in questions:
-        ans = agent_answer(q["question"])
+        try:
+            ans = llm_answer(q["question"])
+        except Exception as e:
+            ans = "I don't know"
+
         answers.append({
             "task_id": q["task_id"],
             "submitted_answer": ans
         })
+
         logs.append({
             "Task ID": q["task_id"],
             "Question": q["question"],
@@ -86,46 +84,29 @@ def run_and_submit(profile: gr.OAuthProfile | None):
         "answers": answers
     }
 
-    try:
-        response = requests.post(
-            f"{API_URL}/submit",
-            json=payload,
-            timeout=60
-        )
-        response.raise_for_status()
-        result = response.json()
-    except Exception as e:
-        return f"❌ Submission failed: {e}", pd.DataFrame(logs)
+    res = requests.post(f"{DEFAULT_API_URL}/submit", json=payload).json()
 
     status = (
         f"✅ Submission Successful!\n"
-        f"User: {result.get('username')}\n"
-        f"Score: {result.get('score')}%\n"
-        f"Correct: {result.get('correct_count')}/{result.get('total_attempted')}\n"
-        f"Message: {result.get('message')}"
+        f"User: {res.get('username')}\n"
+        f"Score: {res.get('score')}%\n"
+        f"Correct: {res.get('correct_count')}/{res.get('total_attempted')}\n"
+        f"Message: {res.get('message')}"
     )
 
     return status, pd.DataFrame(logs)
 
-
-# --------------------------
-# GRADIO UI
-# --------------------------
+# =============================
+# UI
+# =============================
 with gr.Blocks() as demo:
-    gr.Markdown("# 🤖 GAIA Level-1 Agent (Version 2)")
-    gr.Markdown("Login → Run → Submit")
-
+    gr.Markdown("# 🤖 GAIA Level-1 Agent (Unit-4)")
     gr.LoginButton()
-    submit_btn = gr.Button("Run Evaluation & Submit")
 
-    status_box = gr.Textbox(label="Submission Result", lines=6)
-    log_table = gr.DataFrame(label="Agent Answers", wrap=True)
-
-    submit_btn.click(
-        fn=run_and_submit,
-        outputs=[status_box, log_table]
-    )
+    run_btn = gr.Button("Run Evaluation & Submit")
+    status = gr.Textbox(label="Submission Result", lines=5)
+    table = gr.Dataframe(label="Questions & Answers")
 
+    run_btn.click(run_and_submit_all, outputs=[status, table])
 
-if __name__ == "__main__":
-    demo.launch()
+demo.launch()