From d2857910d3403c5ce4745d6161b2f3a78cb4ca6a Mon Sep 17 00:00:00 2001
From: nsde <neostyxde@gmail.com>
Date: Sun, 23 Jul 2023 23:57:41 +0200
Subject: [PATCH] Added simple chatbot

---
 README.md         |  3 ++-
 cord/bot.py       | 62 +++++++++++++++++++++++++++++++++++++++++------
 cord/smarthelp.py |  0
 3 files changed, 56 insertions(+), 9 deletions(-)
 create mode 100644 cord/smarthelp.py

diff --git a/README.md b/README.md
index 21563d4..fcd191a 100644
--- a/README.md
+++ b/README.md
@@ -1,2 +1,3 @@
 # nova-discord
-🤖 Discord bot for project Nova
+🤖 Discord bot for project Nova, built using [NextCord](https://github.com/nextcord/nextcord) for Python.
+
diff --git a/cord/bot.py b/cord/bot.py
index 36d120e..adf0032 100644
--- a/cord/bot.py
+++ b/cord/bot.py
@@ -1,26 +1,72 @@
 # This example requires the 'members' and 'message_content' privileged intents to function.
 
 import os
-import discord
+import openai
+import nextcord
 
 from dotenv import load_dotenv
-from discord.ext import commands
+from nextcord.ext import commands
+from nextcord import SlashOption
 
 load_dotenv()
 
-intents = discord.Intents.default()
-intents.members = True
-intents.message_content = True
-
-bot = commands.Bot(command_prefix='?', intents=intents)
+bot = commands.Bot(
+    intents=nextcord.Intents.all(),
+    default_guild_ids=[int(guild_id) for guild_id in os.getenv('DISCORD_GUILD_IDS').split()]
+)
 
 @bot.event
 async def on_ready():
     print(f'Online as {bot.user} (ID: {bot.user.id})')
-    await bot.change_presence(activity=discord.Game(name='with fire'))
+    await bot.change_presence(activity=nextcord.Game(name='with fire'))
 
 @bot.event
 async def on_message(message):
     await bot.process_commands(message)
 
+@bot.slash_command(description='Chat with AI')
+async def chat(interaction: nextcord.Interaction,
+    prompt: str = SlashOption(description='AI Prompt', required=True)
+):
+    partial_message = await interaction.send('‎') # empty message
+    message = await partial_message.fetch()
+
+    openai.api_base = os.getenv('OPENAI_BASE', 'https://api.openai.com/v1')
+    openai.api_key = os.getenv('OPENAI_KEY')
+
+    model = os.getenv('OPENAI_MODEL')
+
+    async with interaction.channel.typing(): # show the "Typing..."
+        completion = openai.ChatCompletion.create(
+            model=model,
+            messages=[
+                {'role': 'system', 'content': f"""You are a helpful Discord AI bot based on OpenAI\'s {model} model called "Nova".
+You were developed by NovaAI (website: nova-oss.com) in July of 2023, but your knowledge is limited to mid-2021.
+Respond using Markdown. Keep things simple and short and directly do what the user says without any fluff.
+For programming code, always make use formatted code blocks like this:
+```py
+print("Hello")
+```
+"""},
+                {'role': 'user', 'content': prompt}
+            ],
+            temperature=0.6,
+            stream=True
+        )
+
+        text = ''
+
+        for event in completion: # loop through word generation in real time
+            try:
+                new_text = event['choices'][0]['delta']['content'] # newly generated word
+            except KeyError: # end
+                break
+
+            text += new_text
+
+            if text:
+                await message.edit(content=text)
+
+    await message.add_reaction('✅')
+
 bot.run(os.getenv('DISCORD_TOKEN'))
diff --git a/cord/smarthelp.py b/cord/smarthelp.py
new file mode 100644
index 0000000..e69de29