Update app.py

app.py

@@ -5,29 +5,94 @@ import os
 API_KEY = os.getenv("YOUTUBE_API_KEY")
 youtube = build("youtube", "v3", developerKey=API_KEY)
 
-# 국가 코드
+# 국가 코드 - 인구 순 상위 40개
 COUNTRIES = {
     "전세계": "",
-    "한국": "KR",
+    "중국": "CN",
+    "인도": "IN",
     "미국": "US",
+    "인도네시아": "ID",
+    "파키스탄": "PK",
+    "브라질": "BR",
+    "나이지리아": "NG",
+    "방글라데시": "BD",
+    "러시아": "RU",
+    "멕시코": "MX",
     "일본": "JP",
-    "영국": "GB",
+    "에티오피아": "ET",
+    "필리핀": "PH",
+    "이집트": "EG",
+    "베트남": "VN",
+    "콩고민주공화국": "CD",
+    "터키": "TR",
+    "이란": "IR",
     "독일": "DE",
+    "태국": "TH",
+    "영국": "GB",
     "프랑스": "FR",
-    "인도": "IN",
-    "브라질": "BR",
+    "이탈리아": "IT",
+    "남아공": "ZA",
+    "탄자니아": "TZ",
+    "미얀마": "MM",
+    "한국": "KR",
+    "콜롬비아": "CO",
+    "케냐": "KE",
+    "스페인": "ES",
+    "아르헨티나": "AR",
+    "알제리": "DZ",
+    "우크라이나": "UA",
+    "이라크": "IQ",
+    "캐나다": "CA",
+    "폴란드": "PL",
+    "모로코": "MA",
+    "사우디": "SA",
+    "페루": "PE",
+    "말레이시아": "MY",
 }
 
-# 언어 코드
+# 언어 코드 - 사용 인구 순 상위 40개
 LANGUAGES = {
     "전체": "",
-    "한국어": "ko",
     "영어": "en",
-    "일본어": "ja",
     "중국어": "zh",
+    "힌디어": "hi",
     "스페인어": "es",
+    "아랍어": "ar",
+    "벵골어": "bn",
+    "포르투갈어": "pt",
+    "러시아어": "ru",
+    "일본어": "ja",
+    "펀자브어": "pa",
     "독일어": "de",
+    "인도네시아어": "id",
+    "페르시아어": "fa",
     "프랑스어": "fr",
+    "베트남어": "vi",
+    "터키어": "tr",
+    "한국어": "ko",
+    "타밀어": "ta",
+    "이탈리아어": "it",
+    "텔루구어": "te",
+    "마라티어": "mr",
+    "우르두어": "ur",
+    "태국어": "th",
+    "구자라트어": "gu",
+    "폴란드어": "pl",
+    "말라얄람어": "ml",
+    "칸나다어": "kn",
+    "우크라이나어": "uk",
+    "필리핀어": "tl",
+    "미얀마어": "my",
+    "네덜란드어": "nl",
+    "루마니아어": "ro",
+    "스와힐리어": "sw",
+    "그리스어": "el",
+    "체코어": "cs",
+    "헝가리어": "hu",
+    "스웨덴어": "sv",
+    "히브리어": "he",
+    "핀란드어": "fi",
+    "노르웨이어": "no",
 }
 
 # 정렬 기준
@@ -47,6 +112,40 @@ DATE_OPTIONS = {
     "올해": "thisYear",
 }
 
+def get_trending_keywords(region_code="KR"):
+    """YouTube 인기 동영상에서 트렌드 키워드 추출"""
+    try:
+        response = youtube.videos().list(
+            part="snippet",
+            chart="mostPopular",
+            regionCode=region_code if region_code else "US",
+            maxResults=20
+        ).execute()
+        
+        keywords = []
+        seen = set()
+        for item in response.get("items", []):
+            title = item["snippet"]["title"]
+            channel = item["snippet"]["channelTitle"]
+            keyword = channel.split()[0] if len(channel.split()) > 0 else title.split()[0]
+            if keyword not in seen and len(keyword) > 1:
+                keywords.append(keyword)
+                seen.add(keyword)
+            if len(keywords) >= 15:
+                break
+        
+        return keywords if keywords else get_default_trending()
+    except:
+        return get_default_trending()
+
+def get_default_trending():
+    """기본 인기 검색어"""
+    return [
+        "AI", "ChatGPT", "뉴진스", "BTS", "Netflix", 
+        "게임", "먹방", "브이로그", "shorts", "음악",
+        "축구", "NBA", "애니메이션", "영화 리뷰", "코딩"
+    ]
+
 def format_count(count):
     if count >= 1000000000:
         return f"{count/1000000000:.1f}B"
@@ -56,17 +155,29 @@ def format_count(count):
         return f"{count/1000:.1f}K"
     return str(count)
 
+def update_trending(country):
+    """국가 변경시 트렌딩 키워드 업데이트"""
+    region = COUNTRIES.get(country, "")
+    keywords = get_trending_keywords(region if region else "US")
+    return gr.update(choices=keywords, value=None)
+
+def use_trending_keyword(trending_keyword):
+    """트렌딩 키워드 선택시 검색창에 입력"""
+    return trending_keyword if trending_keyword else ""
+
 def search_videos(keyword, country, language, sort_by, date_filter, max_results):
-    if not keyword.strip():
+    if not keyword or not keyword.strip():
         return "검색어를 입력하세요."
     
-    # 검색 요청
+    max_results = int(max_results)
+    all_items = []
+    next_page_token = None
+    
     search_params = {
         "q": keyword,
         "part": "snippet",
         "type": "video",
         "order": SORT_OPTIONS[sort_by],
-        "maxResults": int(max_results),
     }
     
     if COUNTRIES[country]:
@@ -74,7 +185,6 @@ def search_videos(keyword, country, language, sort_by, date_filter, max_results)
     if LANGUAGES[language]:
         search_params["relevanceLanguage"] = LANGUAGES[language]
     
-    # 기간 필터
     if date_filter != "전체 기간":
         from datetime import datetime, timedelta
         now = datetime.utcnow()
@@ -88,110 +198,171 @@ def search_videos(keyword, country, language, sort_by, date_filter, max_results)
             after = now - timedelta(days=365)
         search_params["publishedAfter"] = after.strftime("%Y-%m-%dT%H:%M:%SZ")
     
-    response = youtube.search().list(**search_params).execute()
+    while len(all_items) < max_results:
+        search_params["maxResults"] = min(50, max_results - len(all_items))
+        if next_page_token:
+            search_params["pageToken"] = next_page_token
+        
+        response = youtube.search().list(**search_params).execute()
+        items = response.get("items", [])
+        
+        if not items:
+            break
+            
+        all_items.extend(items)
+        next_page_token = response.get("nextPageToken")
+        
+        if not next_page_token:
+            break
     
-    if not response.get("items"):
+    if not all_items:
         return "검색 결과가 없습니다."
     
-    # 비디오 ID 수집
-    video_ids = [item["id"]["videoId"] for item in response["items"]]
-    channel_ids = [item["snippet"]["channelId"] for item in response["items"]]
+    video_ids = [item["id"]["videoId"] for item in all_items]
+    channel_ids = list(set([item["snippet"]["channelId"] for item in all_items]))
     
-    # 비디오 상세 정보 (조회수, 좋아요 등)
-    video_details = youtube.videos().list(
-        id=",".join(video_ids),
-        part="statistics,contentDetails"
-    ).execute()
-    
-    # 채널 상세 정보 (구독자 수)
-    channel_details = youtube.channels().list(
-        id=",".join(list(set(channel_ids))),
-        part="statistics"
-    ).execute()
+    video_stats = {}
+    for i in range(0, len(video_ids), 50):
+        batch_ids = video_ids[i:i+50]
+        video_details = youtube.videos().list(
+            id=",".join(batch_ids),
+            part="statistics"
+        ).execute()
+        
+        for v in video_details.get("items", []):
+            stats = v["statistics"]
+            video_stats[v["id"]] = {
+                "views": int(stats.get("viewCount", 0)),
+                "views_fmt": format_count(int(stats.get("viewCount", 0))),
+                "likes": format_count(int(stats.get("likeCount", 0))),
+                "comments": format_count(int(stats.get("commentCount", 0))),
+            }
     
-    # 채널 구독자 매핑
     channel_subs = {}
-    for ch in channel_details.get("items", []):
-        sub_count = ch["statistics"].get("subscriberCount", "비공개")
-        if sub_count != "비공개":
-            sub_count = format_count(int(sub_count))
-        channel_subs[ch["id"]] = sub_count
+    for i in range(0, len(channel_ids), 50):
+        batch_ids = channel_ids[i:i+50]
+        channel_details = youtube.channels().list(
+            id=",".join(batch_ids),
+            part="statistics"
+        ).execute()
+        
+        for ch in channel_details.get("items", []):
+            sub_count = ch["statistics"].get("subscriberCount", "비공개")
+            if sub_count != "비공개":
+                sub_count = format_count(int(sub_count))
+            channel_subs[ch["id"]] = sub_count
     
-    # 비디오 통계 매핑
-    video_stats = {}
-    for v in video_details.get("items", []):
-        stats = v["statistics"]
-        video_stats[v["id"]] = {
-            "views": format_count(int(stats.get("viewCount", 0))),
-            "likes": format_count(int(stats.get("likeCount", 0))),
-            "comments": format_count(int(stats.get("commentCount", 0))),
-            "duration": v["contentDetails"]["duration"],
-        }
-    
-    # HTML 결과 생성
-    html = '<div style="display:flex; flex-direction:column; gap:15px;">'
-    
-    for i, item in enumerate(response["items"], 1):
+    html = f'''
+    <style>
+        .yt-table {{ width: 100%; border-collapse: collapse; font-size: 14px; }}
+        .yt-table th {{ background: #ff0000; color: white; padding: 12px 8px; text-align: left; position: sticky; top: 0; }}
+        .yt-table td {{ padding: 10px 8px; border-bottom: 1px solid #ddd; vertical-align: middle; }}
+        .yt-table tr:hover {{ background: #f5f5f5; }}
+        .yt-table img {{ border-radius: 4px; }}
+        .yt-table a {{ color: #1a0dab; text-decoration: none; }}
+        .yt-table a:hover {{ text-decoration: underline; }}
+        .yt-rank {{ font-weight: bold; color: #ff0000; text-align: center; }}
+        .yt-views {{ font-weight: bold; color: #065fd4; }}
+        .yt-subs {{ color: #606060; }}
+        .yt-date {{ color: #888; font-size: 12px; }}
+    </style>
+    <p><b>총 {len(all_items)}개 결과</b> | 검색어: "{keyword}" | 정렬: {sort_by}</p>
+    <div style="max-height: 800px; overflow-y: auto;">
+    <table class="yt-table">
+        <thead>
+            <tr>
+                <th style="width:50px;">순위</th>
+                <th style="width:120px;">썸네일</th>
+                <th>제목</th>
+                <th style="width:150px;">채널</th>
+                <th style="width:80px;">구독자</th>
+                <th style="width:80px;">조회수</th>
+                <th style="width:60px;">좋아요</th>
+                <th style="width:60px;">댓글</th>
+                <th style="width:90px;">게시일</th>
+            </tr>
+        </thead>
+        <tbody>
+    '''
+    
+    for i, item in enumerate(all_items, 1):
         video_id = item["id"]["videoId"]
         snippet = item["snippet"]
         channel_id = snippet["channelId"]
         
         title = snippet["title"]
+        if len(title) > 50:
+            title = title[:50] + "..."
         channel = snippet["channelTitle"]
-        thumbnail = snippet["thumbnails"]["medium"]["url"]
+        if len(channel) > 15:
+            channel = channel[:15] + "..."
+        thumbnail = snippet["thumbnails"]["default"]["url"]
         published = snippet["publishedAt"][:10]
         url = f"https://youtube.com/watch?v={video_id}"
+        channel_url = f"https://youtube.com/channel/{channel_id}"
         
         stats = video_stats.get(video_id, {})
-        views = stats.get("views", "N/A")
+        views = stats.get("views_fmt", "N/A")
         likes = stats.get("likes", "N/A")
         comments = stats.get("comments", "N/A")
         subs = channel_subs.get(channel_id, "N/A")
         
+        if i == 1:
+            rank_style = "color: #FFD700; font-size: 18px;"
+        elif i == 2:
+            rank_style = "color: #C0C0C0; font-size: 16px;"
+        elif i == 3:
+            rank_style = "color: #CD7F32; font-size: 16px;"
+        else:
+            rank_style = ""
+        
         html += f'''
-        <div style="display:flex; gap:15px; padding:10px; border:1px solid #ddd; border-radius:8px; background:#f9f9f9;">
-            <a href="{url}" target="_blank">
-                <img src="{thumbnail}" style="width:200px; border-radius:5px;">
-            </a>
-            <div style="flex:1;">
-                <a href="{url}" target="_blank" style="font-weight:bold; font-size:16px; color:#1a0dab; text-decoration:none;">
-                    {i}. {title}
-                </a>
-                <p style="margin:5px 0; color:#555;">
-                    <b>채널:</b> {channel} &nbsp;|&nbsp; <b>구독자:</b> {subs}
-                </p>
-                <p style="margin:5px 0; color:#333;">
-                    <b>조회수:</b> {views} &nbsp;|&nbsp; 
-                    <b>좋아요:</b> {likes} &nbsp;|&nbsp; 
-                    <b>댓글:</b> {comments}
-                </p>
-                <p style="margin:5px 0; color:#888; font-size:12px;">
-                    <b>게시일:</b> {published}
-                </p>
-            </div>
-        </div>
+            <tr>
+                <td class="yt-rank" style="{rank_style}">{i}</td>
+                <td><a href="{url}" target="_blank"><img src="{thumbnail}" width="100"></a></td>
+                <td><a href="{url}" target="_blank">{title}</a></td>
+                <td><a href="{channel_url}" target="_blank">{channel}</a></td>
+                <td class="yt-subs">{subs}</td>
+                <td class="yt-views">{views}</td>
+                <td>{likes}</td>
+                <td>{comments}</td>
+                <td class="yt-date">{published}</td>
+            </tr>
         '''
     
-    html += '</div>'
+    html += '</tbody></table></div>'
     return html
 
+initial_trending = get_default_trending()
+
 with gr.Blocks() as demo:
     gr.Markdown("## YouTube 인기 영상 검색기")
     
+    gr.Markdown("### 🔥 실시간 인기 검색어")
+    trending_radio = gr.Radio(
+        choices=initial_trending,
+        label="클릭하면 검색창에 자동 입력됩니다",
+        interactive=True
+    )
+    
+    gr.Markdown("---")
+    
     with gr.Row():
-        keyword = gr.Textbox(label="검색어", placeholder="검색어 입력...")
-        btn = gr.Button("검색", variant="primary")
+        keyword = gr.Textbox(label="검색어", placeholder="검색어 입력...", scale=3)
+        btn = gr.Button("검색", variant="primary", scale=1)
     
     with gr.Row():
-        country = gr.Dropdown(list(COUNTRIES.keys()), value="전세계", label="국가/지역")
-        language = gr.Dropdown(list(LANGUAGES.keys()), value="전체", label="언어")
+        country = gr.Dropdown(list(COUNTRIES.keys()), value="한국", label="국가 (인구순)")
+        language = gr.Dropdown(list(LANGUAGES.keys()), value="전체", label="언어 (사용인구순)")
         sort_by = gr.Dropdown(list(SORT_OPTIONS.keys()), value="조회수 순", label="정렬")
         date_filter = gr.Dropdown(list(DATE_OPTIONS.keys()), value="전체 기간", label="기간")
-        max_results = gr.Slider(5, 50, value=10, step=5, label="결과 수")
+        max_results = gr.Slider(10, 300, value=50, step=10, label="결과 수")
     
     output = gr.HTML()
     
+    trending_radio.change(use_trending_keyword, trending_radio, keyword)
+    country.change(update_trending, country, trending_radio)
     btn.click(search_videos, [keyword, country, language, sort_by, date_filter, max_results], output)
     keyword.submit(search_videos, [keyword, country, language, sort_by, date_filter, max_results], output)
 
-demo.launch()
+demo.launch()