main.py

import os

import discord
import openai

token = os.environ['SECRET_KEY']
openai.api_key = os.getenv("OPENAI_API_KEY")

# file = input("Enter 1, 2, or 3 for loading the chat:\n ")
file = "2"
match(file):
  case "1":
    file = "chat1.txt"
  case "2":
    file = "chat2.txt"
  case "3": 
    file = "chat3.txt"
  case _:
    print("Invalid choice.")
    exit()

with open(file, "r") as f:
  chat = f.read()

class MyClient(discord.Client):

  async def on_ready(self):
    print(f'Logged on as {self.user}!')

  async def on_message(self, message):
    global chat
    try:
      chat += f"{message.author}: {message.content}\n"
      print(f'Message from {message.author}: {message.content}')
      if self.user != message.author:
        response = openai.chat.completions.create(
            model="gpt-3.5-turbo",
            messages=[{
                "role": "system",
                "content": f"{chat}\nRanitGPT:"
            }, {
                "role": "user",
                "content": message.content
            }],
            temperature=1,
            max_tokens=500,
            top_p=1,
            frequency_penalty=0,
            presence_penalty=0)
        channel = message.channel
        messageToSend = response.choices[0].message.content
        await channel.send(messageToSend)
    except Exception as e:
      print(e)
      chat = ""


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

client = MyClient(intents=intents)
client.run(token)