Update app.py

app.py

@@ -46,7 +46,6 @@ def youtube_search(query, upload_date, video_type, duration, sort_by, max_result
         "order": sort_by
     }
 
-    # Add filter parameters based on user selection
     if upload_date:
         published_after = get_iso8601_date(upload_date)
         if published_after:
@@ -75,7 +74,6 @@ def youtube_search(query, upload_date, video_type, duration, sort_by, max_result
 
             results = response.json().get("items", [])
             for result in results:
-                # Determine the type of the result and extract the correct ID
                 if result["id"]["kind"] == "youtube#video":
                     video_info = {
                         'thumbnail_url': result["snippet"]["thumbnails"]["high"]["url"],
@@ -98,7 +96,7 @@ def youtube_search(query, upload_date, video_type, duration, sort_by, max_result
                         'type': "playlist"
                     }
                 else:
-                    continue  # Skip if it's an unknown type
+                    continue
                 
                 all_results.append(video_info)
 
@@ -162,12 +160,14 @@ with gr.Blocks(css="""
 """) as demo:
     gr.Markdown("## YouTube Video Search, Selection, and Playback")
     video_ids_state = gr.State()
+    video_types_state = gr.State()
 
     with gr.Row(elem_id="video_container"):
         video_output = gr.HTML(label="Video Player", elem_id="video_output")
 
     with gr.Row():
         selected_video_link = gr.Textbox(label="Selected Video Link", interactive=False)
+        selected_video_type = gr.Textbox(label="Selected Video Type", visible=False, interactive=False)
         play_video_button = gr.Button("Play Video")
 
     with gr.Row():
@@ -176,7 +176,7 @@ with gr.Blocks(css="""
                                             placeholder="Enter your search query here", 
                                             elem_id="search_query_input")
             upload_date_input = gr.Dropdown(label="Upload Date", choices=["", "Last Hour", "Today", "This Week", "This Month", "This Year"])
-            video_type_input = gr.Dropdown(label="Type", choices=["All", "Video", "Channel", "Playlist", "Movie"])
+            video_type_input = gr.Dropdown(label="Type", choices=["All", "Video", "Channel", "Playlist"])
             duration_input = gr.Dropdown(label="Duration", choices=["", "Short (<4 mins)", "Medium (4-20 mins)", "Long (>20 mins)"])
             sort_by_input = gr.Dropdown(label="Sort By", choices=["relevance", "date", "viewCount", "rating"], value="relevance")
             search_button = gr.Button("Search")
@@ -200,13 +200,26 @@ with gr.Blocks(css="""
         index = evt.index
         selected_video_id = video_ids[index]
         selected_video_type = video_types[index]
-        return selected_video_id, selected_video_type
+        video_url = f"https://www.youtube.com/watch?v={selected_video_id}" if selected_video_type == "video" else ""
+        return video_url, selected_video_type
 
     def play_video(video_id, video_type):
         return show_video(video_id, video_type)
 
-    search_button.click(update_search_results, inputs=[search_query_input, upload_date_input, video_type_input, duration_input, sort_by_input], outputs=[search_output, video_ids_state])
-    search_output.select(on_video_select, inputs=[video_ids_state, gr.State()], outputs=[selected_video_link, gr.State()])
-    play_video_button.click(play_video, inputs=[selected_video_link, gr.State()], outputs=video_output)
+    search_button.click(
+        update_search_results,
+        inputs=[search_query_input, upload_date_input, video_type_input, duration_input, sort_by_input],
+        outputs=[search_output, video_ids_state, video_types_state]
+    )
+    search_output.select(
+        on_video_select,
+        inputs=[video_ids_state, video_types_state],
+        outputs=[selected_video_link, selected_video_type]
+    )
+    play_video_button.click(
+        play_video,
+        inputs=[selected_video_link, selected_video_type],
+        outputs=video_output
+    )
 
 demo.launch()