Update app.py

app.py

@@ -5,33 +5,193 @@ import os
 API_KEY = os.getenv("YOUTUBE_API_KEY")
 youtube = build("youtube", "v3", developerKey=API_KEY)
 
-def search_videos(keyword, max_results=10):
-    request = youtube.search().list(
-        q=keyword,
-        part="snippet",
-        type="video",
-        order="viewCount",  # 조회수 순
-        maxResults=max_results
-    )
-    response = request.execute()
-    
-    results = []
-    for item in response["items"]:
+# 국가 코드
+COUNTRIES = {
+    "전세계": "",
+    "한국": "KR",
+    "미국": "US",
+    "일본": "JP",
+    "영국": "GB",
+    "독일": "DE",
+    "프랑스": "FR",
+    "인도": "IN",
+    "브라질": "BR",
+}
+
+# 언어 코드
+LANGUAGES = {
+    "전체": "",
+    "한국어": "ko",
+    "영어": "en",
+    "일본어": "ja",
+    "중국어": "zh",
+    "스페인어": "es",
+    "독일어": "de",
+    "프랑스어": "fr",
+}
+
+# 정렬 기준
+SORT_OPTIONS = {
+    "조회수 순": "viewCount",
+    "최신순": "date",
+    "관련성 순": "relevance",
+    "평점 순": "rating",
+}
+
+# 기간 필터
+DATE_OPTIONS = {
+    "전체 기간": "",
+    "오늘": "today",
+    "이번 주": "thisWeek",
+    "이번 달": "thisMonth",
+    "올해": "thisYear",
+}
+
+def format_count(count):
+    if count >= 1000000000:
+        return f"{count/1000000000:.1f}B"
+    elif count >= 1000000:
+        return f"{count/1000000:.1f}M"
+    elif count >= 1000:
+        return f"{count/1000:.1f}K"
+    return str(count)
+
+def search_videos(keyword, country, language, sort_by, date_filter, max_results):
+    if not keyword.strip():
+        return "검색어를 입력하세요."
+    
+    # 검색 요청
+    search_params = {
+        "q": keyword,
+        "part": "snippet",
+        "type": "video",
+        "order": SORT_OPTIONS[sort_by],
+        "maxResults": int(max_results),
+    }
+    
+    if COUNTRIES[country]:
+        search_params["regionCode"] = COUNTRIES[country]
+    if LANGUAGES[language]:
+        search_params["relevanceLanguage"] = LANGUAGES[language]
+    
+    # 기간 필터
+    if date_filter != "전체 기간":
+        from datetime import datetime, timedelta
+        now = datetime.utcnow()
+        if date_filter == "오늘":
+            after = now - timedelta(days=1)
+        elif date_filter == "이번 주":
+            after = now - timedelta(weeks=1)
+        elif date_filter == "이번 달":
+            after = now - timedelta(days=30)
+        elif date_filter == "올해":
+            after = now - timedelta(days=365)
+        search_params["publishedAfter"] = after.strftime("%Y-%m-%dT%H:%M:%SZ")
+    
+    response = youtube.search().list(**search_params).execute()
+    
+    if not response.get("items"):
+        return "검색 결과가 없습니다."
+    
+    # 비디오 ID 수집
+    video_ids = [item["id"]["videoId"] for item in response["items"]]
+    channel_ids = [item["snippet"]["channelId"] for item in response["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()
+    
+    # 채널 구독자 매핑
+    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
+    
+    # 비디오 통계 매핑
+    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):
         video_id = item["id"]["videoId"]
-        title = item["snippet"]["title"]
-        channel = item["snippet"]["channelTitle"]
+        snippet = item["snippet"]
+        channel_id = snippet["channelId"]
+        
+        title = snippet["title"]
+        channel = snippet["channelTitle"]
+        thumbnail = snippet["thumbnails"]["medium"]["url"]
+        published = snippet["publishedAt"][:10]
         url = f"https://youtube.com/watch?v={video_id}"
-        results.append(f"**{title}**\n채널: {channel}\n{url}\n")
+        
+        stats = video_stats.get(video_id, {})
+        views = stats.get("views", "N/A")
+        likes = stats.get("likes", "N/A")
+        comments = stats.get("comments", "N/A")
+        subs = channel_subs.get(channel_id, "N/A")
+        
+        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>
+        '''
     
-    return "\n".join(results)
+    html += '</div>'
+    return html
 
 with gr.Blocks() as demo:
-    gr.Markdown("## YouTube 인기 영상 검색")
-    keyword = gr.Textbox(label="검색어")
-    count = gr.Slider(5, 50, value=10, step=5, label="결과 수")
-    btn = gr.Button("검색")
-    output = gr.Markdown()
+    gr.Markdown("## YouTube 인기 영상 검색기")
+    
+    with gr.Row():
+        keyword = gr.Textbox(label="검색어", placeholder="검색어 입력...")
+        btn = gr.Button("검색", variant="primary")
+    
+    with gr.Row():
+        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="결과 수")
+    
+    output = gr.HTML()
     
-    btn.click(search_videos, [keyword, count], output)
+    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()