import os
from panna import Flux1Dev

model = Flux1Dev()
os.makedirs("./test/test_image", exist_ok=True)
prompts = [
    "A majestic lion jumping from a big stone at night",
    "Study for a portrait of lady reading newspaper by Francis Bacon, the British painter"
]

for i, p in enumerate(prompts):
    output = model.text2image([p], batch_size=1, seed=42)
    model.export(output[0], f"./test/test_images/test_flux_1_dev.{i}.png")
    output = model.text2image([p], batch_size=1, seed=42, width=1024, height=720)
    model.export(output[0], f"./test/test_images/test_flux_1_dev.{i}.landscape.png")

