Update app.py

app.py

@@ -1,9 +1,8 @@
 import discord
 import logging
 import os
-from huggingface_hub import InferenceClient
-import requests
 import asyncio
+from huggingface_hub import InferenceClient
 from googleapiclient.discovery import build
 from dotenv import load_dotenv
 
@@ -23,17 +22,12 @@ intents.guild_messages = True
 # 추론 API 클라이언트 설정
 hf_client = InferenceClient("CohereForAI/c4ai-command-r-plus", token=os.getenv("HF_TOKEN"))
 
-# Pexels API 키 설정
-PEXELS_API_KEY = os.getenv("PEXELS_API_KEY")
-PEXELS_VIDEO_API_URL = "https://api.pexels.com/videos/search"
-
 # YouTube API 키 설정
 YOUTUBE_API_KEY = os.getenv("YOUTUBE_API_KEY")
 youtube_service = build('youtube', 'v3', developerKey=YOUTUBE_API_KEY)
 
 # 특정 채널 ID
 SPECIFIC_CHANNEL_ID = int(os.getenv("DISCORD_CHANNEL_ID"))
-YOUTUBE_MUSIC_CHANNEL_ID = os.getenv("YOUTUBE_MUSIC_CHANNEL_ID")
 
 # 대화 히스토리를 저장할 전역 변수
 conversation_history = []
@@ -49,7 +43,7 @@ class MyClient(discord.Client):
         # 봇이 시작될 때 안내 메시지를 전송
         channel = self.get_channel(SPECIFIC_CHANNEL_ID)
         if channel:
-            await channel.send("찾고싶은 비디오나 음악에 대한 설명을 한 문장 단위로 입력하세요. 예) 바다에서 서핑하는 사람 또는 흥겨운 여름 음악")
+            await channel.send("트렌드 보고 싶은 주제를 입력하세요. 예) 최신 테크 뉴스")
 
     async def on_message(self, message):
         if message.author == self.user:
@@ -60,18 +54,16 @@ class MyClient(discord.Client):
             return
         self.is_processing = True
         try:
-            # 의미 분석하여 영문 키워드 추출
+            # 주제 키워드 추출
             keywords = await extract_keywords(message)
             if keywords:
-                # Pexels API로 고해상도 비디오 검색
-                video_urls = await search_videos(keywords)
-                # YouTube API로 음악 검색
-                music_urls = await search_music_on_youtube(keywords)
-                if video_urls or music_urls:
+                # YouTube API로 최신 트렌드 및 인기 비디오 검색
+                video_details = await search_trending_videos(keywords)
+                if video_details:
                     # 요청자와의 쓰레드 생성 및 결과 전송
-                    await create_thread_and_send_results(message, keywords, video_urls, music_urls)
+                    await create_thread_and_send_results(message, keywords, video_details)
                 else:
-                    await message.channel.send(f"**{keywords}**에 대한 비디오나 음악을 찾을 수 없습니다.")
+                    await message.channel.send(f"**{keywords}**에 대한 최신 트렌드 비디오를 찾을 수 없습니다.")
             else:
                 await message.channel.send("키워드를 추출할 수 없습니다.")
         finally:
@@ -104,46 +96,35 @@ async def extract_keywords(message):
     logging.debug(f'Extracted keywords: {keywords}')
     return keywords
 
-async def search_videos(keywords):
-    headers = {
-        "Authorization": PEXELS_API_KEY
-    }
-    params = {
-        "query": keywords,
-        "per_page": 10  # 최대 10개 비디오를 가져옵니다
-    }
-    response = requests.get(PEXELS_VIDEO_API_URL, headers=headers, params=params)
-    if response.status_code == 200:
-        data = response.json()
-        return [video['video_files'][0]['link'] for video in data['videos']]
-    return None
-
-async def search_music_on_youtube(keywords):
+async def search_trending_videos(keywords):
     response = youtube_service.search().list(
         q=keywords,
         part='snippet',
         type='video',
-        maxResults=5,
-        channelId=YOUTUBE_MUSIC_CHANNEL_ID  # 특정 채널 ID로 검색
+        order='viewCount',  # 인기 있는 비디오 기준으로 정렬
+        maxResults=5  # 원하는 결과 수
     ).execute()
-    music_urls = []
+    
+    video_details = []
     for item in response.get('items', []):
-        music_urls.append(f"https://www.youtube.com/watch?v={item['id']['videoId']}")
-    return music_urls
-
-async def create_thread_and_send_results(message, keywords, video_urls, music_urls):
+        video_details.append({
+            'title': item['snippet']['title'],
+            'url': f"https://www.youtube.com/watch?v={item['id']['videoId']}",
+            'description': item['snippet']['description'],
+            'thumbnail': item['snippet']['thumbnails']['high']['url']
+        })
+    return video_details
+
+async def create_thread_and_send_results(message, keywords, video_details):
     # 쓰레드 생성
-    thread = await message.channel.create_thread(name=f"{message.author.name}의 검색 결과", message=message)
-    if video_urls:
-        message_content = f"**{keywords}**에 대한 고해상도 비디오 {len(video_urls)}개를 찾았습니다:"
-        await thread.send(message_content)
-        for url in video_urls:
-            await thread.send(url)
-    if music_urls:
-        message_content = f"**{keywords}**에 대한 음악 {len(music_urls)}개를 찾았습니다:"
+    thread = await message.channel.create_thread(name=f"{message.author.name}의 트렌드 검색 결과", message=message)
+    if video_details:
+        message_content = f"**{keywords}**에 대한 최신 트렌드 비디오 {len(video_details)}개를 찾았습니다:"
         await thread.send(message_content)
-        for url in music_urls:
-            await thread.send(url)
+        for video in video_details:
+            embed = discord.Embed(title=video['title'], description=video['description'], url=video['url'])
+            embed.set_thumbnail(url=video['thumbnail'])
+            await thread.send(embed=embed)
 
 if __name__ == "__main__":
     discord_client = MyClient(intents=intents)