Update app.py

app.py

@@ -6,6 +6,7 @@ subprocess.check_call([sys.executable, "-m", "pip", "install", "httpx==0.18.2",
 
 import gradio as gr
 import requests
+import os
 import re
 import yt_dlp
 import logging
@@ -13,6 +14,24 @@ import logging
 # Configure logging for debugging purposes
 logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(levelname)s - %(message)s')
 
+# Fetch the keys from the environment variable and convert them into a list
+YOUTUBE_API_KEYS = os.getenv("YOUTUBE_API_KEYS")
+
+if YOUTUBE_API_KEYS:
+    YOUTUBE_API_KEYS = [key.strip() for key in YOUTUBE_API_KEYS.split(",")]
+else:
+    raise ValueError("API keys not found. Make sure the secret 'YOUTUBE_API_KEYS' is set.")
+
+# Index to keep track of which API key to use
+key_index = 0
+
+def get_api_key():
+    global key_index
+    # Get the current API key and increment the index
+    api_key = YOUTUBE_API_KEYS[key_index]
+    key_index = (key_index + 1) % len(YOUTUBE_API_KEYS)  # Rotate to the next key
+    return api_key
+
 # Function to search YouTube videos using yt-dlp for better reliability
 def youtube_search(query, max_results=50):
     ydl_opts = {
@@ -51,6 +70,50 @@ def youtube_search(query, max_results=50):
         logging.error(error_message)
         return [], error_message
 
+# Function to search YouTube videos using the API with pagination to get up to 1,000 results
+def youtube_api_search(query, max_results=1000):
+    search_url = "https://www.googleapis.com/youtube/v3/search"
+    all_results = []
+    params = {
+        "part": "snippet",
+        "q": query,
+        "type": "video",
+        "maxResults": 50  # YouTube API allows a maximum of 50 per request
+    }
+
+    try:
+        while len(all_results) < max_results:
+            params["key"] = get_api_key()  # Get the current API key
+            response = requests.get(search_url, params=params)
+
+            # If we get a bad response, try the next API key
+            if response.status_code == 403 or response.status_code == 429:
+                logging.debug(f"Quota exceeded or forbidden for API key. Trying next key...")
+                continue
+            response.raise_for_status()  # Raise an error for other bad responses (4xx or 5xx)
+
+            results = response.json().get("items", [])
+            all_results.extend(results)
+
+            # If there is no nextPageToken, we've reached the end
+            if 'nextPageToken' not in response.json() or len(all_results) >= max_results:
+                break
+
+            # Update params with the nextPageToken to get the next batch of results
+            params['pageToken'] = response.json()['nextPageToken']
+
+        # Create a list of tuples with thumbnail URL and video ID as the caption
+        gallery_items = [
+            (result["snippet"]["thumbnails"]["medium"]["url"], result["id"]["videoId"]) for result in all_results
+        ]
+
+        return gallery_items
+
+    except requests.exceptions.RequestException as e:
+        # Print the error message to help debug issues
+        logging.error(f"Error during YouTube API request: {e}")
+        return [], f"Error retrieving video results: {str(e)}"
+
 # Function to display the video using the video URL
 def show_video(video_url):
     # Regular expression to extract the YouTube video ID from the URL
@@ -105,6 +168,8 @@ with gr.Blocks() as demo:
     # Define search button behavior
     def update_search_results(query):
         gallery_items, error_message = youtube_search(query)
+        if not gallery_items:  # If yt-dlp search fails or returns empty, fall back to YouTube API
+            gallery_items = youtube_api_search(query)
         if error_message:
             return [], error_message, gr.update(visible=True)
         # Display videos even if the title or thumbnail is missing by using placeholders
@@ -118,7 +183,7 @@ with gr.Blocks() as demo:
         video_url = f"https://www.youtube.com/watch?v={selected_video_id}"
         logging.debug(f"Video selected: {video_url}")
         return video_url
-
+    
     # Play the video when the Play Video button is clicked
     def play_video(video_url):
         logging.debug(f"Playing video with URL: {video_url}")
@@ -129,4 +194,4 @@ with gr.Blocks() as demo:
     play_video_button.click(play_video, inputs=selected_video_link, outputs=video_output)
 
 # Launch the Gradio interface
-demo.launch()
+demo.launch()