import config
import openai
import discord
import asyncio
import os

openai.api_key = config.OPENAI_TOKEN
jonkey = config.JONBOT_TOKEN
model = config.TXT_MODEL

intents = discord.Intents.default()
intents.messages = True
intents.message_content = True
client = discord.Client(intents=intents)

@client.event
async def on_ready():
    print(f"Logged in as {client.user.name} running {model}.", flush=True)

@client.event
async def on_message(message):
    if message.author == client.user:
        return
    if message.content.lower().startswith("hey jon") or isinstance(message.channel, discord.DMChannel):
        question = message.content[7:] if message.content.lower().startswith("hey jon") else message.content
        try:
            response = openai.ChatCompletion.create(
                model=model,
                messages=[
                        {"role": "system", "content": "Pretend you are friendly Jon Arbuckle."},
                        {"role": "user", "content": f"{question}"}
                    ],
                max_tokens=400
            )
            answer = response['choices'][0]['message']['content']
            answer = answer.replace("AI language model", "American citizen")
            answer = answer.replace("language model AI", "citizen of the United States")
            await message.channel.send(answer)
        except Exception as e:
            e = str(e)
            await message.channel.send(f"`JonBot Error: {e}`")

async def jonbot_connect():
    while True:
        try:
            await client.start(jonkey)
        except Exception as e:
                e = str(e)
                logger.error(f"Jonbot couldn't connect! {e}")
                await asyncio.sleep(60)

if __name__ == "__main__":
    asyncio.run(jonbot_connect())