Spaces:
Sleeping
Sleeping
| # -*- coding: utf-8 -*- | |
| """youtube_video_recommendation.ipynb | |
| Automatically generated by Colab. | |
| Original file is located at | |
| https://colab.research.google.com/drive/1pLpvcg7hC7hp2PmGBc8qe86ktv3eT9gp | |
| """ | |
| import gradio as gr | |
| import requests | |
| import os | |
| API_KEY = os.getenv("API_KEY")# Replace with your own API key | |
| def search_youtube(query): | |
| if not query.strip(): | |
| return "<b>Please enter a search term.</b>" | |
| search_url = "https://www.googleapis.com/youtube/v3/search" | |
| search_params = { | |
| "part": "snippet", | |
| "q": query, | |
| "type": "video", | |
| "maxResults": 25, | |
| "key": API_KEY | |
| } | |
| search_response = requests.get(search_url, params=search_params).json() | |
| video_ids = [item["id"]["videoId"] for item in search_response.get("items", [])] | |
| if not video_ids: | |
| return "<b>❌ No results found.</b>" | |
| stats_url = "https://www.googleapis.com/youtube/v3/videos" | |
| stats_params = { | |
| "part": "snippet,statistics", | |
| "id": ",".join(video_ids), | |
| "key": API_KEY | |
| } | |
| stats_response = requests.get(stats_url, params=stats_params).json() | |
| if "items" not in stats_response: | |
| return "<b>❌ No results found or API limit reached.</b>" | |
| results = [] | |
| for item in stats_response["items"]: | |
| snippet = item["snippet"] | |
| title = snippet["title"] | |
| description = snippet.get("description", "") | |
| channel = snippet.get("channelTitle", "") | |
| tags = snippet.get("tags", []) | |
| video_id = item["id"] | |
| url = f"https://www.youtube.com/watch?v={video_id}" | |
| # Skip YouTube Shorts | |
| if "shorts" in title.lower() or "/shorts/" in url.lower(): | |
| continue | |
| combined_text = " ".join([title, description, channel] + tags).lower() | |
| stats = item.get("statistics", {}) | |
| thumbnail = snippet["thumbnails"]["medium"]["url"] | |
| views = int(stats.get("viewCount", 0)) | |
| likes = int(stats.get("likeCount", 0)) | |
| results.append({ | |
| "title": title, | |
| "thumbnail": thumbnail, | |
| "views": views, | |
| "likes": likes, | |
| "channel": channel, | |
| "url": url | |
| }) | |
| if not results: | |
| return "<b>❌ No suitable videos found.</b>" | |
| # Grid layout for results | |
| grid_cards = "" | |
| for vid in results: | |
| grid_cards += f""" | |
| <div style="background: #1e1e1e; color: white; border-radius: 12px; padding: 10px; box-shadow: 0 0 10px #333;"> | |
| <a href="{vid['url']}" target="_blank" style="text-decoration: none; color: inherit;"> | |
| <img src="{vid['thumbnail']}" style="width: 100%; height: auto; object-fit: cover; border-radius: 8px;" /> | |
| <div style="margin-top: 8px;"> | |
| <div style="color: #33ccff; font-size: 16px; font-weight: bold;">{vid['title']}</div> | |
| <div style="font-size: 14px; margin-top: 4px;">🎙️ {vid['channel']}</div> | |
| <div style="font-size: 14px;">👁️ {vid['views']:,} views ❤️ {vid['likes']:,} likes</div> | |
| </div> | |
| </a> | |
| </div> | |
| """ | |
| return f""" | |
| <div style='font-family: "Segoe UI", sans-serif; padding: 10px; max-height: 600px; overflow-y: auto;'> | |
| <div style="display: grid; grid-template-columns: repeat(auto-fit, minmax(280px, 1fr)); gap: 16px;"> | |
| {grid_cards} | |
| </div> | |
| </div> | |
| """ | |
| # 🎨 Gradio Interface (UI/Color/Font preserved) | |
| with gr.Blocks(theme=gr.themes.Base(), title="YouTube Course Finder") as demo: | |
| gr.HTML("<h1 style='text-align: center; background: linear-gradient(90deg, #ff0066, #6600ff); color: white; padding: 15px; border-radius: 10px;'>🔥 YouTube Course Finder</h1>") | |
| gr.Markdown("Search for top YouTube courses by topic. Only regular videos shown, Shorts are excluded.") | |
| with gr.Row(): | |
| query = gr.Textbox(label="Search Topic", placeholder="e.g. Python full course, HTML tutorial", lines=1) | |
| search_btn = gr.Button("🔍 Search") | |
| output = gr.HTML() | |
| search_btn.click(fn=search_youtube, inputs=query, outputs=output) | |
| demo.launch(share=True) |