Update app.py

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 from uuid import uuid4
-from fileManager import FileManager  # Assuming fileManager is defined elsewhere
+from fileManager import FileManager
 import re
 import os
 import random
@@ -8,7 +8,7 @@ import string
 import glob
 
 ##############################################################
-# File Upload and Storage Functionality
+# File Manager Integration
 ##############################################################
 
 fileManager = FileManager()
@@ -22,7 +22,7 @@ def saveFile(file):
 
 
 ##############################################################
-# Random File Generation Functionality (Optional)
+# Random Text File Generation Function
 ##############################################################
 
 def generate_random_string(length=100):
@@ -33,10 +33,10 @@ num_files=10
 file_length=1000
 
 
-def generate_random_files(directory):
+def generate_text_files(directory):
     if not os.path.exists(directory):
         return
-    
+
     for i in range(num_files):
         file_name = os.path.join(directory, f'random_file_{i}.txt')
         with open(file_name, 'w') as f:
@@ -45,21 +45,25 @@ def generate_random_files(directory):
 
 
 ##############################################################
-# The Gradio App with Combined Functionality
+# The Gradio App with File Management and Text Generation
 ##############################################################
 
 with gr.Blocks() as demo:
-    # File Upload for Saving
-    with gr.Column():
-        uploaded_file = gr.File(label="Upload File")
-        save_button = gr.Button(value="Save File")
-        output_text = gr.Textbox(label="File URL")
-
-    # Random File Generation (Optional)
-    with gr.Column():
-        generate_button = gr.Button(value="Generate Random Files")
-
-    save_button.click(lambda: output_text.setValue(saveFile(uploaded_file)) if uploaded_file else None)
-    generate_button.click(lambda: [generate_random_files(d) for d in ["/home/user/app", "/data"]])
+
+    # File Explorer for Upload (using fileManager integration)
+    gr.File(label="Upload File", upload=True, fn=saveFile)
+
+    # File Explorer for Existing Files (persistent storage)
+    working_dir = gr.FileExplorer(label="Working directory")
+
+    # Generate Random Text Files Button
+    generate_button = gr.Button("Generate Random Text Files")
+
+    @generate_button.click
+    def generate_files(event):
+        # Generate text files in working directory chosen by user
+        generate_text_files(working_dir.value)
+
+    gr.FileExplorer(root=working_dir.value, label="Persistent storage")
 
 demo.launch()