Update app.py

app.py

@@ -4,7 +4,7 @@ import os
 from huggingface_hub import InferenceClient
 import requests
 import asyncio
-import subprocess
+from googleapiclient.discovery import build
 
 # 로깅 설정
 logging.basicConfig(level=logging.DEBUG, format='%(asctime)s:%(levelname)s:%(name)s:%(message)s', handlers=[logging.StreamHandler()])
@@ -23,6 +23,10 @@ hf_client = InferenceClient("CohereForAI/c4ai-command-r-plus", token=os.getenv("
 PEXELS_API_KEY = "5woz23MGx1QrSY0WHFb0BRi29JvbXPu97Hg0xnklYgHUI8G0w23FKH62"
 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"))
 
@@ -36,13 +40,11 @@ class MyClient(discord.Client):
 
     async def on_ready(self):
         logging.info(f'{self.user}로 로그인되었습니다!')
-        subprocess.Popen(["python", "web.py"])
-        logging.info("Web.py server has been started.")
 
         # 봇이 시작될 때 안내 메시지를 전송
         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:
@@ -58,11 +60,13 @@ class MyClient(discord.Client):
             if keywords:
                 # Pexels API로 고해상도 비디오 검색
                 video_urls = await search_videos(keywords)
-                if video_urls:
-                    # 요청자와의 쓰레드 생성 및 고해상도 비디오 전송
-                    await create_thread_and_send_videos(message, keywords, video_urls)
+                # YouTube API로 음악 검색
+                music_urls = await search_music_on_youtube(keywords)
+                if video_urls or music_urls:
+                    # 요청자와의 쓰레드 생성 및 결과 전송
+                    await create_thread_and_send_results(message, keywords, video_urls, music_urls)
                 else:
-                    await message.channel.send(f"**{keywords}**에 대한 고해상도 비디오를 찾을 수 없습니다.")
+                    await message.channel.send(f"**{keywords}**에 대한 비디오나 음악을 찾을 수 없습니다.")
             else:
                 await message.channel.send("키워드를 추출할 수 없습니다.")
         finally:
@@ -101,7 +105,7 @@ async def search_videos(keywords):
     }
     params = {
         "query": keywords,
-        "per_page": 40  # 최대 10개 비디오를 가져옵니다
+        "per_page": 10  # 최대 10개 비디오를 가져옵니다
     }
     response = requests.get(PEXELS_VIDEO_API_URL, headers=headers, params=params)
     if response.status_code == 200:
@@ -109,13 +113,32 @@ async def search_videos(keywords):
         return [video['video_files'][0]['link'] for video in data['videos']]
     return None
 
-async def create_thread_and_send_videos(message, keywords, video_urls):
+async def search_music_on_youtube(keywords):
+    response = youtube_service.search().list(
+        q=keywords,
+        part='snippet',
+        type='video',
+        maxResults=5,
+        videoCategoryId='10'  # 음악 카테고리 ID
+    ).execute()
+    music_urls = []
+    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):
     # 쓰레드 생성
     thread = await message.channel.create_thread(name=f"{message.author.name}의 검색 결과", message=message)
-    message_content = f"**{keywords}**에 대한 고해상도 비디오 {len(video_urls)}개를 찾았습니다:"
-    await thread.send(message_content)
-    for url in video_urls:
-        await thread.send(url)
+    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)}개를 찾았습니다:"
+        await thread.send(message_content)
+        for url in music_urls:
+            await thread.send(url)
 
 if __name__ == "__main__":
     discord_client = MyClient(intents=intents)