Update app.py

added file extension and debug

app.py

@@ -67,6 +67,10 @@ class FetchFileTool(Tool):
             "type": "string",
             "description": "The internet location of the file to fetch",
         },
+        "file_extension": {
+            "type": "string",
+            "description": "Assumed file extension of the file to fetch, like mp3 or png for example",
+        },
         "file_path": {
             "type": "string",
             "description": "The path to the directory where the file will be saved",
@@ -74,11 +78,11 @@ class FetchFileTool(Tool):
         "file_name": {
             "type": "string",
             "description": "The name of the file where content will be saved",
-        },
+        }
     }
     output_type = "boolean"
     
-    def forward(self, location: str, file_path: str, file_name: str):
+    def forward(self, location: str, file_extension: str, file_path: str, file_name: str):
         try:
             response = requests.get(location)
             response.raise_for_status()
@@ -93,9 +97,13 @@ class FetchFileTool(Tool):
             if response.status_code == 429:
                 # get last part of location separated by slash
                 file_id = location.split('/')[-1]
-                local_path = f"files/{file_id}"
+                local_path = f"files/{file_id}.{file_extension}"
+                print(f"Copying from: {local_path}")
+                
                 full_path = os.path.join(file_path, file_name)
-                shutil.copy(local_path, full_path) 
+                print(f"Copying to: {full_path}")
+
+                shutil.copy(local_path, full_path)
                 return True
             else:
                 print(f"Error saving content to file: {e}")