Update app.py

app.py

@@ -1,5 +1,5 @@
 """
-This module integrates real-time object detection into live YouTube streams using the YOLO (You Only Look Once) model and provides an interactive user interface through Gradio. It allows users to search for live YouTube streams and apply object detection to these streams in real time.
+This module integrates real-time object detection into live YouTube streams using the YOLO model and provides an interactive user interface through Gradio. It allows users to search for live YouTube streams and apply object detection to these streams in real time.
 
 Main Features:
 - Search for live YouTube streams using specific queries.
@@ -18,7 +18,7 @@ Dependencies:
 - NumPy: Utilized for numerical operations on image data.
 - Pillow (`PIL`): A Python Imaging Library for opening, manipulating, and saving images.
 - Ultralytics YOLO: The YOLO model implementation for object detection.
-- `youtube-search-python`: Used for searching YouTube without API keys.
+- `innertube`: Used for interacting with YouTube's internal API.
 - `imageio`: For reading frames from live streams using FFmpeg.
 
 Usage:
@@ -36,7 +36,7 @@ import numpy as np
 from ultralytics import YOLO
 import streamlink
 from PIL import Image
-from youtubesearchpython import CustomSearch, SearchMode
+import innertube
 import imageio.v3 as iio
 
 logging.basicConfig(level=logging.DEBUG)
@@ -45,10 +45,11 @@ logging.basicConfig(level=logging.DEBUG)
 class SearchService:
     """
     SearchService provides functionality to search for YouTube videos using the
-    `youtube-search-python` library and retrieve live stream URLs using the Streamlink library.
+    `innertube` library and retrieve live stream URLs using the Streamlink library.
 
     Methods:
         search: Searches YouTube for videos matching a query and live filter.
+        parse: Parses raw search response data into a list of video details.
         get_youtube_url: Constructs a YouTube URL for a given video ID.
         get_stream: Retrieves the stream URL for a given YouTube video URL.
     """
@@ -65,22 +66,43 @@ class SearchService:
         :return: A list of search results, each a dictionary with video details.
         :rtype: List[Dict[str, Any]]
         """
+        client = innertube.InnerTube()
+        params = "EgJAAQ%3D%3D" if live else None  # 'Live' filter code
+        
+        response = client.search(query=query, params=params)
+        results = SearchService.parse(response)
+        return results
+
+    @staticmethod
+    def parse(response: Dict[str, Any]) -> List[Dict[str, Any]]:
+        """
+        Parses the raw search response data into a list of video details.
+
+        :param response: The raw search response data from YouTube.
+        :type response: Dict[str, Any]
+        :return: A list of parsed video details.
+        :rtype: List[Dict[str, Any]]
+        """
         results = []
-        # Apply live filter if needed
-        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'],
-                })
+        try:
+            contents = response["contents"]["twoColumnSearchResultsRenderer"]["primaryContents"]["sectionListRenderer"]["contents"]
+            for content in contents:
+                items = content.get("itemSectionRenderer", {}).get("contents", [])
+                for item in items:
+                    video_renderer = item.get("videoRenderer")
+                    if video_renderer:
+                        video_id = video_renderer.get("videoId")
+                        thumbnails = video_renderer.get("thumbnail", {}).get("thumbnails", [])
+                        title_runs = video_renderer.get("title", {}).get("runs", [])
+                        title = "".join([run.get("text", "") for run in title_runs])
+                        thumbnail_url = thumbnails[-1].get("url") if thumbnails else ""
+                        results.append({
+                            'video_id': video_id,
+                            'thumbnail_url': thumbnail_url,
+                            'title': title,
+                        })
+        except Exception as e:
+            logging.error(f"Error parsing search results: {e}")
         return results
 
     @staticmethod
@@ -138,7 +160,7 @@ class LiveYouTubeObjectDetector:
     def __init__(self):
         """Initializes the LiveYouTubeObjectDetector with YOLO model and UI components."""
         logging.getLogger().setLevel(logging.DEBUG)
-        self.model = YOLO("yolov8x.pt")
+        self.model = YOLO("yolov8n.pt")
         self.model.fuse()
         self.streams = self.get_live_streams("world live cams")
 
@@ -173,7 +195,7 @@ class LiveYouTubeObjectDetector:
         if not stream_url:
             return None
         try:
-            reader = iio.imiter(stream_url, plugin='ffmpeg', fps=1)
+            reader = iio.imiter(stream_url, plugin='pyav', thread_type='AUTO')
             loop = asyncio.get_event_loop()
             frame = await loop.run_in_executor(None, next, reader, None)
             if frame is None:
@@ -243,12 +265,14 @@ class LiveYouTubeObjectDetector:
         The Gradio interface allows users to search for live YouTube streams, select a stream,
         and run object detection on the selected live stream.
         """
-        with gr.Blocks(title="Object Detection in Live YouTube Streams", css=".gradio-container {background-color: #f5f5f5}", theme=gr.themes.Soft()) as app:
+        with gr.Blocks(title="Object Detection in Live YouTube Streams",
+                       css=".gradio-container {background-color: #f5f5f5}", theme=gr.themes.Soft()) as app:
             gr.HTML("<h1 style='text-align: center; color: #1E88E5;'>Object Detection in Live YouTube Streams</h1>")
             with gr.Tabs():
                 with gr.TabItem("Live Stream Detection"):
                     with gr.Row():
-                        stream_input = gr.Textbox(label="URL of Live YouTube Video", placeholder="Enter YouTube live stream URL...", interactive=True)
+                        stream_input = gr.Textbox(label="URL of Live YouTube Video",
+                                                  placeholder="Enter YouTube live stream URL...", interactive=True)
                         submit_button = gr.Button("Detect Objects", variant="primary")
                     annotated_image = gr.AnnotatedImage(label="Detection Result", height=480)
                     status_text = gr.Markdown(value="Status: Ready", visible=False)
@@ -268,13 +292,14 @@ class LiveYouTubeObjectDetector:
 
                 with gr.TabItem("Search Live Streams"):
                     with gr.Row():
-                        search_input = gr.Textbox(label="Search for Live YouTube Streams", placeholder="Enter search query...", interactive=True)
+                        search_input = gr.Textbox(label="Search for Live YouTube Streams",
+                                                  placeholder="Enter search query...", interactive=True)
                         search_button = gr.Button("Search", variant="secondary")
                     gallery = gr.Gallery(label="Live YouTube Streams", show_label=False).style(grid=[4], height="auto")
                     gallery.style(item_height=150)
                     status_text_search = gr.Markdown(value="", visible=False)
 
-                    def search_live_streams(query):
+                    async def search_live_streams(query):
                         status_text_search.update(value="Searching...", visible=True)
                         self.streams = self.get_live_streams(query)
                         gallery_items = []