import discord
import os
import openai

intents = discord.Intents.default()
intents.message_content = True

# Discord 봇 토큰
TOKEN = 'YOUR_DISCORD_BOT_TOKEN'

# OpenAI API 키
OPENAI_API_KEY = 'YOUR_OPENAI_API_KEY'

# Discord 봇 클라이언트 생성
client = discord.Client(intents=intents)

# 봇이 준비되었을 때 실행되는 이벤트 핸들러
@client.event
async def on_ready():
    print(f'Logged in as {client.user.name} - {client.user.id}')

# 메시지가 도착했을 때 실행되는 이벤트 핸들러
@client.event
async def on_message(message):
    # 봇의 메시지는 무시
    if message.author.bot:
        return

    # 사용자 메시지를 GPT-3 API를 사용해 반박하는 함수 호출
    response = generate_response(message.content)

    # 반박한 내용을 디스코드 채널에 전송
    await message.channel.send(response)

# GPT-3 API를 사용해 반박하는 함수
def generate_response(user_message):
    openai.api_key = OPENAI_API_KEY
    prompt = f"User said: {user_message}\nBot:"

    # GPT-3에 요청 보내기
    response = openai.Completion.create(
        engine="text-davinci-003",  # GPT-3 엔진 선택
        prompt=prompt,
        max_tokens=50  # 응답으로 생성되는 최대 토큰 수
    )

    return response.choices[0].text.strip()

# Discord 봇 실행
client.run(TOKEN)