import openai
import os

openai.api_key = os.environ.get("OPENAI_API_KEY")

def chatgpt_prompt(prompt):
    response = openai.ChatCompletion.create(
        model="gpt-3.5-turbo",
        messages=[{"role": "user", "content": prompt}]
    )
    return response['choices'][0]['message']['content']