Update app.py

app.py

@@ -36,7 +36,7 @@ import numpy as np
 from ultralytics import YOLO
 import streamlink
 from PIL import Image
-from youtubesearchpython import VideosSearch
+from youtubesearchpython import CustomSearch, SearchMode
 import imageio.v3 as iio
 
 logging.basicConfig(level=logging.DEBUG)
@@ -67,14 +67,20 @@ class SearchService:
         """
         results = []
         # Apply live filter if needed
-        search_preferences = "EgJAAQ%3D%3D" if live else None  # 'Live' filter code
-        videos_search = VideosSearch(query, limit=20, searchPreferences=search_preferences)
-        for result in videos_search.result()['result']:
-            results.append({
-                'video_id': result['id'],
-                'thumbnail_url': result['thumbnails'][-1]['url'],
-                'title': result['title'],
-            })
+        if live:
+            # Use CustomSearch with SearchMode.live to filter live videos
+            search = CustomSearch(query, SearchMode.live, limit=20)
+        else:
+            # Use CustomSearch with default mode for regular videos
+            search = CustomSearch(query, SearchMode.videos, limit=20)
+        for result in search.result()['result']:
+            if 'video' in result:
+                video = result['video']
+                results.append({
+                    'video_id': video['id'],
+                    'thumbnail_url': video['thumbnails'][-1]['url'],
+                    'title': video['title'],
+                })
         return results
 
     @staticmethod
@@ -170,6 +176,10 @@ class LiveYouTubeObjectDetector:
             reader = iio.imiter(stream_url, plugin='ffmpeg', fps=1)
             loop = asyncio.get_event_loop()
             frame = await loop.run_in_executor(None, next, reader, None)
+            if frame is None:
+                return None
+            # Resize frame if needed
+            frame = cv2.resize(frame, (1280, 720))
             return frame
         except StopIteration:
             logging.warning("Could not read frame from stream.")