import openai
openai.api_key = 'APIKEY'
user_input = input("Prompt: ")
response = openai.ChatCompletion.create(
    model="gpt-3.5-turbo",
    messages=[
        {"role": "system", "content": "You are a helpful assistant."},
        {"role": "user", "content": user_input},
    ],
    max_tokens=4096, 
    temperature=0.5,
    top_p=1
)
print(response['choices'][0]['message']['content'])