added discord bot

chat.py

@@ -1,16 +1,16 @@
 import requests
 
-from utils.prompt import prompt
 
-messages = []
-while True:
-    user_prompt = prompt('>> User: ')
-    messages.append({'role': 'user', 'content': user_prompt})
+class ChatClient:
+    messages = []
 
-    response = requests.post('http://localhost:8900/', json=messages)
-    if response.status_code == 200:
-        messages = response.json()
-        print('>> Bot : ' + messages[-1]['content'])
-    else:
-        messages = messages[0:-1]
-        print('### Error from backend')
+    def input(self, message):
+        self.messages.append({'role': 'user', 'content': message})
+
+        response = requests.post('http://localhost:8900/', json=self.messages)
+        if response.status_code == 200:
+            self.messages = response.json()
+            return self.messages[-1]['content']
+        else:
+            self.messages = self.messages[0:-1]
+            return '### Error from backend'

chat_cli.py

@@ -0,0 +1,8 @@
+from chat import ChatClient
+from utils.prompt import prompt
+
+client = ChatClient()
+while True:
+    user_prompt = prompt('>> User: ')
+    response = client.input(user_prompt)
+    print(response)

discord_bot.py

@@ -0,0 +1,26 @@
+import discord
+
+from chat import ChatClient
+from discord_bot_token import discord_bot_token
+
+intents = discord.Intents.default()
+intents.message_content = True
+discord_client = discord.Client(intents=intents)
+
+chat_client = ChatClient()
+
+@discord_client.event
+async def on_ready():
+    print(f'We have logged in as {discord_client.user}')
+
+
+@discord_client.event
+async def on_message(message):
+    if message.author == discord_client.user:
+        return
+
+    response = chat_client.input(message.content)
+    await message.channel.send(response)
+
+
+discord_client.run(discord_bot_token)

discord_bot_token.py

@@ -0,0 +1 @@
+discord_bot_token = '<token here>'