Update app.py

app.py

@@ -3,7 +3,93 @@ import requests
 import os
 import re
 
-# ... [previous code remains the same] ...
+# 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 the API
+def youtube_search(query, max_results=50):
+    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()
+            response = requests.get(search_url, params=params)
+
+            if response.status_code == 403 or response.status_code == 429:
+                print(f"Quota exceeded or forbidden for API key. Trying next key...")
+                continue
+            response.raise_for_status()
+
+            results = response.json().get("items", [])
+            for result in results:
+                video_info = {
+                    'thumbnail_url': result["snippet"]["thumbnails"]["high"]["url"],
+                    'video_id': result["id"]["videoId"],
+                    'title': result["snippet"]["title"],
+                    'description': result["snippet"]["description"]
+                }
+                all_results.append(video_info)
+
+            if 'nextPageToken' not in response.json() or len(all_results) >= max_results:
+                break
+
+            params['pageToken'] = response.json()['nextPageToken']
+
+        return all_results
+
+    except requests.exceptions.RequestException as e:
+        print(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):
+    video_id = None
+    patterns = [
+        r"youtube\.com/watch\?v=([^\&\?\/]+)",
+        r"youtube\.com/embed/([^\&\?\/]+)",
+        r"youtube\.com/v/([^\&\?\/]+)",
+        r"youtu\.be/([^\&\?\/]+)"
+    ]
+
+    for pattern in patterns:
+        match = re.search(pattern, video_url)
+        if match:
+            video_id = match.group(1)
+            break
+
+    if not video_id:
+        return "Invalid YouTube URL. Please enter a valid YouTube video link."
+
+    embed_url = f"https://www.youtube.com/embed/{video_id}"
+
+    html_code = f'''
+    <iframe width="560" height="315" src="{embed_url}" 
+    frameborder="0" allow="accelerometer; autoplay; clipboard-write; encrypted-media; gyroscope; picture-in-picture" 
+    allowfullscreen></iframe>
+    '''
+    return html_code
 
 # Create the Gradio interface
 with gr.Blocks(css="""
@@ -36,7 +122,7 @@ with gr.Blocks(css="""
             search_output = gr.HTML(label="Search Results")
 
         with gr.Column(scale=2):
-            selected_video_link = gr.Textbox(label="Selected Video Link", interactive=False)
+            selected_video_link = gr.Textbox(label="Selected Video Link", interactive=False, elem_id='selected_video_link')
             play_video_button = gr.Button("Play Video")
             video_output = gr.HTML(label="Video Player")
 
@@ -60,10 +146,24 @@ with gr.Blocks(css="""
             '''
         html_code += '''
         <script>
+        function getGradioApp() {
+            const gradioApp = document.querySelector('gradio-app');
+            if (gradioApp.shadowRoot) {
+                return gradioApp.shadowRoot;
+            } else {
+                return document;
+            }
+        }
+
         function selectVideo(video_id) {
-            const videoLink = 'https://www.youtube.com/watch?v=' + video_id;
-            document.querySelector('gradio-app').querySelector('#selected_video_link input').value = videoLink;
-            document.querySelector('gradio-app').querySelector('#selected_video_link input').dispatchEvent(new Event('input', { bubbles: true }));
+            const gradioApp = getGradioApp();
+            const textbox = gradioApp.querySelector('#selected_video_link textarea');
+            if (!textbox) {
+                console.error('Selected video link textbox not found');
+                return;
+            }
+            textbox.value = 'https://www.youtube.com/watch?v=' + video_id;
+            textbox.dispatchEvent(new Event('input', { bubbles: true }));
         }
         </script>
         '''
@@ -78,4 +178,4 @@ with gr.Blocks(css="""
     play_video_button.click(play_video, inputs=selected_video_link, outputs=video_output)
 
 # Launch the Gradio interface
-demo.launch()
+demo.launch()