import gradio as gr

# Load your model
model = gr.load("models/John6666/wai-ani-nsfw-ponyxl-v7-sdxl")

# Define a function to generate an image from text
def generate_image(text):
    # Use the model to generate an image based on the input text
    # Assuming your model takes a text input and returns an image
    return model(text)

# Set up the Gradio Blocks UI
with gr.Blocks() as demo:
    gr.Markdown("## Text to Image Generation")
    text_input = gr.Textbox(label="Enter text prompt:", placeholder="Type your description here...")
    image_output = gr.Image(label="Generated Image")
    
    # Create a button to generate the image
    generate_button = gr.Button("Generate Image")
    
    # Set the action for the button to call the generate_image function
    generate_button.click(generate_image, inputs=text_input, outputs=image_output)

# Launch the app
demo.launch(debug=True)