Trying new YouTube method

Dockerfile

@@ -22,6 +22,7 @@ COPY . /app
 
 RUN pip install gradio
 RUN pip install yt-dlp
+RUN pip install pytube
    
 EXPOSE 7860
 ENV PYTHONUNBUFFERED=1 \

app_backend.py

@@ -3,6 +3,7 @@ import os
 import csv
 from yt_dlp import YoutubeDL
 from dataclasses import dataclass
+import pytube
 
 device = 'cpu'
 model_path = './model/best.pt'
@@ -24,24 +25,42 @@ def parse_inputs(student_name_tb, simulation_videos_tb, real_videos_tb):
     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):
+    """Downloads a YouTube video, renames it to the first three characters, and returns the downloaded file path."""
+
+    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])
+        #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])
+        #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