chat.py

chat.py

from colorama import Fore, Back, Style
import os
import openai

# Set up OpenAI API key
openai.api_key = os.getenv("OPENAI_API_KEY")
prompt = """Your prompts here"""
# Set up initial message
messages = [{"role": "system", "content": }]


while True:
    # Get user input
    user_input = input("You: ")
    # Exit loop if the user inputs "exit"
    if user_input.lower() == "exit":
        break
    # Add user message to messages list
    messages.append({"role": "user", "content": user_input})
    # Call OpenAI API
    completion = openai.ChatCompletion.create(
        model="gpt-3.5-turbo",
        messages=messages
    )
    # Get response from API and add to messages list
    response = completion.choices[0].message
    messages.append(response)
    # Print response
    print(Fore.RED + f"Bot: {response['content']}"+Style.RESET_ALL)