Remove deprecated pafy dependency, replace with yt-dlp

CV_Studio.spec

@@ -33,7 +33,6 @@ hiddenimports += collect_submodules('numpy')
 hiddenimports += collect_submodules('librosa')
 hiddenimports += collect_submodules('soundfile')
 hiddenimports += collect_submodules('matplotlib')
-hiddenimports += collect_submodules('pafy')
 hiddenimports += collect_submodules('youtube_dl')
 hiddenimports += collect_submodules('yt_dlp')
 hiddenimports += collect_submodules('filterpy')
@@ -68,7 +67,6 @@ hiddenimports += [
     'src.utils.logging',
     'src.utils.gpu_utils',
     'src.core',
-    'pafy',
     'youtube_dl',
     'yt_dlp',
     'filterpy',

build_exe.py

@@ -197,7 +197,6 @@ def generate_spec_file():
 hiddenimports += collect_submodules('librosa')
 hiddenimports += collect_submodules('soundfile')
 hiddenimports += collect_submodules('matplotlib')
-hiddenimports += collect_submodules('pafy')
 hiddenimports += collect_submodules('youtube_dl')
 hiddenimports += collect_submodules('yt_dlp')
 hiddenimports += collect_submodules('filterpy')
@@ -232,7 +231,6 @@ def generate_spec_file():
     'src.utils.logging',
     'src.utils.gpu_utils',
     'src.core',
-    'pafy',
     'youtube_dl',
     'yt_dlp',
     'filterpy',

node/InputNode/node_webrtc.py

@@ -3,9 +3,9 @@
 import time
 
 import cv2
-import pafy
 import numpy as np
 import dearpygui.dearpygui as dpg
+import yt_dlp
 
 from node_editor.util import dpg_get_value, dpg_set_value
 
@@ -384,10 +384,27 @@ def _button(self, sender, data, user_data):
                     dpg.set_item_label(tag_node_button_value_name,
                                        self._loading_label)
 
-                    pafy_video = pafy.new(youtube_url)
-                    pafy_best_video = pafy_video.getbest(preftype="mp4")
-                    youtube_capture = YoutubeCapture(pafy_best_video.url)
-                    self._youtube_capture[youtube_url] = youtube_capture
+                    # Use yt-dlp to get the best video URL
+                    ydl_opts = {
+                        'quiet': True,
+                        'format': 'best[ext=mp4]/best',  # Prefer MP4, fallback to best available
+                    }
+                    try:
+                        with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+                            info = ydl.extract_info(youtube_url, download=False)
+                            video_url = info.get('url', None)
+                            if not video_url:
+                                raise ValueError("No video URL found")
+                            youtube_capture = YoutubeCapture(video_url)
+                            self._youtube_capture[youtube_url] = youtube_capture
+                    except yt_dlp.utils.DownloadError as e:
+                        print(f"Error downloading video info: {e}")
+                        dpg.set_item_label(tag_node_button_value_name, self._start_label)
+                        return
+                    except Exception as e:
+                        print(f"Unexpected error extracting video URL: {e}")
+                        dpg.set_item_label(tag_node_button_value_name, self._start_label)
+                        return
 
                     dpg.set_item_label(tag_node_button_value_name,
                                        self._stop_label)

node/InputNode/node_youtube.py

@@ -5,7 +5,6 @@
 from threading import Lock
 
 import cv2
-import pafy
 import numpy as np
 import dearpygui.dearpygui as dpg
 import yt_dlp

node/StatsNode/node_bar.py

@@ -3,7 +3,6 @@
 import time
 
 import cv2
-import pafy
 import numpy as np
 import dearpygui.dearpygui as dpg
 

requirements.txt

@@ -8,7 +8,6 @@ dearpygui>=1.11.0
 mediapipe
 protobuf>=3.20.0
 filterpy>=1.4.5
-pafy
 ffmpeg-python
 serial
 pytest