Update app.py

app.py

@@ -5,9 +5,7 @@ import pandas as pd
 from smolagents import ToolCallingAgent, tool
 from duckduckgo_search import DDGS
 import math
-
-# --- Constants ---
-DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
+import openai
 
 # --- Tools ---
 @tool
@@ -22,24 +20,23 @@ def duck_search(query: str) -> str:
         A string summarizing the top search results.
     """
     try:
-        results = ddg(query, max_results=3)
-        if results:
-            return "\n\n".join([f"{r['title']}: {r['body']}" for r in results])
-        else:
-            return "No results found."
+        with DDGS() as ddgs:
+            results = ddgs.text(query, max_results=3)
+            summaries = [f"{r['title']}: {r['body']}" for r in results]
+            return "\n\n".join(summaries) if summaries else "No results found."
     except Exception as e:
         return f"Search error: {e}"
 
 @tool
 def calculator(expression: str) -> str:
     """
-    Safely evaluates math expressions using Python's math module.
+    Evaluates basic math expressions using math module.
 
     Args:
-        expression: A valid math expression as a string (e.g., 'sqrt(16) + 10').
+        expression: A math expression as a string.
 
     Returns:
-        The result of the evaluated expression or an error message.
+        The result or an error message.
     """
     try:
         result = eval(expression, {"__builtins__": {}}, math.__dict__)
@@ -52,12 +49,13 @@ class WebSearchAgent:
     def __init__(self):
         self.agent = ToolCallingAgent(
             name="GAIAWebToolAgent",
-            description="Agent that answers questions using web search and calculator tools.",
+            description="An agent using DuckDuckGo and calculator tools.",
             tools=[duck_search, calculator],
+            model="gpt-3.5-turbo",  # You can use gpt-4o if you have access
             step_limit=5,
             system_prompt=(
-                "You're a helpful reasoning agent. Use available tools like web search "
-                "and calculator to answer the user's question accurately and concisely."
+                "You're a helpful reasoning agent. Use the provided tools "
+                "to help answer the user's questions accurately."
             ),
         )
         print("✅ Agent initialized.")
@@ -70,7 +68,10 @@ class WebSearchAgent:
             print(f"❌ Agent error: {e}")
             return f"Error: {e}"
 
-# --- Evaluation and Submission ---
+# --- Constants ---
+DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
+
+# --- Evaluation & Submission ---
 def run_and_submit_all(profile: gr.OAuthProfile | None):
     space_id = os.getenv("SPACE_ID")
     if profile:
@@ -89,13 +90,11 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
         return f"Agent initialization error: {e}", None
 
     try:
-        print("📥 Fetching questions...")
         response = requests.get(questions_url, timeout=15)
         response.raise_for_status()
         questions = response.json()
         if not questions:
             return "No questions received.", None
-        print(f"✅ Retrieved {len(questions)} questions.")
     except Exception as e:
         return f"Failed to fetch questions: {e}", None
 
@@ -109,27 +108,15 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
             continue
         try:
             answer = agent(question)
-            results_log.append({
-                "Task ID": task_id,
-                "Question": question,
-                "Submitted Answer": answer
-            })
-            answers_payload.append({
-                "task_id": task_id,
-                "submitted_answer": answer
-            })
+            results_log.append({"Task ID": task_id, "Question": question, "Submitted Answer": answer})
+            answers_payload.append({"task_id": task_id, "submitted_answer": answer})
         except Exception as e:
             error_msg = f"Agent error: {e}"
-            results_log.append({
-                "Task ID": task_id,
-                "Question": question,
-                "Submitted Answer": error_msg
-            })
+            results_log.append({"Task ID": task_id, "Question": question, "Submitted Answer": error_msg})
 
     if not answers_payload:
         return "No answers were generated.", pd.DataFrame(results_log)
 
-    print("📤 Submitting answers...")
     try:
         response = requests.post(submit_url, json={
             "username": username.strip(),
@@ -149,14 +136,9 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     except Exception as e:
         return f"❌ Submission failed: {e}", pd.DataFrame(results_log)
 
-# --- Gradio UI ---
+# --- UI ---
 with gr.Blocks() as demo:
-    gr.Markdown("# 🤖 GAIA Agent with Web Search & Calculator Tools")
-    gr.Markdown("""
-    - ✅ Log in to your Hugging Face account
-    - 🚀 Click the button to run and submit your agent
-    - 🧠 Agent uses DuckDuckGo search + calculator
-    """)
+    gr.Markdown("# 🤖 GAIA Agent with Web Search & Calculator")
     gr.LoginButton()
     run_btn = gr.Button("Run Evaluation & Submit All Answers")
     status_box = gr.Textbox(label="Status", lines=5)
@@ -165,5 +147,4 @@ with gr.Blocks() as demo:
     run_btn.click(fn=run_and_submit_all, outputs=[status_box, result_table])
 
 if __name__ == "__main__":
-    print("🚀 Starting Gradio App...")
-    demo.launch(debug=True, share=False)
+    demo.launch(debug=True, share=False)