|
import discord |
|
import logging |
|
import os |
|
from huggingface_hub import InferenceClient |
|
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 |
|
|
|
|
|
hf_client = InferenceClient("CohereForAI/c4ai-command-r-plus", token=os.getenv("HF_TOKEN")) |
|
|
|
|
|
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.") |
|
|
|
async def on_message(self, message): |
|
if message.author == self.user: |
|
return |
|
if message.channel.id != SPECIFIC_CHANNEL_ID: |
|
return |
|
if self.is_processing: |
|
return |
|
self.is_processing = True |
|
try: |
|
response = await generate_response(message.content) |
|
await message.channel.send(response) |
|
finally: |
|
self.is_processing = False |
|
|
|
async def generate_response(user_input): |
|
|
|
|
|
if __name__ == "__main__": |
|
discord_client = MyClient(intents=intents) |
|
discord_client.run(os.getenv('DISCORD_TOKEN')) |
|
|