Update app.py

app.py

@@ -4,27 +4,29 @@ import gradio as gr
 import requests
 import pandas as pd
 
-from smolagents import CodeAgent, OpenAIServerModel, DuckDuckGoSearchTool
+from smolagents import CodeAgent, OpenAIServerModel
+from smolagents.tools import WebSearchTool  # Correct tool
 
 # Constants
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 MAX_QUESTION_LENGTH = 4000
+MAX_TOKENS_PER_QUESTION = 8000  # Safety margin to avoid 8192-token error
 
 # Reliable search tool with retry
-class ReliableDuckDuckGoTool(DuckDuckGoSearchTool):
+class ReliableWebSearchTool(WebSearchTool):
     def run(self, query: str) -> str:
         for attempt in range(3):
             try:
                 return super().run(query)
             except Exception as e:
                 if "rate" in str(e).lower():
-                    print(f"[DuckDuckGo] Rate limit, retry {attempt+1}/3")
+                    print(f"[WebSearchTool] Rate limit, retry {attempt+1}/3")
                     time.sleep(2 * (attempt + 1))
                 else:
                     raise
-        raise RuntimeError("DuckDuckGo search failed after retries")
+        raise RuntimeError("WebSearchTool failed after retries")
 
-# Main agent using GPT-4 & tools
+# Main agent using GPT-4 and tools
 class SmartGAIAAgent:
     def __init__(self):
         key = os.getenv("OPENAI_API_KEY")
@@ -32,7 +34,7 @@ class SmartGAIAAgent:
             raise ValueError("Missing OPENAI_API_KEY")
         model = OpenAIServerModel(model_id="gpt-4", api_key=key)
         self.agent = CodeAgent(
-            tools=[ReliableDuckDuckGoTool()],
+            tools=[ReliableWebSearchTool()],
             model=model,
             add_base_tools=True
         )
@@ -51,23 +53,25 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     if not username:
         return "Please Login to Hugging Face", None
 
-    # instantiate agent
+    # Instantiate agent
     try:
         agent = SmartGAIAAgent()
     except Exception as e:
         return f"Error initializing agent: {e}", None
 
-    # get questions
+    # Get questions
     resp = requests.get(f"{DEFAULT_API_URL}/questions", timeout=15)
     resp.raise_for_status()
     questions = resp.json()
 
     payload, logs = [], []
-    skip_kw = ['.mp3','.wav','.png','.jpg','youtube','video','watch','listen']
+    skip_kw = ['.mp3', '.wav', '.png', '.jpg', 'youtube', 'video', 'watch', 'listen']
     for item in questions:
         tid = item.get("task_id")
-        q = item.get("question","")
-        if not tid or not q or len(q)>MAX_QUESTION_LENGTH or any(k in q.lower() for k in skip_kw):
+        q = item.get("question", "")
+        if not tid or not q:
+            continue
+        if len(q) > MAX_QUESTION_LENGTH or any(k in q.lower() for k in skip_kw):
             continue
         ans = agent(q)
         payload.append({"task_id": tid, "submitted_answer": ans})
@@ -76,10 +80,15 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     if not payload:
         return "No valid questions to submit.", pd.DataFrame(logs)
 
-    sub = {"username": username, "agent_code": f"https://huggingface.co/spaces/{os.getenv('SPACE_ID')}/tree/main", "answers": payload}
+    sub = {
+        "username": username,
+        "agent_code": f"https://huggingface.co/spaces/{os.getenv('SPACE_ID')}/tree/main",
+        "answers": payload
+    }
     resp = requests.post(f"{DEFAULT_API_URL}/submit", json=sub, timeout=60)
     resp.raise_for_status()
     result = resp.json()
+
     status = f"Score: {result.get('score')}% ({result.get('correct_count')}/{result.get('total_attempted')})"
     return status, pd.DataFrame(logs)