Update app.py

app.py

@@ -9,27 +9,32 @@ openai.api_key = os.getenv("OPENAI_API_KEY")
 
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
-# --- Smart Agent Definition ---
+# --- Improved Smart Agent ---
 class SmartAgent:
     def __init__(self):
         print("SmartAgent initialized.")
 
     def __call__(self, question: str) -> str:
-        print(f"[Agent] Processing question: {question[:50]}...")
-        prompt = f"""
-You are an expert AI agent. Answer the following question clearly and precisely. 
-If external data is required (e.g. Wikipedia), use your knowledge base.
-Only return the final answer. Do NOT include 'Answer:' or any extra words.
-
-Question:
-{question}
-"""
+        print(f"[Agent] Processing question: {question[:60]}...")
         try:
             response = openai.ChatCompletion.create(
                 model="gpt-4",
-                messages=[{"role": "user", "content": prompt}],
-                temperature=0.4,
-                max_tokens=300
+                messages=[
+                    {
+                        "role": "system",
+                        "content": (
+                            "You are a highly logical AI assistant specialized in answering "
+                            "knowledge and reasoning questions precisely and concisely. "
+                            "Provide only the final answer, with no extra commentary or labels."
+                        )
+                    },
+                    {"role": "user", "content": question}
+                ],
+                temperature=0.2,  # Lower temp for more deterministic answers
+                max_tokens=300,
+                top_p=1,
+                frequency_penalty=0,
+                presence_penalty=0,
             )
             answer = response.choices[0].message["content"].strip()
             print(f"[Agent] Answer: {answer}")
@@ -103,6 +108,21 @@ with gr.Blocks() as demo:
 
     run_btn.click(fn=run_and_submit_all, outputs=[status, table])
 
+# --- Optional local test for debugging ---
+def local_test():
+    print("Running local test...")
+    agent = SmartAgent()
+    test_questions = [
+        "How many studio albums did Mercedes Sosa publish between 2000 and 2009?",
+        "What is the capital city of France?",
+        "List the vegetables in this list: milk, eggs, flour, whole bean coffee, Oreos, sweet potatoes, fresh basil, plums, green beans, rice, corn, bell pepper, whole allspice, acorns, broccoli, celery, zucchini, lettuce, peanuts."
+    ]
+    for q in test_questions:
+        ans = agent(q)
+        print(f"Q: {q}\nA: {ans}\n{'-'*40}")
+
 if __name__ == "__main__":
     print("Launching GAIA Agent App")
+    local_test()
     demo.launch()
+