Update app.py

app.py

@@ -7,6 +7,7 @@ from datetime import datetime
 import pandas as pd
 from filelock import FileLock, Timeout
 from pathlib import Path
+import tempfile
 
 
 class VideoArenaManager:
@@ -33,12 +34,9 @@ class VideoArenaManager:
         """Retrieve a list of data files in the /data directory."""
         data_path = Path("/data")
         files = [
-            {
-                "name": file.name,
-                "size": file.stat().st_size,
-                "path": str(file)
-            }
-            for file in data_path.glob("*") if file.is_file()
+            {"name": file.name, "size": file.stat().st_size, "path": str(file)}
+            for file in data_path.glob("*")
+            if file.is_file()
         ]
         return files
 
@@ -261,24 +259,29 @@ def create_arena_interface():
         with gr.Accordion("Download Results File", open=False):
             files = manager.get_data_files()
             file_names = [file["name"] for file in files]
-            file_select = gr.Dropdown(
-                choices=file_names, 
-                label="Select Results File to Download",
-                interactive=True
-            )
+            file_select = gr.Dropdown(choices=file_names, label="Select Results File to Download", interactive=True)
             download_button = gr.Button("Download Selected File", size="sm")  # Smaller button
 
         def download_file(file_name: str):
             """Prepare the selected file for download."""
+            if not file_name:
+                return None
+
             file_path = f"/data/{file_name}"
-            with open(file_path, "rb") as f:
-                file_content = f.read()
-            return {'name': file_name, 'data': file_content}
-        
+            try:
+                # Create a temporary file to handle the download
+                with tempfile.NamedTemporaryFile(delete=False, suffix=os.path.splitext(file_name)[1]) as tmp_file:
+                    with open(file_path, "rb") as f:
+                        tmp_file.write(f.read())
+                    return tmp_file.name
+            except Exception as e:
+                print(f"Error preparing download: {e}")
+                return None
+
         download_button.click(
             fn=download_file,
             inputs=[file_select],
-            outputs=gr.File(),
+            outputs=gr.File(label="Download"),
         )
 
     return demo