call_openai_api.py

def ask_chatgpt(input_text: str) -> str:
    """ Call OpenAI's gpt-3.5-turbo model API with prompt by 'generate_prompt()' function """

    openai.api_key = os.getenv('OPENAI_API_KEY')

    prompt = generate_prompt(input_text)

    messages = [{"role": "user", "content": prompt}]
    response = openai.ChatCompletion.create(
        model="gpt-3.5-turbo",
        messages=messages
    )

    return response.choices[0].message["content"]