From 7e0eeaa7eedda51c3bdfd189c3e7787d1ee892f7 Mon Sep 17 00:00:00 2001
From: William Lane <will@laneguage.com>
Date: Tue, 4 Apr 2023 12:53:40 -0700
Subject: [PATCH] add logging

---
 app/discord_bot/connect_discord.py | 16 +++++++++++++---
 app/mabel/connect_openai.py        | 23 ++++++++++++++---------
 2 files changed, 27 insertions(+), 12 deletions(-)

diff --git a/app/discord_bot/connect_discord.py b/app/discord_bot/connect_discord.py
index 74141e7..b5db9da 100644
--- a/app/discord_bot/connect_discord.py
+++ b/app/discord_bot/connect_discord.py
@@ -2,18 +2,28 @@ from dotenv import load_dotenv
 import os
 import discord
 from app.mabel.connect_openai import chatgpt_response
+import logging
+logging.basicConfig(level=logging.INFO)
 
 #load_dotenv()
 discord_token=os.getenv('DISCORD_TOKEN')
 class MyClient(discord.Client):
     async def on_ready(self):
-        print('Logged in as: ', self.user)
+        logging.info('Logged in as: ', self.user)
     async def on_message(self, message):
+        logging.info('Message from {0.author}: {0.content}'.format(message))
         if message.author == client.user:
             return
         if "mabel" in message.content.lower():
-            response = await chatgpt_response(message.content)
-            await message.channel.send(response)
+            try:
+                response = await chatgpt_response(message.content)
+            except Exception as e:
+                logging.info('CHATGPT Error: ', e)
+            try:
+                logging.info('Response: ', response)
+                await message.channel.send(response)
+            except Exception as e:
+                logging.info('DISCORD Error: ', e)
             return
 
 intents=discord.Intents.default()
diff --git a/app/mabel/connect_openai.py b/app/mabel/connect_openai.py
index eabb3cf..8674620 100644
--- a/app/mabel/connect_openai.py
+++ b/app/mabel/connect_openai.py
@@ -28,15 +28,20 @@ async def chatgpt_response(prompt):
 
     if "mabel" in prompt.lower():
         logger.info("PROMPT: " + prompt.lower())
-        response = openai.ChatCompletion.create(
-        model="gpt-3.5-turbo",
-        messages=messages,
-        max_tokens=1800,
-        )
-        response_content = response['choices'][0]['message']['content']
-        logger.info("RESPONSE: " + response_content)
-        DIALOGUE_STACK.append({"role": "assistant", "content": response_content })
-        return response_content
+        try:
+            response = openai.ChatCompletion.create(
+            model="gpt-3.5-turbo",
+            messages=messages,
+            max_tokens=1800,
+            )
+            response_content = response['choices'][0]['message']['content']
+            logger.info("RESPONSE: " + response_content)
+            DIALOGUE_STACK.append({"role": "assistant", "content": response_content })
+            return response_content
+        except Exception as e:
+            logger.info("OPENAI Error: " + str(e))
+            return "NO RESPONSE"
+        
     else:
         DIALOGUE_STACK.append({"role": "user", "content": prompt})
         return
\ No newline at end of file