Update app.py

app.py

@@ -5,34 +5,34 @@ import requests
 import pandas as pd
 
 from smolagents import CodeAgent, OpenAIServerModel
-from langchain_community.tools import DuckDuckGoSearchRun
+from smolagents.tools import DuckDuckGoSearchTool
 
 # Constants
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 MAX_QUESTION_LENGTH = 4000
-MAX_WEBPAGE_CONTENT = 3000
 
-# --- Reliable DuckDuckGo Tool with Retry ---
-class ReliableDuckDuckGoTool(DuckDuckGoSearchRun):
-    def _run(self, query: str) -> str:
+# --- Reliable DuckDuckGo Tool using smolagents ---
+class ReliableDuckDuckGoTool(DuckDuckGoSearchTool):
+    def run(self, query: str) -> str:
         for attempt in range(3):
             try:
-                return super()._run(query)
+                return super().run(query)
             except Exception as e:
-                if "ratelimit" in str(e).lower() or "202" in str(e):
-                    print(f"Rate limited. Retry {attempt + 1}/3...")
-                    time.sleep(3 * (attempt + 1))
+                if "rate" in str(e).lower():
+                    print(f"[DuckDuckGo] Rate limit hit. Retrying ({attempt + 1}/3)...")
+                    time.sleep(2 * (attempt + 1))
                 else:
                     raise e
         raise RuntimeError("DuckDuckGo search failed after retries")
 
-# --- Smart GAIA Agent ---
+# --- Main Agent ---
 class SmartGAIAAgent:
     def __init__(self):
         self.api_key = os.getenv("OPENAI_API_KEY")
         if not self.api_key:
             raise ValueError("Missing OPENAI_API_KEY")
         self.model = OpenAIServerModel(model_id="gpt-4", api_key=self.api_key)
+
         self.agent = CodeAgent(
             tools=[ReliableDuckDuckGoTool()],
             model=self.model,
@@ -44,14 +44,13 @@ class SmartGAIAAgent:
 
     def __call__(self, question: str) -> str:
         try:
-            clean_question = self.truncate_question(question)
-            result = self.agent.run(clean_question)
-            return result.strip()
+            question = self.truncate_question(question)
+            return self.agent.run(question).strip()
         except Exception as e:
             print(f"Agent error: {e}")
             return "error"
 
-# --- Evaluation + Submission ---
+# --- Evaluation and Submission Logic ---
 def run_and_submit_all(profile: gr.OAuthProfile | None):
     space_id = os.getenv("SPACE_ID")
     if profile:
@@ -93,7 +92,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
             'attached', '.mp3', '.wav', '.png', '.jpg', '.jpeg',
             'youtube', '.mp4', 'video', 'listen', 'watch'
         ]):
-            print(f"Skipping unsupported media question: {task_id}")
+            print(f"Skipping unsupported question: {task_id}")
             continue
 
         try:
@@ -125,17 +124,17 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
         response.raise_for_status()
         result_data = response.json()
         final_status = (
-            f"Submission Successful!\\n"
-            f"User: {result_data.get('username')}\\n"
+            f"Submission Successful!\n"
+            f"User: {result_data.get('username')}\n"
             f"Score: {result_data.get('score')}% "
-            f"({result_data.get('correct_count')}/{result_data.get('total_attempted')})\\n"
+            f"({result_data.get('correct_count')}/{result_data.get('total_attempted')})\n"
             f"Message: {result_data.get('message')}"
         )
         return final_status, pd.DataFrame(results_log)
     except Exception as e:
         return f"Submission failed: {e}", pd.DataFrame(results_log)
 
-# --- Gradio Interface ---
+# --- Gradio UI ---
 with gr.Blocks() as demo:
     gr.Markdown("# 🧠 GAIA Agent Evaluation")
     gr.Markdown("""