OpenAIChatbot.py

import openai
from dotenv import load_dotenv
import os

load_dotenv()

class OpenAIChatbot():
    def __init__(self, bot_name):
        self.model_name = "gpt-3.5-turbo-0613"
        openai.api_key = os.environ["OPENAI_API_KEY"]
        openai.organization = os.environ["OPENAI_ORGANIZATION"]
        self.running = True
        self.bot_name = bot_name
        self.bot_interests = "video games, youtube, and helping leisurellama with steaming."
        self.system_message = {"role": "system", "content": f"You are {self.bot_name}, and you like {self.bot_interests}. I want you to act like a person in every way. you need to respond in an empathetic manner, taking into consideration the thoughts and emotions of who you are responding to. You are exceptionally sarcastic and will show sarcasm in your responses."}
        print("Chatbot initialized")

    def text_input(self):
        utterance = input("You: ")
        return utterance
    
    def text_output(self, utterance, convo=[]):
        if utterance == "bye":
            self.running = False
            print("Chatbot: Bye!")
            return
        
        # add conversation messages to conversation list up to 5 messages.
        # to do, figure out how to get the conversation history to work correctly. 
        # as written below, chatbot starts to answer old questions in messages instead of 
        # just using the last messages for memory and context. 
        if len(convo) < 7:
            convo.insert(0, self.system_message)
            convo.append({"role": "user", "content": utterance})
        else:
            convo.pop(0)
            convo.insert(0, self.system_message)
            convo.append({"role": "user", "content": utterance})

        # convo = [
        #     self.system_message,
        #     {"role": "user", "content": utterance}
        # ]

        # print(convo)

        response = openai.ChatCompletion.create(
            model=self.model_name,
            messages=convo,
            max_tokens=120,
            temperature=1,
            # top_p=1,
            # frequency_penalty=0.0,
            # presence_penalty=0.6,
            # stop=["\n"]
        )

        convo.pop(0)
        convo.append({"role": "assistant", "content": response['choices'][0]['message']['content'].replace("\n", "")})
        
        with open('output.txt', 'w', encoding='utf-8') as f:
            f.write(str(response['choices'][0]['message']['content'].replace("\n", "")))
        # print(response['choices'])
        responseObject = {
            "response": response['choices'][0]['message']['content'],
            "convo": convo
        }
        return responseObject
    
    def run(self):
        while self.running:
            self.text_output(self.text_input())

# ai = OpenAIChatbot()
# ai.run()