Update app.py

app.py

@@ -4,37 +4,32 @@ import requests
 import inspect
 import pandas as pd
 import openai
+from smolagents.agents import ToolCallingAgent
+from smolagents.tools import CodeInterpreterTool
+from langchain_community.tools import DuckDuckGoSearchRun
 
-# (Keep Constants as is)
-# --- Constants ---
-DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
-
-# --- Basic Agent Definition ---
-# ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
-class BasicAgent:
+class SmartGAIAAgent:
     def __init__(self):
-        openai.api_key = os.getenv("OPENAI_API_KEY")  # Set this in HF secrets
-
-    def __call__(self, question: str) -> str:
-        system_prompt = (
-            "You are a smart AI assistant that answers factual questions clearly and concisely. "
-            "If a question requires step-by-step reasoning, do it in your head before producing a short answer. "
-            "Answer with a single sentence or phrase, no extra explanation, and no 'Final answer' phrase."
+        openai.api_key = os.getenv("OPENAI_API_KEY")
+
+        self.search = DuckDuckGoSearchRun()
+        self.calculator = CodeInterpreterTool()
+
+        self.agent = ToolCallingAgent(
+            tools=[self.search, self.calculator],
+            model="gpt-4",  # or gpt-3.5-turbo
+            max_steps=8,
+            system_prompt=(
+                "You are an expert assistant answering complex factual and reasoning questions. "
+                "Use tools only when necessary. Only return the final answer, nothing else."
+            )
         )
 
+    def __call__(self, question: str) -> str:
         try:
-            response = openai.ChatCompletion.create(
-                model="gpt-4",
-                messages=[
-                    {"role": "system", "content": system_prompt},
-                    {"role": "user", "content": question}
-                ],
-                temperature=0.3,
-            )
-            answer = response['choices'][0]['message']['content'].strip()
-            return answer
+            return self.agent.run(question)
         except Exception as e:
-            print(f"Error during OpenAI call: {e}")
+            print(f"Agent error: {e}")
             return "error"
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):