Fix: File download

app.py

@@ -255,23 +255,24 @@ def get_audio_duration(file_path):
     return audio.info.length  # Duration in seconds
 
 def get_storage(password):
-    #source: https://discuss.huggingface.co/t/accessing-data-folder-of-persistent-storage/46681/2
+    # Check if the password is correct
     if password == password_files:
+        # Get the list of file paths and calculate the total usage
         files = [
-            {
-                "orig_name": file.name,
-                "name": file.resolve(),
-                "size": file.stat().st_size,
-                "data": None,
-                "is_file": True,
-            }
-            for file in persistent_storage.glob("**/*.csv")
-            if file.is_file()
+            file for file in persistent_storage.glob("**/*.csv") if file.is_file()
         ]
-        usage = sum([f['size'] for f in files])
+        
+        # Calculate total usage (in bytes)
+        usage = sum([file.stat().st_size for file in files])
+        
+        # Convert file paths to strings for Gradio's File component
+        file_paths = [str(file.resolve()) for file in files]
+        
+        # Return the file paths (as strings) and the total usage in GB
+        return file_paths, f"{usage / (1024.0 ** 3):.3f}GB"
+    
     else:
-        gr.Warning("Please provide the correct password")
-    return files, f"{usage/(1024.0 ** 3):.3f}GB"
+        return gr.Warning("Please provide the correct password"), None
 
 
 def load_first_example(participant_id, ann_completed, current_index):
@@ -495,15 +496,17 @@ with (gr.Blocks(theme=gr.themes.Soft(), css = css) as demo):
         audio_player.play(count_clicks, [n_clicks], [n_clicks])
     
     with gr.Tab("Access Files"):
+
         with gr.Row():
-            with gr.Column():
-                password = gr.Textbox(label='Password', interactive = True)
-                get_files_button = gr.Button("Get Files")
-
-            with gr.Column():
-                files = gr.Files(label="Files")
-                storage = gr.Text(label="Total Usage")
-        get_files_button.click(get_storage, inputs= [password], outputs=[files, storage], postprocess=False)
+            password = gr.Textbox(label="Enter Password", type="password")
+            get_files_button = gr.Button("Get Files")
+        
+        with gr.Column():
+            files = gr.File(label="Download Files", file_count="multiple", interactive=True)
+            storage = gr.Text(label="Total Usage")
+        
+        get_files_button.click(fn=get_storage, inputs=[password], outputs=[files, storage])
+
 
         
 demo.launch(allowed_paths = ['/data'])