import gradio as gr
import random
import os


model1 = gr.load("models/pimpilikipilapi1/NSFW_master")
model2 = gr.load("models/DiegoJR1973/NSFW-TrioHMH-Flux")
model3 = gr.load("models/prashanth970/flux-lora-uncensored")


def generate_images(text):
    result_image1 = model1(text)
    result_image2 = model2(text)
    result_image3 = model3(text)
    
    return result_image1, result_image2, result_image3

interface = gr.Interface(
    fn=generate_images,
    inputs=[
        gr.Textbox(label="Type here your imagination:", placeholder="Type or click an example..."),
    ],
    outputs=[
        gr.Image(label="Model 1 Output"),
        gr.Image(label="Model 2 Output"),
        gr.Image(label="Model 3 Output")
    ],
    theme="NoCrypt/miku",
    description="Sorry for the inconvenience. The model is currently running on the CPU, which might affect performance. We appreciate your understanding.",
)

interface.launch()