Update tools.py

added logic to handle different file types

tools.py

@@ -1,5 +1,7 @@
 from smolagents import tool, DuckDuckGoSearchTool, VisitWebpageTool
 import os
+import pandas as pd
+from pypdf import PdfReader
 
 # Instantiate built-in tools
 search_tool = DuckDuckGoSearchTool()
@@ -7,22 +9,40 @@ visit_webpage = VisitWebpageTool()
 
 # Custom tool for GAIA files
 @tool
-def hanlde_file(file_path: str) -> str:
+def handle_file(file_path: str) -> str:
     """
-    A tool that reads the content of a file provided in a GAIA task.
-    Supports .txt files. 
-
+    This tool extracts content from different file types (PDF, Excel, CSV, TXT).
+    
     Args:
-        file_path: The local path to the file to be read.
+        file_path: The local path to the file.
     """
     if not os.path.exists(file_path):
-        return f"Error: FIle {file_path} not found"
-
+        return f"Error: File {file_path} not found."
+    
+    ext = os.path.splitext(file_path)[1].lower()
+    
     try:
-        with open(file_path, 'r', encoding='utf-8') as f:
-            return f.read()
+        # Handle Excel
+        if ext in ['.xlsx', '.xls', '.csv']:
+            df = pd.read_csv(file_path) if ext == '.csv' else pd.read_excel(file_path)
+            # We return a markdown version of the head and info to save tokens
+            return f"Dataframe Summary:\n{df.head(10).to_markdown()}\n\nFull shape: {df.shape}"
+            
+        # Handle PDFs
+        elif ext == '.pdf':
+            reader = PdfReader(file_path)
+            text = ""
+            for page in reader.pages:
+                text += page.extract_text() + "\n"
+            return text[:10000] # Cap text to avoid context window issues
+            
+        # Handle Text files
+        else:
+            with open(file_path, 'r', encoding='utf-8') as f:
+                return f.read()
+                
     except Exception as e:
-        return f"Error reading file: {str(e)}"
+        return f"Error processing {ext} file: {str(e)}"
 
 # all tools in a list ready for export
 all_tools = [search_tool, visit_webpage, hanlde_file]