Update app.py

app.py

@@ -1,16 +1,16 @@
 import gradio as gr
 from googleapiclient.discovery import build
+from groq import Groq
 import os
 import sqlite3
 from datetime import datetime, timedelta
 import re
-import anthropic
 
 API_KEY = os.getenv("YOUTUBE_API_KEY")
-ANTHROPIC_API_KEY = os.getenv("ANTHROPIC_API_KEY")
+GROQ_API_KEY = os.getenv("GROQ_API_KEY")
 
 youtube = build("youtube", "v3", developerKey=API_KEY)
-claude_client = anthropic.Anthropic(api_key=ANTHROPIC_API_KEY) if ANTHROPIC_API_KEY else None
+groq_client = Groq(api_key=GROQ_API_KEY) if GROQ_API_KEY else None
 
 # ============================================
 # 🎨 Comic Classic Theme CSS
@@ -177,62 +177,71 @@ def format_count(count):
     return str(count)
 
 def get_ai_pick_rating(videos_data):
-    """Claude API로 AI Pick 등급 산정"""
-    if not claude_client or not videos_data:
-        # API 없으면 자체 알고리즘으로 계산
+    """Groq API (GPT-OSS-120B)로 AI Pick 등급 산정"""
+    if not groq_client or not videos_data:
         return calculate_local_rating(videos_data)
     
     try:
-        # 상위 100개만 분석 (API 비용 절감)
-        sample = videos_data[:100]
+        # 상위 50개만 분석 (토큰 제한)
+        sample = videos_data[:50]
         
         video_info = "\n".join([
-            f"#{i+1}. 제목: {v['title'][:50]}, 채널: {v['channel_name']}, "
-            f"조회수: {v['views']}, 좋아요: {v['likes']}, 댓글: {v['comments']}, "
-            f"구독자: {v.get('subs', 0)}"
+            f"#{i+1}. 제목:{v['title'][:40]}, 채널:{v['channel_name'][:15]}, "
+            f"조회수:{v['views']}, 좋아요:{v['likes']}, 댓글:{v['comments']}, 구독자:{v.get('subs',0)}"
             for i, v in enumerate(sample)
         ])
         
-        response = claude_client.messages.create(
-            model="claude-sonnet-4-20250514",
-            max_tokens=2000,
-            messages=[{
-                "role": "user",
-                "content": f"""다음 YouTube 영상 목록을 분석하여 각 영상의 종합 품질 등급을 매겨주세요.
+        prompt = f"""YouTube 영상 목록을 분석하여 각 영상의 종합 품질 등급을 매겨주세요.
 
 평가 기준:
 - 조회수 대비 좋아요/댓글 비율 (engagement)
 - 채널 구독자 대비 조회수 (viral potential)
-- 전반적인 인기도와 성장 가능성
+- 전반적인 인기도
 
-등급 (숫자로만 응답):
-0 = 없음 (평균 이하)
-1 = ♥ (평균)
-2 = ⭐ (좋음)
-3 = ⭐⭐ (매우 좋음)
-4 = ⭐⭐⭐ (최고)
+등급 (숫자로만):
+0=없음(평균이하), 1=♥(평균), 2=⭐(좋음), 3=⭐⭐(매우좋음), 4=⭐⭐⭐(최고)
 
-각 영상 번호와 등급만 쉼표로 구분하여 응답하세요.
-예시: 1:3,2:2,3:4,4:1,5:0
+각 영상 번호와 등급만 응답하세요.
+형식: 1:3,2:2,3:4,4:1,5:0
 
 영상 목록:
 {video_info}
 
 응답 (번호:등급 형식으로만):"""
-            }]
+
+        # Groq API 호출 (스트리밍)
+        completion = groq_client.chat.completions.create(
+            model="openai/gpt-oss-120b",
+            messages=[{"role": "user", "content": prompt}],
+            temperature=0.7,
+            max_completion_tokens=2000,
+            top_p=1,
+            stream=True,
+            stop=None
         )
         
+        # 스트리밍 응답 수집
+        result_text = ""
+        for chunk in completion:
+            if chunk.choices[0].delta.content:
+                result_text += chunk.choices[0].delta.content
+        
         # 응답 파싱
         ratings = {}
-        result_text = response.content[0].text.strip()
         pairs = re.findall(r'(\d+):(\d)', result_text)
         for idx, rating in pairs:
             ratings[int(idx)-1] = int(rating)
         
+        # 나머지 영상은 로컬 알고리즘으로
+        if len(videos_data) > 50:
+            local_ratings = calculate_local_rating(videos_data[50:])
+            for k, v in local_ratings.items():
+                ratings[k + 50] = v
+        
         return ratings
     
     except Exception as e:
-        print(f"AI Rating Error: {e}")
+        print(f"Groq AI Rating Error: {e}")
         return calculate_local_rating(videos_data)
 
 def calculate_local_rating(videos_data):
@@ -242,7 +251,6 @@ def calculate_local_rating(videos_data):
     if not videos_data:
         return ratings
     
-    # 통계 계산
     views_list = [v['views'] for v in videos_data if v['views'] > 0]
     if not views_list:
         return ratings
@@ -256,52 +264,35 @@ def calculate_local_rating(videos_data):
         comments = v['comments']
         subs = v.get('subs', 0)
         
-        # 스코어 계산
         score = 0
         
         # 조회수 점수 (0-40)
         if views > 0:
-            view_score = min(40, (views / max_views) * 40)
-            score += view_score
+            score += min(40, (views / max_views) * 40)
         
         # engagement 점수 (0-30)
         if views > 0:
             engagement = ((likes + comments * 2) / views) * 100
-            engagement_score = min(30, engagement * 3)
-            score += engagement_score
+            score += min(30, engagement * 3)
         
-        # viral 점수 (구독자 대비 조회수) (0-30)
+        # viral 점수 (0-30)
         if subs > 0 and views > 0:
-            viral = views / subs
-            viral_score = min(30, viral * 10)
-            score += viral_score
+            score += min(30, (views / subs) * 10)
         elif views > avg_views:
             score += 15
         
         # 등급 결정
-        if score >= 70:
-            ratings[i] = 4  # ⭐⭐⭐
-        elif score >= 50:
-            ratings[i] = 3  # ⭐⭐
-        elif score >= 30:
-            ratings[i] = 2  # ⭐
-        elif score >= 15:
-            ratings[i] = 1  # ♥
-        else:
-            ratings[i] = 0  # 없음
+        if score >= 70: ratings[i] = 4
+        elif score >= 50: ratings[i] = 3
+        elif score >= 30: ratings[i] = 2
+        elif score >= 15: ratings[i] = 1
+        else: ratings[i] = 0
     
     return ratings
 
 def get_rating_display(rating):
     """등급을 이모지로 변환"""
-    displays = {
-        0: "",
-        1: "♥",
-        2: "⭐",
-        3: "⭐⭐",
-        4: "⭐⭐⭐"
-    }
-    return displays.get(rating, "")
+    return {0: "", 1: "♥", 2: "⭐", 3: "⭐⭐", 4: "⭐⭐⭐"}.get(rating, "")
 
 def get_real_trending_keywords(region_code="KR", language="ko"):
     try:
@@ -484,7 +475,7 @@ def search_videos(keyword, country, language, sort_by, date_filter, max_results)
             "subs": channel_subs_raw.get(snip["channelId"], 0),
         })
     
-    # AI Pick 등급 계산
+    # AI Pick 등급 계산 (Groq GPT-OSS-120B)
     ai_ratings = get_ai_pick_rating(videos_data)
     
     # DB 저장
@@ -519,7 +510,7 @@ def search_videos(keyword, country, language, sort_by, date_filter, max_results)
     
     <div class="result-header">
         🎬 총 <b>{len(all_items)}개</b> 결과 | 🔍 "{keyword}" | 🌍 {country} | 📊 {sort_by}
-        <br><span style="font-size:0.9rem;">💡 헤더 클릭 = 정렬 | AI Pick: ♥(평균) ⭐(좋음) ⭐⭐(매우좋음) ⭐⭐⭐(최고)</span>
+        <br><span style="font-size:0.9rem;">💡 헤더 클릭 = 정렬 | 🤖 AI Pick (GPT-OSS-120B): ♥ ⭐ ⭐⭐ ⭐⭐⭐</span>
     </div>
     
     <div style="max-height:700px; overflow-y:auto; border:3px solid #1F2937; border-radius:8px;">
@@ -625,11 +616,10 @@ def search_videos(keyword, country, language, sort_by, date_filter, max_results)
     return html, stats_html
 
 def show_trending_alerts():
-    """급상승 알림 + 실시간 데이터 갱신"""
+    """급상승 알림"""
     conn = sqlite3.connect("youtube_data.db")
     c = conn.cursor()
     
-    # 최근 급상승 알림
     c.execute('''
         SELECT ta.video_id, v.title, v.channel_name, ta.old_value, ta.new_value, ta.change_percent, ta.detected_at
         FROM trending_alerts ta
@@ -640,14 +630,13 @@ def show_trending_alerts():
     conn.close()
     
     if not alerts:
-        # 알림 없으면 현재 인기 급상승 영상 직접 조회
         try:
             resp = youtube.videos().list(part="snippet,statistics", chart="mostPopular", regionCode="KR", maxResults=20).execute()
             
             html = '''
-            <div style="background:#FACC15; padding:15px; border:3px solid #1F2937; border-radius:8px; margin-bottom:15px;">
+            <div style="background:#FACC15; padding:15px; border:3px solid #1F2937; border-radius:8px; margin-bottom:15px; font-family:'Comic Neue',cursive;">
                 <h3 style="font-family:'Bangers',cursive; margin:0;">🔥 현재 급상승 인기 영상 (한국)</h3>
-                <p style="margin:5px 0 0 0; font-size:0.9rem;">데이터 축적 후 변화율 기반 알림이 표시됩니다</p>
+                <p style="margin:5px 0 0 0; font-size:0.9rem;">검색을 여러 번 실행하면 변화율 기반 알림이 표시됩니다</p>
             </div>
             <div style="display:flex; flex-wrap:wrap; gap:15px;">
             '''
@@ -655,24 +644,23 @@ def show_trending_alerts():
             for i, item in enumerate(resp.get("items", [])[:20], 1):
                 snip = item["snippet"]
                 stats = item["statistics"]
-                vid = item["id"]
                 title = snip["title"][:30] + "..." if len(snip["title"]) > 30 else snip["title"]
                 views = format_count(int(stats.get("viewCount", 0)))
-                url = f"https://youtube.com/watch?v={vid}"
+                url = f"https://youtube.com/watch?v={item['id']}"
                 thumb = snip["thumbnails"]["medium"]["url"]
                 
                 html += f'''
-                <div style="width:180px; background:#FFF; border:3px solid #1F2937; border-radius:8px; padding:10px; box-shadow:4px 4px 0 #1F2937;">
+                <div style="width:180px; background:#FFF; border:3px solid #1F2937; border-radius:8px; padding:10px; box-shadow:4px 4px 0 #1F2937; font-family:'Comic Neue',cursive;">
                     <a href="{url}" target="_blank"><img src="{thumb}" style="width:100%; border-radius:5px; border:2px solid #1F2937;"></a>
                     <p style="margin:8px 0 5px 0; font-size:12px; font-weight:700; color:#1F2937;">{i}. {title}</p>
-                    <p style="margin:0; font-size:14px; color:#EF4444; font-weight:700;">👀 {views}</p>
+                    <p style="margin:0; font-size:14px; color:#EF4444; font-weight:700; font-family:'Bangers',cursive;">👀 {views}</p>
                 </div>
                 '''
             
             html += '</div>'
             return html
         except Exception as e:
-            return f"<p>데이터 로드 오류: {e}</p>"
+            return f"<p style='font-family:Comic Neue,cursive;'>데이터 로드 오류: {e}</p>"
     
     html = '''
     <div style="background:#EF4444; color:white; padding:15px; border:3px solid #1F2937; border-radius:8px; margin-bottom:15px;">
@@ -681,11 +669,8 @@ def show_trending_alerts():
     <table style="width:100%; border-collapse:collapse; font-family:'Comic Neue',cursive;">
         <thead><tr style="background:#1F2937; color:#FACC15;">
             <th style="padding:10px; border:2px solid #1F2937;">영상</th>
-            <th style="padding:10px;">채널</th>
-            <th style="padding:10px;">이전</th>
-            <th style="padding:10px;">현재</th>
-            <th style="padding:10px;">증가율</th>
-            <th style="padding:10px;">감지시간</th>
+            <th style="padding:10px;">채널</th><th style="padding:10px;">이전</th>
+            <th style="padding:10px;">현재</th><th style="padding:10px;">증가율</th><th style="padding:10px;">감지시간</th>
         </tr></thead><tbody>
     '''
     
@@ -707,7 +692,7 @@ def show_trending_alerts():
     return html
 
 def show_top_growing():
-    """급성장 TOP - DB + 실시간 조회"""
+    """급성장 TOP"""
     conn = sqlite3.connect("youtube_data.db")
     c = conn.cursor()
     cutoff = (datetime.now() - timedelta(hours=48)).isoformat()
@@ -727,26 +712,22 @@ def show_top_growing():
     conn.close()
     
     if not results:
-        # DB 없으면 실시간 인기 영상 표시
         try:
             resp = youtube.videos().list(part="snippet,statistics", chart="mostPopular", regionCode="KR", maxResults=20).execute()
             
             html = '''
             <div style="background:#3B82F6; color:white; padding:15px; border:3px solid #1F2937; border-radius:8px; margin-bottom:15px;">
                 <h3 style="font-family:'Bangers',cursive; margin:0;">📈 현재 인기 급상승 TOP 20</h3>
-                <p style="margin:5px 0 0 0; font-size:0.9rem;">검색을 여러 번 수행하면 성장률 데이터가 축적됩니다</p>
+                <p style="margin:5px 0 0 0; font-size:0.9rem; font-family:'Comic Neue',cursive;">검색을 여러 번 실행하면 성장률 데이터가 축적됩니다</p>
             </div>
             <div style="display:flex; flex-wrap:wrap; gap:15px;">
             '''
             
-            items = resp.get("items", [])
-            for i, item in enumerate(items[:20], 1):
+            for i, item in enumerate(resp.get("items", [])[:20], 1):
                 snip = item["snippet"]
                 stats = item["statistics"]
                 views = int(stats.get("viewCount", 0))
                 likes = int(stats.get("likeCount", 0))
-                
-                # 간단한 engagement 계산
                 engagement = (likes / views * 100) if views > 0 else 0
                 
                 title = snip["title"][:25] + "..." if len(snip["title"]) > 25 else snip["title"]
@@ -754,11 +735,11 @@ def show_top_growing():
                 thumb = snip["thumbnails"]["medium"]["url"]
                 
                 html += f'''
-                <div style="width:180px; background:#FFF; border:3px solid #1F2937; border-radius:8px; padding:10px; box-shadow:4px 4px 0 #1F2937;">
+                <div style="width:180px; background:#FFF; border:3px solid #1F2937; border-radius:8px; padding:10px; box-shadow:4px 4px 0 #1F2937; font-family:'Comic Neue',cursive;">
                     <a href="{url}" target="_blank"><img src="{thumb}" style="width:100%; border-radius:5px; border:2px solid #1F2937;"></a>
                     <p style="margin:8px 0 5px 0; font-size:12px; font-weight:700;">{i}. {title}</p>
                     <p style="margin:0; font-size:11px; color:#666;">{snip["channelTitle"][:15]}</p>
-                    <p style="margin:5px 0 0 0; font-size:14px; color:#3B82F6; font-weight:700;">👀 {format_count(views)}</p>
+                    <p style="margin:5px 0 0 0; font-size:14px; color:#3B82F6; font-weight:700; font-family:'Bangers',cursive;">👀 {format_count(views)}</p>
                     <p style="margin:2px 0 0 0; font-size:12px; color:#EF4444;">❤️ {engagement:.2f}%</p>
                 </div>
                 '''
@@ -780,7 +761,7 @@ def show_top_growing():
         url = f"https://youtube.com/watch?v={vid}"
         
         html += f'''
-        <div style="width:180px; background:#FFF; border:3px solid #1F2937; border-radius:8px; padding:10px; box-shadow:4px 4px 0 #1F2937;">
+        <div style="width:180px; background:#FFF; border:3px solid #1F2937; border-radius:8px; padding:10px; box-shadow:4px 4px 0 #1F2937; font-family:'Comic Neue',cursive;">
             <a href="{url}" target="_blank"><img src="{thumb}" style="width:100%; border-radius:5px; border:2px solid #1F2937;"></a>
             <p style="margin:8px 0 5px 0; font-size:12px; font-weight:700;">{i}. {title}</p>
             <p style="margin:0; font-size:11px; color:#666;">{channel[:15]}</p>
@@ -886,3 +867,4 @@ with gr.Blocks(css=css) as demo:
     history_btn.click(show_search_history, outputs=history_out)
 
 demo.launch()
+