Update app.py

app.py

@@ -21,7 +21,7 @@ hf_client = InferenceClient("CohereForAI/c4ai-command-r-plus", token=os.getenv("
 
 # Pexels API 키 설정
 PEXELS_API_KEY = "5woz23MGx1QrSY0WHFb0BRi29JvbXPu97Hg0xnklYgHUI8G0w23FKH62"
-PEXELS_API_URL = "https://api.pexels.com/v1/search"
+PEXELS_API_VIDEO_URL = "https://api.pexels.com/videos/search"
 
 # 특정 채널 ID
 SPECIFIC_CHANNEL_ID = int(os.getenv("DISCORD_CHANNEL_ID"))
@@ -42,7 +42,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:
@@ -56,13 +56,13 @@ class MyClient(discord.Client):
             # 의미 분석하여 영문 키워드 추출
             keywords = await extract_keywords(message)
             if keywords:
-                # Pexels API로 고해상도 이미지 검색
-                image_urls = await search_images(keywords)
-                if image_urls:
-                    # 요청자와의 쓰레드 생성 및 고해상도 이미지 전송
-                    await create_thread_and_send_images(message, keywords, image_urls)
+                # Pexels API로 동영상 검색
+                video_urls = await search_videos(keywords)
+                if video_urls:
+                    # 요청자와의 쓰레드 생성 및 동영상 전송
+                    await create_thread_and_send_videos(message, keywords, video_urls)
                 else:
-                    await message.channel.send(f"**{keywords}**에 대한 고해상도 이미지를 찾을 수 없습니다.")
+                    await message.channel.send(f"**{keywords}**에 대한 동영상을 찾을 수 없습니다.")
             else:
                 await message.channel.send("키워드를 추출할 수 없습니다.")
         finally:
@@ -95,26 +95,26 @@ async def extract_keywords(message):
     logging.debug(f'Extracted keywords: {keywords}')
     return keywords
 
-async def search_images(keywords):
+async def search_videos(keywords):
     headers = {
         "Authorization": PEXELS_API_KEY
     }
     params = {
         "query": keywords,
-        "per_page": 80  # 최대 20개 이미지를 가져옵니다
+        "per_page": 80  # 최대 20개 동영상을 가져옵니다
     }
-    response = requests.get(PEXELS_API_URL, headers=headers, params=params)
+    response = requests.get(PEXELS_API_VIDEO_URL, headers=headers, params=params)
     if response.status_code == 200:
         data = response.json()
-        return [photo['src']['large2x'] for photo in data['photos']]
+        return [video['video_files'][0]['link'] for video in data['videos']]
     return None
 
-async def create_thread_and_send_images(message, keywords, image_urls):
+async def create_thread_and_send_videos(message, keywords, video_urls):
     # 쓰레드 생성
     thread = await message.channel.create_thread(name=f"{message.author.name}의 검색 결과", message=message)
-    message_content = f"**{keywords}**에 대한 고해상도 이미지 {len(image_urls)}장을 찾았습니다:"
+    message_content = f"**{keywords}**에 대한 동영상 {len(video_urls)}개를 찾았습니다:"
     await thread.send(message_content)
-    for url in image_urls:
+    for url in video_urls:
         await thread.send(url)
 
 if __name__ == "__main__":