Update multi_video_app.py

multi_video_app.py

@@ -9,7 +9,7 @@ import torch
 from src.video_model import describe_video 
 from src.utils import parse_string, parse_annotations
 
-# Utility functions (from your provided utilities)
+
 
 # Function to save data to a CSV file
 def save_to_csv(observations: List[Dict], output_dir: str = "outputs") -> str:
@@ -77,9 +77,8 @@ def process_single_video(video_path: str, sitting, hands, location, screen) -> D
         return {"error": f"An error occurred with {video_name}: {e}"}
 
 # Function to process all videos in a folder
-def process_folder_of_videos(videos_folder: str, sitting, hands, location, screen):
+def process_multiple_videos(video_files: List[str], sitting, hands, location, screen):
     all_observations = []
-    video_files = [os.path.join(videos_folder, f) for f in os.listdir(videos_folder) if f.endswith(('.mp4', '.avi', '.mkv'))]
 
     for video_path in video_files:
         observation = process_single_video(video_path, sitting, hands, location, screen)
@@ -96,11 +95,11 @@ def process_folder_of_videos(videos_folder: str, sitting, hands, location, scree
     return "Processing completed. Download the CSV file.", csv_file
 
 # Gradio interface
-def gradio_interface(videos_folder, sitting, hands, location, screen):
-    return process_folder_of_videos(videos_folder, sitting, hands, location, screen)
+def gradio_interface(video_files, sitting, hands, location, screen):
+    return process_multiple_videos(video_files, sitting, hands, location, screen)
 
 # Inputs
-videos_folder = gr.Directory(label="Upload a folder of videos")
+video_files = gr.File(file_count="multiple", file_types=["video"], label="Upload multiple videos")
 sitting = gr.Checkbox(label="Sitting/Standing")
 hands = gr.Checkbox(label="Hands Free/Not Free")
 location = gr.Checkbox(label="Indoors/Outdoors")
@@ -113,10 +112,10 @@ download_link = gr.File(label="Download CSV")
 # Gradio interface setup
 interface = gr.Interface(
     fn=gradio_interface,
-    inputs=[videos_folder, sitting, hands, location, screen],
+    inputs=[video_files, sitting, hands, location, screen],
     outputs=[response, download_link],
     title="Batch Video Annotation",
-    description="Upload a folder of videos and process them sequentially, saving the results to a downloadable CSV file.",
+    description="Upload multiple videos and process them sequentially, saving the results to a downloadable CSV file.",
     theme=gr.themes.Soft(primary_hue="red", secondary_hue="red"),
     allow_flagging="never"
 )