utils.py

import openai
import yaml

# read api key from keys.yaml
with open("keys.yaml", "r") as f:
    keys = yaml.load(f, Loader=yaml.FullLoader)
    openai.api_key = keys["key"]


def get_answer(phrase: str):
    completion = openai.ChatCompletion.create(model="gpt-3.5-turbo", messages=[{"role": "user", "content": phrase}])
    return completion.choices[0].message.content


def translator(phrase: str, language: str):
    if language == "English":
        return phrase
    elif language == "Spanish":
        return get_answer(f"Translate from Spanish to English: {phrase}")
    elif language == "German":
        return get_answer(f"Translate from German to English: {phrase}")
    else:
        raise ValueError("Language not supported")


def text_generator(phrase: str):
    return get_answer(f"Convert the following text or bullet points into a report. Focus on the facts and only give information that is actually included in the text. The report is technical, so there should be no sentiments. There should not be a conclusion. Text: {phrase}")