Removed YouTube Downloader

Dockerfile

@@ -21,8 +21,6 @@ RUN chmod -R a+r /app
 COPY . /app
 
 RUN pip install gradio
-RUN pip install yt-dlp
-RUN pip install pytube
    
 EXPOSE 7860
 ENV PYTHONUNBUFFERED=1 \

app.py

@@ -3,8 +3,8 @@ import app_backend
 
 gr.set_static_paths(['.'])
 
-def detect_crashes(name, sim_paths, real_path, frame_out, text_output, file_out, video_out):
-    for outputs in app_backend.backend(name, sim_paths, real_path, frame_out, text_output, file_out, video_out):
+def detect_crashes(name, sim_files, real_files, frame_out, text_output, file_out, video_out):
+    for outputs in app_backend.backend(name, sim_files, real_files, frame_out, text_output, file_out, video_out):
         yield outputs
 
 with gr.Blocks(title="Drone Crash Detection") as demo:
@@ -13,18 +13,39 @@ with gr.Blocks(title="Drone Crash Detection") as demo:
     video_output = gr.Gallery(interactive=False)
     file_output = gr.File(interactive=False)
 
-    label = gr.Label("Drone Crash Classification Using Yolov8.", scale=0,show_label=False)
+    label = gr.Label("Drone Crash Classification Using Yolov8.", scale=0, show_label=False)
 
-    student_name_tb = gr.Textbox(lines=1, placeholder="Enter student's name. Ex: \'First Last\'", visible=True, show_label=False)
-    simulation_videos_tb = gr.Textbox(lines=1, placeholder="Enter simulation flying videos. Ex: \'https://youtu.be/Gd12g9XqSo4,https://youtu.be/Vzj78FsgQZg,...,https://youtu.be/gf7cp6_XFvQ\'", visible=True, show_label=False)
-    real_videos_tb = gr.Textbox(lines=1, placeholder="Enter real flying videos. Ex: \'https://youtu.be/Gd12g9XqSo4,https://youtu.be/Vzj78FsgQZg,...,https://youtu.be/gf7cp6_XFvQ\'", visible=True, show_label=False)
+    student_name_tb = gr.Textbox(
+        lines=1, 
+        placeholder="Enter student's name. Ex: 'First Last'", 
+        visible=True, 
+        show_label=True, 
+        label="Student's Name"
+    )
+    simulation_videos_upload = gr.File(
+        label="Upload Simulation Drone Videos", 
+        type="file", 
+        file_count="multiple"
+    )
+    real_videos_upload = gr.File(
+        label="Upload Real Drone Videos", 
+        type="file", 
+        file_count="multiple"
+    )
     with gr.Row():
         btn_prompt = gr.Button("Detect Crashes")
         btn_prompt.click(
             fn=detect_crashes,
-            inputs=[student_name_tb, simulation_videos_tb, real_videos_tb, frame_output, text_output, file_output, video_output],
+            inputs=[
+                student_name_tb, 
+                simulation_videos_upload, 
+                real_videos_upload, 
+                frame_output, 
+                text_output, 
+                file_output, 
+                video_output
+            ],
             outputs=[frame_output, text_output, file_output, video_output]
         )
 
-
-demo.launch()
+demo.launch()

app_backend.py

@@ -1,9 +1,7 @@
-import model.video_classifier as classifier
 import os
 import csv
-from yt_dlp import YoutubeDL
 from dataclasses import dataclass
-import pytube
+import model.video_classifier as classifier
 
 device = 'cpu'
 model_path = './model/best.pt'
@@ -17,57 +15,15 @@ class VideoResult:
     video_path: str
     label_counts: dict
     crash_events: list
-    video_link: str
+    video_name: str
     is_simulation: int
 
-def parse_inputs(student_name_tb, simulation_videos_tb, real_videos_tb):
-    simulation_video_list = [url.strip() for url in simulation_videos_tb.split(',') if url.strip()]
-    real_video_list = [url.strip() for url in real_videos_tb.split(',') if url.strip()]
-    return student_name_tb.strip(), simulation_video_list, real_video_list
-
-def download_youtube_video(youtube_url,rename):
-    try:
-        yt = pytube.YouTube(youtube_url)
-        video = yt.streams.filter(progressive=True, file_extension='mp4').order_by('resolution').desc().first()
-        video.download(f'{video_folder}')
-        original_filename = video.default_filename
-        os.rename(original_filename, rename)
-        return new_filename
-
-    except Exception as e:
-        print("An error occurred:", e)
-        return None
-
-
-def download_videos(simulation_links, real_links):
-    simulation_paths = []
-    real_paths = []
-    for i, url in enumerate(simulation_links, start=1):
-        #ydl_opts = {
-        #    'format': 'best',
-        #    'outtmpl': f'{video_folder}/sim_video{i}.%(ext)s'
-        #}
-        #with YoutubeDL(ydl_opts) as ydl:
-        #    ydl.download([url])
-        download_youtube_video(url,f"sim_video{i}.mp4")
-        simulation_paths.append(f'{video_folder}/sim_video{i}.mp4')
-    for i, url in enumerate(real_links, start=1):
-        #ydl_opts = {
-        #    'format': 'best',
-        #    'outtmpl': f'{video_folder}/real_video{i}.%(ext)s'
-        #}
-        #with YoutubeDL(ydl_opts) as ydl:
-        #    ydl.download([url])
-        download_youtube_video(url,f"real_video{i}.mp4")
-        real_paths.append(f'{video_folder}/real_video{i}.mp4')
-    return simulation_paths, real_paths
-
 def result_csv(student, simulation_results, real_results):
     results = simulation_results + real_results
     file_name = 'crash_results.csv'
     csv_file = os.path.join(output_folder, file_name)
     csv_columns = [
-        'Student', 'YouTubeLink', 'IsSimulation', 'UniqueCrashes', 'CrashFrames', 'FlightFrames',
+        'Student', 'VideoFileName', 'IsSimulation', 'UniqueCrashes', 'CrashFrames', 'FlightFrames',
         'NoDroneFrames', 'NoSignalFrames', 'NoStartedFrames', 'StartedFrames', 'UnstableFrames',
         'LandingFrames', 'CrashTimes', 'CrashDurations'
     ]
@@ -94,7 +50,7 @@ def result_csv(student, simulation_results, real_results):
 
             writer.writerow({
                 'Student': student,
-                'YouTubeLink': result.video_link,
+                'VideoFileName': result.video_name,
                 'IsSimulation': result.is_simulation,
                 'UniqueCrashes': unique_crashes,
                 'CrashFrames': result.label_counts.get("Crash", 0),
@@ -110,12 +66,6 @@ def result_csv(student, simulation_results, real_results):
             })
     return csv_file
 
-def delete_youtube_videos(simulation_paths, real_paths):
-    all_video_paths = simulation_paths + real_paths
-    for video_file in all_video_paths:
-        if os.path.exists(video_file):
-            os.remove(video_file)
-
 def remove_videos_and_csv(folder_path):
     video_extensions = {'.mp4'}
     csv_extension = '.csv'
@@ -126,20 +76,45 @@ def remove_videos_and_csv(folder_path):
             if extension.lower() in video_extensions or extension.lower() == csv_extension:
                 os.remove(file_path)
 
-def backend(student_name_tb, simulation_videos_tb, real_videos_tb, frame_output, text_output, file_output, video_output):
+def delete_uploaded_videos(simulation_paths, real_paths):
+    all_video_paths = simulation_paths + real_paths
+    for video_file in all_video_paths:
+        if os.path.exists(video_file):
+            os.remove(video_file)
+
+def backend(student_name_tb, simulation_files, real_files, frame_output, text_output, file_output, video_output):
     yield frame_output, "Starting process...", file_output, video_output
     remove_videos_and_csv(output_folder)
-    yield frame_output, "Parsing Input...", file_output, video_output
-    name, simulation_links, real_links = parse_inputs(student_name_tb, simulation_videos_tb, real_videos_tb)
-    yield frame_output, "Downloading YouTube Videos...", file_output, video_output
-    simulation_paths, real_paths = download_videos(simulation_links, real_links)
+    yield frame_output, "Processing Input...", file_output, video_output
+    name = student_name_tb.strip()
+    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())
+        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())
+        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 = []
 
-    for i, path in enumerate(simulation_paths, start=1): # for each sim video
-        # setup output paths
+    # Process simulation videos
+    for i, path in enumerate(simulation_paths, start=1):
+        # Setup output paths
         crash_video_filename = f'Simulation_Video_{i}_Crashes.mp4'
         crash_video_path = os.path.join(output_folder, crash_video_filename)
         labeled_video_filename = f'Simulation_Video_{i}_Labeled.mp4'
@@ -163,7 +138,7 @@ def backend(student_name_tb, simulation_videos_tb, real_videos_tb, frame_output,
                     video_path=path,
                     label_counts=label_counts,
                     crash_events=crash_events,
-                    video_link=simulation_links[i - 1],
+                    video_name=simulation_filenames[i - 1],
                     is_simulation=1))
                 break
 
@@ -192,12 +167,13 @@ def backend(student_name_tb, simulation_videos_tb, real_videos_tb, frame_output,
                     video_path=path,
                     label_counts=label_counts,
                     crash_events=crash_events,
-                    video_link=real_links[i - 1],
+                    video_name=real_filenames[i - 1],
                     is_simulation=0))
                 break
 
     yield frame_output, "Outputting Results to CSV...", file_output, video_output
     csv_file = result_csv(name, simulation_results, real_results)
     yield frame_output, "Finishing process...", file_output, video_output
-    delete_youtube_videos(simulation_paths, real_paths)
+    # Delete the uploaded video files from the server
+    delete_uploaded_videos(simulation_paths, real_paths)
     yield frame_output, "Download Ready.", csv_file, output_video_paths