Update app.py

app.py

@@ -15,6 +15,7 @@ intents.message_content = True
 intents.messages = True
 intents.guilds = True
 intents.guild_messages = True
+intents.threads = True  # 스레드 권한 추가
 
 # 추론 API 클라이언트 설정
 hf_client = InferenceClient("CohereForAI/c4ai-command-r-plus", token=os.getenv("HF_TOKEN"))
@@ -57,10 +58,10 @@ class MyClient(discord.Client):
             keywords = await extract_keywords(message)
             if keywords:
                 # Pexels API로 고해상도 이미지 검색
-                image_infos = await search_images(keywords)
-                if image_infos:
-                    # 디스코드 채널에 이미지 전송
-                    await send_images(message.channel, keywords, image_infos)
+                image_urls = await search_images(keywords)
+                if image_urls:
+                    # 요청자와의 쓰레드 생성 및 고해상도 이미지 전송
+                    await create_thread_and_send_images(message, keywords, image_urls)
                 else:
                     await message.channel.send(f"**{keywords}**에 대한 고해상도 이미지를 찾을 수 없습니다.")
             else:
@@ -106,31 +107,16 @@ async def search_images(keywords):
     response = requests.get(PEXELS_API_URL, headers=headers, params=params)
     if response.status_code == 200:
         data = response.json()
-        return [{
-            "url": photo['src']['large2x'],
-            "thumbnail": photo['src']['medium'],  # 썸네일을 medium 사이즈로 사용
-            "title": photo['alt']
-        } for photo in data['photos']]
+        return [photo['src']['large2x'] for photo in data['photos']]
     return None
 
-async def send_images(channel, keywords, image_infos):
-    message_content = f"**{keywords}**에 대한 고해상도 이미지 {len(image_infos)}장을 찾았습니다:"
-    await channel.send(message_content)
-    embeds = []
-
-    for info in image_infos:
-        embed = discord.Embed(title=info["title"], url=info["url"], description=f"[고해상도 이미지 보기]({info['url']})")
-        embed.set_thumbnail(url=info["thumbnail"])
-        embeds.append(embed)
-
-        # 디스코드 임베드 메시지는 한 번에 최대 10개까지 보내므로 한 번에 보내기
-        if len(embeds) == 10:
-            await channel.send(embeds=embeds)
-            embeds = []
-
-    # 남은 임베드 전송
-    if embeds:
-        await channel.send(embeds=embeds)
+async def create_thread_and_send_images(message, keywords, image_urls):
+    # 쓰레드 생성
+    thread = await message.channel.create_thread(name=f"{message.author.name}의 검색 결과", message=message)
+    message_content = f"**{keywords}**에 대한 고해상도 이미지 {len(image_urls)}장을 찾았습니다:"
+    await thread.send(message_content)
+    for url in image_urls:
+        await thread.send(url)
 
 if __name__ == "__main__":
     discord_client = MyClient(intents=intents)