Spaces:
Sleeping
Sleeping
File size: 3,187 Bytes
8bf3938 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 |
import gradio as gr
import requests
from bs4 import BeautifulSoup
import re
# Function to search YouTube videos using web scraping
def youtube_search(query, max_results=50):
search_url = f"https://www.youtube.com/results?search_query={query}"
headers = {
"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537.3"
}
try:
# Make the HTTP request to YouTube search results page
response = requests.get(search_url, headers=headers)
response.raise_for_status() # Raise an error for bad responses (4xx or 5xx)
# Parse the HTML content
soup = BeautifulSoup(response.text, "html.parser")
video_elements = soup.find_all("a", href=True, class_="yt-uix-tile-link")
# Extract up to 'max_results' video details
gallery_items = []
for idx, video in enumerate(video_elements):
if idx >= max_results:
break
video_id = video['href'].split('=')[-1]
video_title = video.get_text(strip=True)
thumbnail_url = f"https://img.youtube.com/vi/{video_id}/mqdefault.jpg"
# Append tuple (thumbnail, video ID)
gallery_items.append((thumbnail_url, video_id))
return gallery_items
except requests.exceptions.RequestException as e:
# Print the error message to help debug issues
print(f"Error during YouTube web scraping request: {e}")
return []
# Update Gradio-related parts
with gr.Blocks() as demo:
gr.Markdown("## YouTube Video Search, Selection, and Playback")
with gr.Row():
with gr.Column(scale=3):
search_query_input = gr.Textbox(label="Search YouTube", placeholder="Enter your search query here")
search_button = gr.Button("Search")
search_output = gr.Gallery(label="Search Results", columns=5, height="1500px")
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):
gallery_items = youtube_search(query)
return gallery_items
# Update the selected video link field when a video is clicked in the gallery
def on_video_select(evt: gr.SelectData):
# Extract the video ID from the event value, which is a dictionary containing details of the selected item
selected_video_id = evt.value["caption"]
video_url = f"https://www.youtube.com/watch?v={selected_video_id}"
return video_url
# Play the video when the Play Video button is clicked
def play_video(video_url):
return show_video(video_url)
search_button.click(update_search_results, inputs=search_query_input, outputs=search_output)
search_output.select(on_video_select, inputs=None, outputs=selected_video_link)
play_video_button.click(play_video, inputs=selected_video_link, outputs=video_output)
# Launch the Gradio interface
demo.launch()
|