Update app.py

app.py

@@ -2,22 +2,20 @@ import os
 import gradio as gr
 import requests
 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"))
+from smolagents import ToolCallingAgent, tool
+from instructor import OpenAIChat
+from duckduckgo_search import DDGS
 
 # --- Tools ---
 @tool
 def web_search(query: str) -> str:
     """Performs a web search using DuckDuckGo.
-    
+
     Args:
         query: The search query string.
-        
+
     Returns:
         A formatted string with search results.
     """
@@ -34,27 +32,26 @@ def web_search(query: str) -> str:
 @tool
 def calculate(expression: str) -> str:
     """Evaluates mathematical expressions.
-    
+
     Args:
         expression: The math expression to evaluate.
-        
+
     Returns:
         The result as a string or error message.
     """
     try:
-        safe_dict = {k: v for k, v in math.__dict__.items() 
-                    if not k.startswith("__")}
+        safe_dict = {k: v for k, v in math.__dict__.items() if not k.startswith("__")}
         safe_dict.update({
             '__builtins__': None,
             'abs': abs,
             'round': round
         })
-        result = eval(expression, {'__builtins__': None}, safe_dict)
+        result = eval(expression, {"__builtins__": None}, safe_dict)
         return str(result)
     except Exception as e:
         return f"Calculation error: {str(e)}"
 
-# --- Agent with Proper Initialization ---
+# --- Agent ---
 class GAIAAgent:
     def __init__(self):
         self.system_prompt = """You are an AI assistant that answers questions using tools:
@@ -63,13 +60,11 @@ class GAIAAgent:
 - Be concise and accurate"""
 
         try:
-            # Initialize with proper configuration
             self.agent = ToolCallingAgent(
                 name="GAIA_Agent",
                 description=self.system_prompt,
                 tools=[web_search, calculate],
-                model=client,  # Pass the OpenAI client directly
-                model_kwargs={"model": "gpt-3.5-turbo"},  # Specify model here
+                model=OpenAIChat(model="gpt-3.5-turbo"),  # ✅ Fixed model wrapper
                 planning_interval=3
             )
             print("✅ Agent initialized successfully")
@@ -77,35 +72,30 @@ class GAIAAgent:
             raise RuntimeError(f"Agent init failed: {str(e)}")
 
     def __call__(self, question: str) -> str:
-        """Process a question with proper error handling."""
         try:
             response = self.agent.run(question)
             return str(response) if response else "No answer generated"
         except Exception as e:
             return f"Error: {str(e)}"
 
-
-# --- Gradio Interface ---
+# --- Gradio Submission Logic ---
 def submit_answers(profile: gr.OAuthProfile | None):
     if not profile:
         return "Please login to Hugging Face", None
 
     try:
         agent = GAIAAgent()
-        response = requests.get(
-            "https://agents-course-unit4-scoring.hf.space/questions",
-            timeout=20
-        )
+        response = requests.get("https://agents-course-unit4-scoring.hf.space/questions", timeout=20)
         questions = response.json() or []
-        
+
         answers = []
         results = []
-        
-        for item in questions[:15]:
+
+        for item in questions[:15]:  # Limit to 15 for speed
             task_id = item.get("task_id", "")
             question = item.get("question", "")
             answer = agent(question)
-            
+
             answers.append({
                 "task_id": task_id,
                 "submitted_answer": answer[:1000]
@@ -115,7 +105,7 @@ def submit_answers(profile: gr.OAuthProfile | None):
                 "Question": question[:100],
                 "Answer": answer[:200]
             })
-            
+
         submit_response = requests.post(
             "https://agents-course-unit4-scoring.hf.space/submit",
             json={
@@ -126,24 +116,25 @@ def submit_answers(profile: gr.OAuthProfile | None):
             timeout=60
         )
         data = submit_response.json()
-        
+
         return (
             f"Submitted {len(answers)} answers\n"
             f"Score: {data.get('score', 'N/A')}%\n"
             f"Correct: {data.get('correct_count', 0)}/{len(answers)}",
             pd.DataFrame(results)
         )
-        
+
     except Exception as e:
         return f"Error: {str(e)}", None
 
+# --- Gradio Interface ---
 with gr.Blocks() as demo:
-    gr.Markdown("# GAIA Agent")
+    gr.Markdown("# GAIA Agent with DuckDuckGo + Calculator")
     gr.LoginButton()
-    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])
+    btn = gr.Button("Run Evaluation", variant="primary")
+    status = gr.Textbox(label="Results")
+    df = gr.DataFrame(label="Evaluation Details")
+    btn.click(submit_answers, outputs=[status, df])
 
 if __name__ == "__main__":
     demo.launch()