Update app.py

app.py

@@ -91,8 +91,33 @@ def show_video(video_url):
     '''
     return html_code
 
+# Function to handle video selection
+def select_video(video_id):
+    video_url = f"https://www.youtube.com/watch?v={video_id}"
+    return video_url
+
 # Create the Gradio interface
-with gr.Blocks() as demo:
+with gr.Blocks(css="""
+.search-item {
+    display: flex;
+    align-items: flex-start;
+    margin-bottom: 30px;
+    cursor: pointer;
+}
+.search-item img {
+    width: 300px;
+    height: auto;
+    margin-right: 20px;
+}
+.search-item h3 {
+    font-size: 24px;
+    margin: 0 0 10px 0;
+}
+.search-item p {
+    font-size: 18px;
+    margin: 0;
+}
+""") as demo:
     gr.Markdown("## YouTube Video Search, Selection, and Playback")
 
     with gr.Row():
@@ -103,19 +128,20 @@ with gr.Blocks() as demo:
 
         with gr.Column(scale=2):
             selected_video_link = gr.Textbox(label="Selected Video Link", interactive=False)
+            play_video_button = gr.Button("Play Video")
             video_output = gr.HTML(label="Video Player")
 
     # Define search button behavior
     def update_search_results(query):
         search_results = youtube_search(query)
-        html_code = '<div id="search-results">'
+        html_code = '<div>'
         for item in search_results:
             video_id = item['video_id']
             thumbnail_url = item['thumbnail_url']
             title = item['title']
             description = item['description']
             html_code += f'''
-            <div class="search-item" data-video-id="{video_id}">
+            <div class="search-item" onclick="selectVideo('{video_id}')">
                 <img src="{thumbnail_url}" alt="{title}">
                 <div>
                     <h3>{title}</h3>
@@ -123,17 +149,41 @@ with gr.Blocks() as demo:
                 </div>
             </div>
             '''
+        html_code += '''
+        <script>
+        function selectVideo(video_id) {
+            fetch('/select_video', {
+                method: 'POST',
+                headers: { 'Content-Type': 'application/json' },
+                body: JSON.stringify({ 'video_id': video_id })
+            })
+            .then(response => response.json())
+            .then(data => {
+                const gradioApp = document.querySelector('gradio-app').shadowRoot || document;
+                const textbox = gradioApp.querySelector('#selected_video_link textarea');
+                if (!textbox) {
+                    console.error('Selected video link textbox not found');
+                    return;
+                }
+                textbox.value = data;
+                textbox.dispatchEvent(new Event('input', { bubbles: true }));
+            })
+            .catch(error => {
+                console.error('Error selecting video:', error);
+            });
+        }
+        </script>
+        '''
         html_code += '</div>'
         return html_code
 
-    # Function to handle video selection and playback
-    def select_and_play_video(video_id: str):
-        video_url = f"https://www.youtube.com/watch?v={video_id}"
-        return video_url, show_video(video_url)
+    # Play the video when the Play Video button is clicked
+    def play_video(video_url):
+        return show_video(video_url)
 
-    # Set up event listeners
     search_button.click(update_search_results, inputs=search_query_input, outputs=search_output)
-    search_output.select(select_and_play_video, inputs=gr.Textbox(visible=False), outputs=[selected_video_link, video_output])
+    play_video_button.click(play_video, inputs=selected_video_link, outputs=video_output)
+    select_video_event = gr.Event(fn=select_video, inputs="video_id", outputs=selected_video_link, api_name="select_video")
 
 # Launch the Gradio interface
-demo.launch()
+demo.launch()