from openai import OpenAI
from dotenv import dotenv_values

config = dotenv_values(".env")
client = OpenAI(api_key=config["OPENAI_API_KEY"])
user_input = input("You: ")
prompt = {"role": "user" , "content": user_input }
res = client.chat.completions.create(
       model="gpt-3.5-turbo",
       messages=[prompt],
     )
print("Assistant: " + res.choices[0].message.content)