Update app.py

app.py

@@ -130,10 +130,21 @@ def write_file(file_path, content):
     try:
         with open(file_path, "w") as file:
             file.write(content)
-        return f"Successfully written to {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 f"Successfully written to {file_path}.", href
     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
+
 # Function to append to a file
 def append_file(file_path, content):
     try:
@@ -187,12 +198,14 @@ with gr.Blocks(css = """#col_container {width: 1400px; margin-left: auto; margin
     fileContent = gr.TextArea(label="File Content")
     completedMessage = gr.Textbox(label="Completed")
     label = gr.Label()
+    
     with gr.Row():
         listFiles = gr.Button("📄 List File(s)")
         readFile = gr.Button("📖 Read File")
         saveFile = gr.Button("💾 Save File")
         deleteFile = gr.Button("🗑️ Delete File")
         appendFile = gr.Button("➕ Append File")
+        
     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)