import discord
import logging
import os
from huggingface_hub import InferenceClient
import asyncio
import subprocess  # 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

# 추론 API 클라이언트 설정
hf_client = InferenceClient("CohereForAI/c4ai-command-r-plus", token=os.getenv("HF_TOKEN"))

# 특정 채널 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}로 로그인되었습니다!')
        # web.py를 새로운 프로세스로 실행합니다.
        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):
    # (중략: 기존의 generate_response 함수 로직 유지)

if __name__ == "__main__":
    discord_client = MyClient(intents=intents)
    discord_client.run(os.getenv('DISCORD_TOKEN'))