Update app.py

app.py

@@ -3,6 +3,9 @@ import os
 
 DOWNLOADS_PATH = '/tmp/downloads'  # Adjust if necessary
 
+# Ensure the downloads path exists
+os.makedirs(DOWNLOADS_PATH, exist_ok=True)
+
 def list_directories():
     return [d for d in os.listdir(DOWNLOADS_PATH) if os.path.isdir(os.path.join(DOWNLOADS_PATH, d))]
 
@@ -10,10 +13,12 @@ def upload_file(file, directory):
     if not file:
         return 'No file selected'
     
-    if directory not in list_directories():
+    directories = list_directories()
+    if directory not in directories:
         return 'Invalid directory selected'
     
     filepath = os.path.join(DOWNLOADS_PATH, directory, file.name)
+    os.makedirs(os.path.dirname(filepath), exist_ok=True)  # Ensure the directory exists
     with open(filepath, 'wb') as f:
         f.write(file.read())
     
@@ -34,6 +39,12 @@ with gr.Blocks() as demo:
             upload_btn = gr.Button("Upload")
             upload_status = gr.Textbox(label="Status")
 
+    # Update dropdown choices when interface is loaded
+    def update_directories():
+        return list_directories()
+    
+    directory_dropdown.update(choices=update_directories())
+    
     # Set up the button click action
     upload_btn.click(
         fn=upload_file,
@@ -42,4 +53,4 @@ with gr.Blocks() as demo:
     )
 
 # Launch Gradio app
-demo.launch(server_name="0.0.0.0",share=True, server_port=7860, debug=True)
+demo.launch(server_name="0.0.0.0", share=True,server_port=7860, debug=True)