from handler import EndpointHandler
import os

# Create results directory
os.makedirs("./results", exist_ok=True)

# init handler
my_handler = EndpointHandler(path=".")

# prepare sample prompts
prompts = [
    "A dynamic city skyline, stylized and white-hued, stretches across the midsection, featuring abstract buildings and architectural shapes against a white background. The top section reveals a person composing a photograph on a bridge, providing a sense of scale and adventure. The image carries black and white elements, with the city block emerging prominently from the vivid environment, blending creativity and urban aesthetics.",
    "A digital artwork showcases a futuristic cityscape at dusk, viewed through a wide-angle lens. The buildings are sleek and towering, reflecting a mix of modern and cyberpunk architecture, set against a vivid sunset. Swirls of cloud and city lights create a dynamic, lively atmosphere. The image is bordered with intricate orange and black geometric patterns, enhancing contemporary aesthetic against a plain white background."
]

# Generate and save images
for i, prompt in enumerate(prompts, 1):
    image = my_handler({"inputs": prompt})
    #image.save(f"./results/cityscape_{i}.png")
    print(image)
    #print(f"Saved image {i} to ./results/cityscape_{i}.png")

# non_holiday_pred [{'label': 'joy', 'score': 0.9985942244529724}]
# holiday_payload [{'label': 'happy', 'score': 1}]