import gradio as gr

model_1 = gr.load("models/pimpilikipilapi1/NSFW_master")
model_2 = gr.load("models/DiegoJR1973/NSFW-TrioHMH-Flux")
model_3 = gr.load("models/prashanth970/flux-lora-uncensored")

def generate_image_model_1(prompt):
    prompt += " 2k"
    return model_1(prompt)

def generate_image_model_2(prompt):
    prompt += " 8k"
    return model_2(prompt)

def generate_image_model_3(prompt):
    prompt += " 10k"
    return model_3(prompt)

interface = gr.Interface(
    fn=lambda prompt: (
        generate_image_model_1(prompt),
        generate_image_model_2(prompt),
        generate_image_model_3(prompt)
    ),
    inputs=[
        gr.Textbox(label="Type your prompt here: ✍️", placeholder="Describe what you want..."),
    ],
    outputs=[
        gr.Image(label="Generated Image - Model 1"),
        gr.Image(label="Generated Image - Model 2"),
        gr.Image(label="Generated Image - Model 3"),
    ],
    title="Text to Image (NSFW) 🔞",
    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()