Update app.py

app.py

@@ -5,6 +5,10 @@ import pandas as pd
 from smolagents import ToolCallingAgent, tool
 from duckduckgo_search import DDGS
 import math
+from openai import OpenAI
+
+# Initialize OpenAI client
+client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
 
 # --- Tools ---
 @tool
@@ -15,7 +19,7 @@ def web_search(query: str) -> str:
         query: The search query string.
         
     Returns:
-        A formatted string with search results or error message.
+        A formatted string with search results.
     """
     try:
         with DDGS() as ddgs:
@@ -29,7 +33,7 @@ def web_search(query: str) -> str:
 
 @tool
 def calculate(expression: str) -> str:
-    """Evaluates mathematical expressions safely.
+    """Evaluates mathematical expressions.
     
     Args:
         expression: The math expression to evaluate.
@@ -40,36 +44,37 @@ def calculate(expression: str) -> str:
     try:
         safe_dict = {k: v for k, v in math.__dict__.items() 
                     if not k.startswith("__")}
-        safe_dict.update({'__builtins__': None, 'abs': abs, 'round': round})
-        return str(eval(expression, {'__builtins__': None}, safe_dict))
+        safe_dict.update({
+            '__builtins__': None,
+            'abs': abs,
+            'round': round
+        })
+        result = eval(expression, {'__builtins__': None}, safe_dict)
+        return str(result)
     except Exception as e:
         return f"Calculation error: {str(e)}"
 
-# --- Agent with System Prompt ---
+# --- Agent with Proper Model Initialization ---
 class GAIAAgent:
     def __init__(self):
-        self.system_prompt = """You are an AI assistant designed to answer questions accurately using these tools:
-
-1. web_search - For finding current information
-2. calculate - For mathematical calculations
-
-Guidelines:
-- For factual questions, always use web_search
-- For math problems, use calculate
-- Be concise but thorough
-- Verify information when possible
-- If unsure, say so rather than guessing"""
+        self.system_prompt = """You are an AI assistant that answers questions using tools:
+- Use web_search for factual queries
+- Use calculate for math problems
+- Be concise and accurate"""
 
         try:
+            # Initialize with proper ChatModel instance
             self.agent = ToolCallingAgent(
                 name="GAIA_Agent",
                 description=self.system_prompt,
                 tools=[web_search, calculate],
-                model="gpt-3.5-turbo",
+                model=client.chat.completions,  # Proper model instance
+                model_name="gpt-3.5-turbo",
                 planning_interval=3
             )
+            print("✅ Agent initialized with OpenAI model")
         except Exception as e:
-            raise RuntimeError(f"Agent initialization failed: {str(e)}")
+            raise RuntimeError(f"Agent init failed: {str(e)}")
 
     def __call__(self, question: str) -> str:
         """Process a question with proper error handling."""
@@ -95,11 +100,11 @@ def submit_answers(profile: gr.OAuthProfile | None):
         answers = []
         results = []
         
-        for item in questions[:15]:  # Process first 15 for testing
+        for item in questions[:15]:
             task_id = item.get("task_id", "")
             question = item.get("question", "")
-            
             answer = agent(question)
+            
             answers.append({
                 "task_id": task_id,
                 "submitted_answer": answer[:1000]
@@ -110,7 +115,6 @@ def submit_answers(profile: gr.OAuthProfile | None):
                 "Answer": answer[:200]
             })
             
-        # Submit answers
         submit_response = requests.post(
             "https://agents-course-unit4-scoring.hf.space/submit",
             json={
@@ -138,7 +142,6 @@ with gr.Blocks() as demo:
     submit_btn = gr.Button("Run Evaluation", variant="primary")
     output = gr.Textbox(label="Results")
     table = gr.DataFrame(label="Details")
-    
     submit_btn.click(submit_answers, outputs=[output, table])
 
 if __name__ == "__main__":