Update app.py

app.py

@@ -5,13 +5,14 @@ import requests
 import pandas as pd
 
 from smolagents import CodeAgent, OpenAIServerModel
-from smolagents.tools.web_search import WebSearchTool
+from smolagents.tools import WebSearchTool, BaseTools
 
 # Constants
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 MAX_QUESTION_LENGTH = 4000
 
-class RetryableWebSearchTool(WebSearchTool):
+# Reliable search tool with retry
+class ReliableWebSearch(WebSearchTool):
     def run(self, query: str) -> str:
         for attempt in range(3):
             try:
@@ -22,8 +23,9 @@ class RetryableWebSearchTool(WebSearchTool):
                     time.sleep(2 * (attempt + 1))
                 else:
                     raise
-        raise RuntimeError("Web search failed after retries")
+        raise RuntimeError("WebSearchTool failed after retries")
 
+# Main agent using GPT-4 & tools
 class SmartGAIAAgent:
     def __init__(self):
         key = os.getenv("OPENAI_API_KEY")
@@ -31,19 +33,20 @@ class SmartGAIAAgent:
             raise ValueError("Missing OPENAI_API_KEY")
         model = OpenAIServerModel(model_id="gpt-4", api_key=key)
         self.agent = CodeAgent(
+            tools=[ReliableWebSearch()],
             model=model,
-            tools=[RetryableWebSearchTool()],
             add_base_tools=True
         )
 
     def __call__(self, question: str) -> str:
-        question = question[:MAX_QUESTION_LENGTH]
+        q = question[:MAX_QUESTION_LENGTH]
         try:
-            return self.agent.run(question).strip()
+            return self.agent.run(q).strip()
         except Exception as e:
             print("Agent error:", e)
             return "error"
 
+# Fetch, filter, run, and submit answers
 def run_and_submit_all(profile: gr.OAuthProfile | None):
     username = profile.username if profile else None
     if not username:
@@ -54,34 +57,30 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     except Exception as e:
         return f"Error initializing agent: {e}", None
 
-    response = requests.get(f"{DEFAULT_API_URL}/questions", timeout=15)
-    response.raise_for_status()
-    questions = response.json()
-
-    payload = []
-    logs = []
+    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']
     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:
+        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):
             continue
-        if any(skip in q.lower() for skip in ['.mp3', '.wav', '.jpg', '.png', 'youtube', 'video', 'watch', 'listen']):
-            continue
-        answer = agent(q)
-        payload.append({"task_id": tid, "submitted_answer": answer})
-        logs.append({"Task ID": tid, "Question": q, "Submitted Answer": answer})
+        ans = agent(q)
+        payload.append({"task_id": tid, "submitted_answer": ans})
+        logs.append({"Task ID": tid, "Question": q, "Submitted Answer": ans})
 
     if not payload:
         return "No valid questions to submit.", pd.DataFrame(logs)
 
-    submission = {
+    sub = {
         "username": username,
         "agent_code": f"https://huggingface.co/spaces/{os.getenv('SPACE_ID')}/tree/main",
-        "answers": payload,
+        "answers": payload
     }
-
-    resp = requests.post(f"{DEFAULT_API_URL}/submit", json=submission, timeout=30)
+    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')})"