Update helper.py

helper.py

@@ -96,7 +96,7 @@ def get_travily_api_search_tool(tavily_api_key: str) -> Tool:
 import requests
 from langchain.tools import Tool
 import os
-
+'''
 def download_file_from_url(url: str, local_filename: str = None) -> str | None:
     """
     Downloads a file from a given URL and saves it locally.
@@ -169,7 +169,60 @@ file_download_tool = Tool(
     """,
     func=download_file_from_url,
 )
+'''
+
+
+def download_and_save_file(args: dict) -> str:
+    """
+    Downloads a file from a given URL and saves it to a specified local filename.
+    Useful for downloading binary files like spreadsheets, images, or archives that
+    need to be processed by other tools locally (e.g., python_repl).
+
+    **Input:** A JSON string with 'url' and 'local_filename' keys.
+    Example: {"url": "https://example.com/data.xlsx", "local_filename": "data.xlsx"}
+
+    **Output:** Returns a success message with the local filename, or an error message.
+    """
+    try:
+        # Ensure the input is parsed if it comes as a string (common from LLMs)
+        if isinstance(args, str):
+            import json
+            args = json.loads(args)
+
+        url = args.get("url")
+        local_filename = args.get("local_filename")
+
+        if not url or not local_filename:
+            return "Error: Both 'url' and 'local_filename' must be provided."
+
+        response = requests.get(url, stream=True, timeout=30)
+        response.raise_for_status() # Raise an HTTPError for bad responses (4xx or 5xx)
+
+        # Ensure the directory exists if local_filename includes a path
+        os.makedirs(os.path.dirname(local_filename) or '.', exist_ok=True)
 
+        with open(local_filename, 'wb') as f:
+            for chunk in response.iter_content(chunk_size=8192):
+                f.write(chunk)
+
+        return f"File downloaded successfully to {local_filename}"
+
+    except requests.exceptions.RequestException as e:
+        return f"Error downloading file from {url}: {e}"
+    except Exception as e:
+        return f"An unexpected error occurred: {e}"
+
+# Add this tool to your list of tools
+file_saver_tool = Tool(
+    name="file_saver",
+    description="""
+    Downloads a file from a URL and saves it to a specified local filename.
+    Input must be a JSON string with 'url' and 'local_filename' keys.
+    Use this for downloading binary files like Excel sheets (.xlsx), images, or PDFs.
+    Example: '{"url": "https://example.com/data.xlsx", "local_filename": "my_data.xlsx"}'
+    """,
+    func=download_and_save_file,
+)
 
 import speech_recognition as sr
 from pydub import AudioSegment