Update app.py

app.py

@@ -1,7 +1,15 @@
 import gradio as gr
 from uuid import uuid4
-from fileManager import FileManager
+from fileManager import FileManager  # Assuming fileManager is defined elsewhere
 import re
+import os
+import random
+import string
+import glob
+
+##############################################################
+# File Upload and Storage Functionality
+##############################################################
 
 fileManager = FileManager()
 fileManager.start()
@@ -10,19 +18,11 @@ fileManager.start()
 def saveFile(file):
   with open(file.name, "rb") as rfile:
     file_infor = fileManager.saveFile(rfile.read(), file.orig_name)
-  return f"https://astraos-nx-storage.hf.space/file=./{file_infor['path']}"
-
-
+  return f"https://emee-nx-storage.hf.space/file=./{file_infor['path']}"
 
 
-import gradio as gr
-import os
-import random
-import string
-import glob
-
 ##############################################################
-# Generate some text files and save them in persistent storage
+# Random File Generation Functionality (Optional)
 ##############################################################
 
 def generate_random_string(length=100):
@@ -33,9 +33,9 @@ num_files=10
 file_length=1000
 
 
-for directory in ["/home/user/app", "/data"]:
+def generate_random_files(directory):
     if not os.path.exists(directory):
-        break
+        return
     
     for i in range(num_files):
         file_name = os.path.join(directory, f'random_file_{i}.txt')
@@ -45,12 +45,21 @@ for directory in ["/home/user/app", "/data"]:
 
 
 ##############################################################
-# The Gradio app
+# The Gradio App with Combined Functionality
 ##############################################################
 
-#with gr.Blocks() as demo:
-demo=[gr.Interface(fn=saveFile, inputs="file", outputs="text"),
-    gr.FileExplorer(label="Working directory"),
-    gr.FileExplorer(root="/home/user/app", label="Persistent storage")]
+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"]])
 
 demo.launch()