# Note: you need to be using OpenAI Python v0.27.0 for the code below to work
import openai

# openai.api_key = "YOUR_API_KEY" # OPENAI_API_KEY is an environment variable

messages=[
        {"role": "system", "content": "You are a helpful assistant."}]

while True:
    user_input = input("You: ")
    messages.append({"role": "user", "content": user_input})


    response = openai.ChatCompletion.create(
    model="gpt-3.5-turbo",
        messages = messages,
    )
    print("Assistant: " + response['choices'][0]['message']['content'])
    messages.append({"role": "assistant", "content": response['choices'][0]['message']['content']})
    # print("Assistant: " + response.choices[0].text)