Spaces:
Runtime error
Runtime error
import discord | |
import logging | |
import os | |
from huggingface_hub import InferenceClient | |
import requests | |
import asyncio | |
import subprocess | |
# λ‘κΉ μ€μ | |
logging.basicConfig(level=logging.DEBUG, format='%(asctime)s:%(levelname)s:%(name)s:%(message)s', handlers=[logging.StreamHandler()]) | |
# μΈν νΈ μ€μ | |
intents = discord.Intents.default() | |
intents.message_content = True | |
intents.messages = True | |
intents.guilds = True | |
intents.guild_messages = True | |
# μΆλ‘ API ν΄λΌμ΄μΈνΈ μ€μ | |
hf_client = InferenceClient("CohereForAI/c4ai-command-r-plus", token=os.getenv("HF_TOKEN")) | |
# Pexels API ν€ μ€μ | |
PEXELS_API_KEY = "5woz23MGx1QrSY0WHFb0BRi29JvbXPu97Hg0xnklYgHUI8G0w23FKH62" | |
PEXELS_API_URL = "https://api.pexels.com/v1/search" | |
# νΉμ μ±λ ID | |
SPECIFIC_CHANNEL_ID = int(os.getenv("DISCORD_CHANNEL_ID")) | |
# λν νμ€ν 리λ₯Ό μ μ₯ν μ μ λ³μ | |
conversation_history = [] | |
class MyClient(discord.Client): | |
def __init__(self, *args, **kwargs): | |
super().__init__(*args, **kwargs) | |
self.is_processing = False | |
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("μ°Ύκ³ μΆμ μ¬μ§μ λν μ€λͺ μ ν λ¬Έμ₯ λ¨μλ‘ μ λ ₯νμΈμ. μ) λλ°μ λ°μ΄ λκ³ μλ νμ€ν€") | |
async def on_message(self, message): | |
if message.author == self.user: | |
return | |
if not self.is_message_in_specific_channel(message): | |
return | |
if self.is_processing: | |
return | |
self.is_processing = True | |
try: | |
# μλ―Έ λΆμνμ¬ μλ¬Έ ν€μλ μΆμΆ | |
keywords = await extract_keywords(message) | |
if keywords: | |
# Pexels APIλ‘ κ³ ν΄μλ μ΄λ―Έμ§ κ²μ | |
image_urls = await search_images(keywords) | |
if image_urls: | |
# λμ€μ½λ μ±λμ μ΄λ―Έμ§ μ μ‘ | |
await send_images(message.channel, keywords, image_urls) | |
else: | |
await message.channel.send(f"**{keywords}**μ λν κ³ ν΄μλ μ΄λ―Έμ§λ₯Ό μ°Ύμ μ μμ΅λλ€.") | |
else: | |
await message.channel.send("ν€μλλ₯Ό μΆμΆν μ μμ΅λλ€.") | |
finally: | |
self.is_processing = False | |
def is_message_in_specific_channel(self, message): | |
# λ©μμ§κ° μ§μ λ μ±λμ΄κ±°λ, ν΄λΉ μ±λμ μ°λ λμΈ κ²½μ° True λ°ν | |
return message.channel.id == SPECIFIC_CHANNEL_ID or ( | |
isinstance(message.channel, discord.Thread) and message.channel.parent_id == SPECIFIC_CHANNEL_ID | |
) | |
async def extract_keywords(message): | |
user_input = message.content | |
system_prompt = "λ€μ λ¬Έμ₯μ μλ―Έμ λ§λ μλ¬Έ ν€μλλ₯Ό μΆμΆνμΈμ: " | |
logging.debug(f'Extracting keywords from user input: {user_input}') | |
messages = [{"role": "system", "content": system_prompt + user_input}] | |
logging.debug(f'Messages to be sent to the model: {messages}') | |
loop = asyncio.get_event_loop() | |
response = await loop.run_in_executor(None, lambda: hf_client.chat_completion( | |
messages, max_tokens=10, temperature=0.7, top_p=0.85)) | |
# LLM μλ΅μ ν μ€νΈμΌ μ μμΌλ―λ‘ λ¨μν κ²°ν©νμ¬ λ°ν | |
keywords = response.strip() | |
logging.debug(f'Extracted keywords: {keywords}') | |
return keywords | |
async def search_images(keywords): | |
headers = { | |
"Authorization": PEXELS_API_KEY | |
} | |
params = { | |
"query": keywords, | |
"per_page": 20 # μ΅λ 20κ° μ΄λ―Έμ§λ₯Ό κ°μ Έμ΅λλ€ | |
} | |
response = requests.get(PEXELS_API_URL, headers=headers, params=params) | |
if response.status_code == 200: | |
data = response.json() | |
return [photo['src']['large2x'] for photo in data['photos']] | |
return None | |
async def send_images(channel, keywords, image_urls): | |
message_content = f"**{keywords}**μ λν κ³ ν΄μλ μ΄λ―Έμ§ {len(image_urls)}μ₯μ μ°Ύμμ΅λλ€:" | |
await channel.send(message_content) | |
for url in image_urls: | |
await channel.send(url) | |
if __name__ == "__main__": | |
discord_client = MyClient(intents=intents) | |
discord_client.run(os.getenv('DISCORD_TOKEN')) | |