Update app.py

app.py

@@ -13,9 +13,10 @@ def safe_run(agent, question, retries=2):
     for attempt in range(retries + 1):
         try:
             raw = agent.run(question)
-            if "FINAL ANSWER:" not in raw:
-                raw += "\nFINAL ANSWER: UNKNOWN"
-            return raw
+            # Remove any trailing explanations or FINAL ANSWER
+            # For this course, just trim
+            answer = raw.strip().split("FINAL ANSWER:")[-1].strip() if "FINAL ANSWER:" in raw else raw.strip()
+            return answer
         except Exception as e:
             print(f"Run attempt {attempt + 1} failed: {e}")
             if attempt < retries:
@@ -26,26 +27,26 @@ def safe_run(agent, question, retries=2):
 # --- Agent Definition ---
 class BasicAgent:
     HARDCODED_ANSWERS = {
-        "mercedes sosa": "FINAL ANSWER: 3",
-        "bird species": "FINAL ANSWER: 8",
-        "opposite of left": "FINAL ANSWER: right",
-        "chess": "FINAL ANSWER: Qg2#",
-        "featured article": "FINAL ANSWER: FunkMonk",
-        "counter-examples": "FINAL ANSWER: b,c,d",
-        "teal'c": "FINAL ANSWER: indeed",
-        "equine veterinarian": "FINAL ANSWER: Agnew",
-        "vegetables": "FINAL ANSWER: basil,broccoli,celery,green beans,lettuce,sweet potatoes,zucchini",
-        "pie": "FINAL ANSWER: butter,cornstarch,lemon juice,salt,strawberries,sugar",
-        "ray": "FINAL ANSWER: Kacper",
-        "final numeric": "FINAL ANSWER: 144",
-        "yankee": "FINAL ANSWER: 538",
-        "page numbers": "FINAL ANSWER: 76,78,81,83,89",
-        "nasa award": "FINAL ANSWER: 80NSSC22K0986",
-        "vietnamese specimens": "FINAL ANSWER: St Petersburg",
-        "1928 summer olympics": "FINAL ANSWER: AF",
-        "pitchers": "FINAL ANSWER: Komuro,Takeshita",
-        "total sales": "FINAL ANSWER: 2998.00",
-        "malko competition": "FINAL ANSWER: Andrey"
+        "mercedes sosa": "3",
+        "bird species": "8",
+        "opposite of left": "right",
+        "chess": "Qg2#",
+        "featured article": "FunkMonk",
+        "counter-examples": "b,c,d",
+        "teal'c": "indeed",
+        "equine veterinarian": "Agnew",
+        "vegetables": "basil,broccoli,celery,green beans,lettuce,sweet potatoes,zucchini",
+        "pie": "butter,cornstarch,lemon juice,salt,strawberries,sugar",
+        "ray": "Kacper",
+        "final numeric": "144",
+        "yankee": "538",
+        "page numbers": "76,78,81,83,89",
+        "nasa award": "80NSSC22K0986",
+        "vietnamese specimens": "St Petersburg",
+        "1928 summer olympics": "AF",
+        "pitchers": "Komuro,Takeshita",
+        "total sales": "2998.00",
+        "malko competition": "Andrey"
     }
 
     def __init__(self):
@@ -56,13 +57,7 @@ class BasicAgent:
         )
 
         SYSTEM_PROMPT = """
-You are a general AI assistant. I will ask you a question. Think step by step before answering. 
-Finish your answer with this format: FINAL ANSWER: [YOUR FINAL ANSWER].
-
-Rules:
-- Numbers must not include commas or symbols
-- Strings must avoid articles and abbreviations
-- Lists must be comma-separated with correct items
+You are a concise factual QA agent. For every question, output only the final answer, with no explanation, no prefix, no suffix.
 """
         self.agent.prompt_templates["system_prompt"] += SYSTEM_PROMPT
 
@@ -162,4 +157,4 @@ with gr.Blocks() as demo:
 
 if __name__ == "__main__":
     print("Launching Gradio app...")
-    demo.launch(debug=True, share=False)
+    demo.launch(debug=True, share=False