text_to_img.py

from together import Together
import base64
import time
import os
from dotenv import load_dotenv

load_dotenv()

TOGETHER_API_KEY = os.getenv("TOGETHER_AI_API_KEY")
client = Together(api_key=TOGETHER_API_KEY)

def main(myprompt: str, img_file_name: str): # Images main format
    response = client.images.generate(
        prompt=myprompt,
        model="black-forest-labs/FLUX.1-schnell-Free",
        width=1024,
        height=768,
        steps=1,
        n=1,
        response_format="b64_json",
    )
    imgstring: str = response.data[0].b64_json
    imgdata: bytes = base64.b64decode(imgstring)
    filename: str = f'{img_file_name}.png'
    with open(filename, 'wb') as f:
        f.write(imgdata)

if __name__=="__main__":
    main("Cat eating burger", "burger-cat")