Update new_agent.py

new_agent.py

@@ -8,7 +8,7 @@ from smolagents import (
     InferenceClientModel,
     CodeAgent,
     VisitWebpageTool,
-    WebSearchTool,
+    DuckDuckGoSearchTool,
     WikipediaSearchTool,
     tool,
 )
@@ -21,22 +21,6 @@ token = os.environ["HF_KEY"]
 model_id = "meta-llama/Llama-3.3-70B-Instruct"
 
 
-@tool
-def arxiv_search(query: str) -> str:
-    """Search Arxiv for a query and return maximum 3 result.
-
-    Args:
-        query: The search query."""
-    search_docs = ArxivLoader(query=query, load_max_docs=3).load()
-    formatted_search_docs = "\n\n---\n\n".join(
-        [
-            f'<Document source="{doc.metadata["source"]}" page="{doc.metadata.get("page", "")}"/>\n{doc.page_content[:1000]}\n</Document>'
-            for doc in search_docs
-        ]
-    )
-    return {"arxiv_results": formatted_search_docs}
-
-
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
@@ -50,9 +34,8 @@ class BasicAgent:
         self.agent = CodeAgent(
             tools=[
                 VisitWebpageTool(),
-                WebSearchTool(),
+                DuckDuckGoSearchTool(),
                 WikipediaSearchTool(),
-                arxiv_search,
             ],
             model=model,
             additional_authorized_imports=[
@@ -88,10 +71,10 @@ class BasicAgent:
             try:
                 if file_path.endswith(".xlsx"):
                     df = pd.read_excel(file_path)
-                    file_info = f"\n[Attached Excel file preview (first 10 rows)]:\n{df.head(10).to_string(index=False)}\n"
+                    file_info = f"\n[Attached Excel file preview]:\n{df.head(500).to_string(index=False)}\n"
                 elif file_path.endswith(".py"):
                     with open(file_path, "r", encoding="utf-8") as f:
-                        code = "".join([line for _, line in zip(range(400), f)])
+                        code = "".join([line for _, line in zip(range(500), f)])
                     file_info = f"\n[Attached Python file]:\n{code}\n"
                 elif file_path.endswith(".png"):
                     with open(file_path, "rb") as img_file:
@@ -172,14 +155,14 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     for item in questions_data:
         task_id = item.get("task_id")
         question_text = item.get("question")
-        file_name = item.get("file")  # Check for file key
+        file_name = item.get("file_name")  # Updated key
         file_path = None
 
         if not task_id or question_text is None:
             print(f"Skipping item with missing task_id or question: {item}")
             continue
 
-        # Download file if present
+        # Download file if present and file_name is not empty
         if file_name:
             files_url = f"{api_url}/files/{task_id}"
             print(f"Downloading file for task {task_id} from {files_url}")