Fixing file upload

app.py

@@ -24,12 +24,12 @@ with gr.Blocks(title="Drone Crash Detection") as demo:
     )
     simulation_videos_upload = gr.File(
         label="Upload Simulation Drone Videos", 
-        type="file", 
+        type="filepath",  # Changed 'file' to 'filepath'
         file_count="multiple"
     )
     real_videos_upload = gr.File(
         label="Upload Real Drone Videos", 
-        type="file", 
+        type="filepath",  # Changed 'file' to 'filepath'
         file_count="multiple"
     )
     with gr.Row():

app_backend.py

@@ -1,5 +1,6 @@
 import os
 import csv
+import shutil  # Added to copy files
 from dataclasses import dataclass
 import model.video_classifier as classifier
 
@@ -8,6 +9,10 @@ model_path = './model/best.pt'
 video_folder = './videos'
 output_folder = './results'
 
+# Ensure the directories exist
+os.makedirs(video_folder, exist_ok=True)
+os.makedirs(output_folder, exist_ok=True)
+
 model = classifier.get_model(model_path)
 
 @dataclass
@@ -90,24 +95,27 @@ def backend(student_name_tb, simulation_files, real_files, frame_output, text_ou
     simulation_paths = []
     real_paths = []
 
-    # Save the uploaded simulation videos to disk
-    for i, file in enumerate(simulation_files, start=1):
-        sim_video_path = os.path.join(video_folder, f'sim_video{i}.mp4')
-        with open(sim_video_path, 'wb') as f:
-            f.write(file.read())
+    # Ensure the video_folder exists
+    os.makedirs(video_folder, exist_ok=True)
+
+    # Store original filenames
+    simulation_filenames = [os.path.basename(file_path) for file_path in simulation_files]
+    real_filenames = [os.path.basename(file_path) for file_path in real_files]
+
+    # Save the uploaded simulation videos to the 'video_folder'
+    for i, file_path in enumerate(simulation_files, start=1):
+        ext = os.path.splitext(file_path)[1]
+        sim_video_path = os.path.join(video_folder, f'sim_video{i}{ext}')
+        shutil.copy(file_path, sim_video_path)
         simulation_paths.append(sim_video_path)
 
-    # Save the uploaded real videos to disk
-    for i, file in enumerate(real_files, start=1):
-        real_video_path = os.path.join(video_folder, f'real_video{i}.mp4')
-        with open(real_video_path, 'wb') as f:
-            f.write(file.read())
+    # Save the uploaded real videos to the 'video_folder'
+    for i, file_path in enumerate(real_files, start=1):
+        ext = os.path.splitext(file_path)[1]
+        real_video_path = os.path.join(video_folder, f'real_video{i}{ext}')
+        shutil.copy(file_path, real_video_path)
         real_paths.append(real_video_path)
 
-    # Store original filenames
-    simulation_filenames = [file.name for file in simulation_files]
-    real_filenames = [file.name for file in real_files]
-
     simulation_results = []
     real_results = []
     output_video_paths = []