Update app.py

app.py

@@ -36,6 +36,16 @@ class NewAgent:
     def __call__(self, question: str, file_name: str = "", task_id: str = "") -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
         
+        # Auto-download file if file_name and task_id are provided
+        local_file_path = ""
+        if file_name and task_id:
+            print(f"Auto-downloading file: {file_name} for task: {task_id}")
+            local_file_path = download_file(task_id, file_name)
+            if local_file_path.startswith("Error"):
+                print(f"File download failed: {local_file_path}")
+            else:
+                print(f"File downloaded successfully to: {local_file_path}")
+        
         # Initialize the web search tool
         search_tool = DuckDuckGoSearchRun()
         # Initialize the Hub stats tool
@@ -56,20 +66,12 @@ class NewAgent:
             func=read_excel_file,
             description="Reads an Excel file and returns structured information about its contents."
         )
-        # Initialize the Download File tool
-        download_file_tool = Tool(
-            name="download_file",
-            func=download_file,
-            description="Downloads a file associated with a task_id. Input should be 'task_id,filename'"
-        )
         # Generate the chat interface, including the tools
         tools = [
             search_tool,
-            # weather_info_tool,
             hub_stats_tool,
             image_analysis_tool,
             read_excel_tool,
-            download_file_tool,
         ]
         # llm = ChatOpenAI(model="gpt-4o")
         llm = ChatOpenAI(model="gpt-4.1")
@@ -92,7 +94,9 @@ class NewAgent:
                 If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise.
                 If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string.
                 Current task_id: {task_id}
-                Current file (if any): {file_name}"""
+                Current file (if any): {file_name}
+                Downloaded file path: {local_file_path}
+                """
                                    )
             return {
                 "messages": [llm_with_tools.invoke([sys_msg] + state["messages"])],