Update app.py

app.py

@@ -138,12 +138,23 @@ def write_file(file_path, content):
     except:
         return "Error occurred while writing to file."
 
-def download_link(file_path):
-    with open(file_path, "rb") as file:
-        contents = file.read()
-    b64 = base64.b64encode(contents).decode()
-    href = f'<a href="data:application/octet-stream;base64,{b64}" download="{file_path}">Download</a>'
-    return href
+# def download_link(file_path):
+#    with open(file_path, "rb") as file:
+#        contents = file.read()
+#    b64 = base64.b64encode(contents).decode()
+#    href = f'<a href="data:application/octet-stream;base64,{b64}" download="{file_path}">Download</a>'
+#    return href
+
+def download_file(file_path, content):
+    # Get the user's desktop directory
+    desktop_dir = os.path.join(os.path.join(os.path.expanduser('~')), 'Desktop')
+
+    # Write the input to a new file on the desktop
+    with open(os.path.join(desktop_dir, file_path), "w") as f:
+        f.write(content)
+
+    return "File saved to desktop!"
+
 
 # Function to append to a file
 def append_file(file_path, content):
@@ -210,7 +221,7 @@ with gr.Blocks(css = """#col_container {width: 1400px; margin-left: auto; margin
     listFiles.click(list_files, inputs=fileName, outputs=fileContent)
     readFile.click(read_file, inputs=fileName, outputs=fileContent)
     saveFile.click(write_file, inputs=[fileName, fileContent], outputs=completedMessage)
-    downloadFile.click(download_link, inputs=[fileName], outputs=completedMessage)
+    downloadFile.click(download_file, inputs=[fileName, fileContent], outputs=completedMessage)
     deleteFile.click(delete_file, inputs=fileName, outputs=completedMessage)
     appendFile.click(append_file, inputs=[fileName, fileContent], outputs=completedMessage )