'''
Write an openai chat bot in python
'''
import openai
import os

# Set up the OpenAI API client
openai.api_key = os.environ["OPENAI_API_KEY"]

# Define a function that takes in a user's message and returns the bot's response
def get_bot_response(user_message):
    prompt = f"User: {user_message}\nBot:"
    response = openai.Completion.create(
        engine="davinci",
        prompt=prompt,
        max_tokens=50,
        n=1,
        stop=None,
        temperature=0.7,
    )
    message = response.choices[0].text.strip()
    return message

# Let's test the bot
while True:
    user_message = input("You: ")
    if user_message.lower() in ['bye', 'goodbye']:
        print("Bot: Goodbye!")
        break
    bot_response = get_bot_response(user_message)
    print(f"Bot: {bot_response}")