Update app.py

app.py

@@ -11,8 +11,7 @@ from smolagents.tools.web_search import WebSearchTool
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 MAX_QUESTION_LENGTH = 4000
 
-# Retryable search tool
-class ReliableWebSearchTool(WebSearchTool):
+class RetryableWebSearchTool(WebSearchTool):
     def run(self, query: str) -> str:
         for attempt in range(3):
             try:
@@ -25,7 +24,6 @@ class ReliableWebSearchTool(WebSearchTool):
                     raise
         raise RuntimeError("Web search failed after retries")
 
-# Agent definition
 class SmartGAIAAgent:
     def __init__(self):
         key = os.getenv("OPENAI_API_KEY")
@@ -33,20 +31,19 @@ class SmartGAIAAgent:
             raise ValueError("Missing OPENAI_API_KEY")
         model = OpenAIServerModel(model_id="gpt-4", api_key=key)
         self.agent = CodeAgent(
-            tools=[ReliableWebSearchTool()],
             model=model,
+            tools=[RetryableWebSearchTool()],
             add_base_tools=True
         )
 
     def __call__(self, question: str) -> str:
-        q = question[:MAX_QUESTION_LENGTH]
+        question = question[:MAX_QUESTION_LENGTH]
         try:
-            return self.agent.run(q).strip()
+            return self.agent.run(question).strip()
         except Exception as e:
             print("Agent error:", e)
             return "error"
 
-# Submission logic
 def run_and_submit_all(profile: gr.OAuthProfile | None):
     username = profile.username if profile else None
     if not username:
@@ -57,36 +54,40 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     except Exception as e:
         return f"Error initializing agent: {e}", None
 
-    resp = requests.get(f"{DEFAULT_API_URL}/questions", timeout=15)
-    resp.raise_for_status()
-    questions = resp.json()
+    response = requests.get(f"{DEFAULT_API_URL}/questions", timeout=15)
+    response.raise_for_status()
+    questions = response.json()
+
+    payload = []
+    logs = []
 
-    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 or any(k in q.lower() for k in skip_kw):
+        if not tid or not q or len(q) > MAX_QUESTION_LENGTH:
             continue
-        ans = agent(q)
-        payload.append({"task_id": tid, "submitted_answer": ans})
-        logs.append({"Task ID": tid, "Question": q, "Submitted Answer": ans})
+        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})
 
     if not payload:
         return "No valid questions to submit.", pd.DataFrame(logs)
 
-    sub = {
+    submission = {
         "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=sub, timeout=60)
+
+    resp = requests.post(f"{DEFAULT_API_URL}/submit", json=submission, timeout=30)
     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)
 
-# UI
+# Gradio UI
 with gr.Blocks() as demo:
     gr.Markdown("# GAIA Agent")
     gr.LoginButton()