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_VIDEO_URL = "https://api.pexels.com/videos/search"
+PEXELS_VIDEO_API_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로 고해상도 동영상 검색
+                # Pexels API로 고해상도 비디오 검색
                 video_urls = await search_videos(keywords)
                 if video_urls:
-                    # 요청자와의 쓰레드 생성 또는 기존 쓰레드 사용
-                    await create_or_use_thread_and_send_videos(message, keywords, 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:
@@ -101,46 +101,22 @@ async def search_videos(keywords):
     }
     params = {
         "query": keywords,
-        "per_page": 20  # 최대 20개 동영상을 가져옵니다
+        "per_page": 10  # 최대 10개 비디오를 가져옵니다
     }
-    response = requests.get(PEXELS_API_VIDEO_URL, headers=headers, params=params)
+    response = requests.get(PEXELS_VIDEO_API_URL, headers=headers, params=params)
     if response.status_code == 200:
         data = response.json()
-        video_urls = []
-        for video in data['videos']:
-            highest_quality_video = max(video['video_files'], key=lambda x: x['width'] * x['height'])
-            video_urls.append(highest_quality_video['link'])
-        return video_urls
+        return [video['video_files'][0]['link'] for video in data['videos']]
     return None
 
-async def create_or_use_thread_and_send_videos(message, keywords, video_urls):
-    # 메시지에 대해 이미 생성된 쓰레드를 찾음
-    thread = await find_existing_thread(message)
-    
-    if not thread:
-        # 새 쓰레드 생성
-        thread = await message.channel.create_thread(name=f"{message.author.name}의 검색 결과", message=message)
-    
-    message_content = f"**{keywords}**에 대한 고해상도 동영상 {len(video_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(video_urls)}개를 찾았습니다:"
     await thread.send(message_content)
     for url in video_urls:
         await thread.send(url)
 
-async def find_existing_thread(message):
-    # 채널의 모든 활성 쓰레드 가져오기
-    active_threads = await message.channel.active_threads()
-    for thread in active_threads:
-        if thread.starter_message_id == message.id:
-            return thread
-    
-    # 채널의 모든 보관된 쓰레드 가져오기
-    archived_threads = await message.channel.archived_threads().flatten()
-    for thread in archived_threads:
-        if thread.starter_message_id == message.id:
-            return thread
-
-    return None
-
 if __name__ == "__main__":
     discord_client = MyClient(intents=intents)
     discord_client.run(os.getenv('DISCORD_TOKEN'))