openaiapi.py

from openai import OpenAI
from openaikey import OPENAI_KEY
import requests

client = OpenAI(api_key=OPENAI_KEY)

def getCompletion(prompt, system_prompt="", messages=[]):
 
    if system_prompt != "" and len(messages) == 0:
        messages.append({"role": "system", "content": system_prompt})

    messages.append({"role": "user", "content": prompt})

    try: 
        completion = client.chat.completions.create(
            model="gpt-4o",
            messages=messages
        )
        response = completion.choices[0].message.content
        messages.append({"role": "assistant", "content": response})
        return response
    
    except Exception as e:
        print("OpenAI Exception" + str(e))
        return None
    
def generateImage(prompt, resolution="1024x1024", filename="", cheap_mode=False):

    model = "dall-e-3"
    if cheap_mode:
        model="dall-e-2"
        resolution="512x512"
    try:
        response = client.images.generate(
        model=model,
        prompt=prompt,
        n=1,
        quality="standard",
        size=resolution
        )
        image_url = response.data[0].url

        if filename != "":
            image_data = requests.get(image_url).content
            with open(filename, 'wb') as file:
                file.write(image_data)

        return image_url
    except Exception as e:
        print("OpenAI Exception" + str(e))
        return None
    
def textToSpeech(text, filename):
    response = client.audio.speech.create(
    model="tts-1",
    voice="alloy",
    input=text,
    )
    response.write_to_file(filename)


if __name__ == "__main__":
    print(getCompletion("Bonjour"))
    #print(generateImage("a black fox", "1024x1024", "black_fox.png", cheap_mode=True))
    #textToSpeech("hello, I'm Vincent", "hello.mp3")