prodia-studio / app.py
zenafey's picture
Update app.py
b21a9fd
raw
history blame
9.15 kB
"""
Original code by Zenafey
@zenafey
"""
import gradio as gr
from engine import generate_sd, generate_sdxl, transform_sd, controlnet_sd, image_upscale, get_models
from const import CMODELS, CMODULES, SAMPLER_LIST, SDXL_MODEL_LIST
with gr.Blocks() as demo:
gr.Markdown("""
<h1><center>Prodia Studio</center></h>
<h2><center>powered by Prodia Stable Diffusion API</center></h2>""")
with gr.Tab("/sdxl/generate [BETA]"):
with gr.Row():
with gr.Column(scale=6, min_width=600):
prompt = gr.Textbox("puppies in a cloud, 4k", placeholder="Prompt", show_label=False, lines=3)
negative_prompt = gr.Textbox(placeholder="Negative Prompt", show_label=False, lines=3)
with gr.Row():
with gr.Column():
sampler = gr.Dropdown(value="Euler a", show_label=True, label="Sampling Method",
choices=SAMPLER_LIST)
model = gr.Dropdown(
interactive=True,
value="sd_xl_base_1.0.safetensors [be9edd61]",
show_label=True,
label="Stable Diffusion XL Checkpoint",
choices=SDXL_MODEL_LIST
)
seed = gr.Number(label="Seed", value=-1)
with gr.Column():
steps = gr.Slider(label="Sampling Steps", minimum=1, maximum=50, value=25, step=1)
cfg_scale = gr.Slider(label="CFG Scale", minimum=1, maximum=20, value=7, step=1)
text_button = gr.Button("Generate", variant='primary')
with gr.Column(scale=7):
image_output = gr.Image()
text_button.click(generate_sdxl,
inputs=[prompt, negative_prompt, model, steps, sampler, cfg_scale, seed], outputs=image_output)
with gr.Tab("/sd/generate"):
with gr.Row():
with gr.Column(scale=6, min_width=600):
prompt = gr.Textbox("puppies in a cloud, 4k", placeholder="Prompt", show_label=False, lines=3)
negative_prompt = gr.Textbox(placeholder="Negative Prompt", show_label=False, lines=3)
with gr.Row():
with gr.Column():
sampler = gr.Dropdown(value="Euler a", show_label=True, label="Sampling Method",
choices=SAMPLER_LIST)
model = gr.Dropdown(
interactive=True,
value=get_models()[1],
show_label=True,
label="Stable Diffusion Checkpoint",
choices=get_models()
)
upscale = gr.Checkbox(label="Upscale", value=True)
seed = gr.Number(label="Seed", value=-1)
with gr.Column():
width = gr.Slider(label="Width", maximum=1024, value=512, step=8)
height = gr.Slider(label="Height", maximum=1024, value=512, step=8)
steps = gr.Slider(label="Sampling Steps", minimum=1, maximum=50, value=25, step=1)
cfg_scale = gr.Slider(label="CFG Scale", minimum=1, maximum=20, value=7, step=1)
text_button = gr.Button("Generate", variant='primary')
with gr.Column(scale=7):
image_output = gr.Image()
text_button.click(generate_sd,
inputs=[prompt, negative_prompt, model, steps, sampler, cfg_scale, width, height, seed,
upscale], outputs=image_output)
with gr.Tab("/sd/transform"):
with gr.Row():
with gr.Row():
with gr.Column(scale=6, min_width=600):
with gr.Row():
with gr.Column():
image_input = gr.Image(type='filepath')
with gr.Column():
prompt = gr.Textbox("puppies in a cloud, 4k", label='Prompt', placeholder="Prompt", lines=3)
negative_prompt = gr.Textbox(placeholder="badly drawn", label='Negative Prompt', lines=3)
with gr.Row():
with gr.Column():
sampler = gr.Dropdown(value="Euler a", show_label=True, label="Sampling Method", choices=SAMPLER_LIST)
model = gr.Dropdown(
interactive=True,
value=get_models()[1],
show_label=True,
label="Stable Diffusion Checkpoint",
choices=get_models()
)
upscale = gr.Checkbox(label="Upscale", value=True)
seed = gr.Number(label="Seed", value=-1)
with gr.Column():
steps = gr.Slider(label="Sampling Steps", minimum=1, maximum=30, value=25, step=1)
cfg_scale = gr.Slider(label="CFG Scale", minimum=1, maximum=20, value=7, step=1)
denoising_strength = gr.Slider(label="Denoising Strength", minimum=0.1, maximum=1.0, value=0.7, step=0.1)
text_button = gr.Button("Generate", variant='primary')
with gr.Column(scale=7):
image_output = gr.Image()
text_button.click(transform_sd,
inputs=[image_input, model, prompt, denoising_strength, negative_prompt, steps, cfg_scale, seed, upscale, sampler
], outputs=image_output)
with gr.Tab("/sd/controlnet"):
with gr.Row():
with gr.Row():
with gr.Column(scale=6, min_width=600):
with gr.Row():
with gr.Column():
image_input = gr.Image(type='filepath')
with gr.Column():
prompt = gr.Textbox("puppies in a cloud, 4k", label='Prompt', placeholder="Prompt", lines=3)
negative_prompt = gr.Textbox(placeholder="badly drawn", label='Negative Prompt', lines=3)
with gr.Row():
with gr.Column():
sampler = gr.Dropdown(value="Euler a", show_label=True, label="Sampling Method", choices=SAMPLER_LIST)
model = gr.Dropdown(
interactive=True,
value="control_v11p_sd15_canny [d14c016b]",
show_label=True,
label="ControlNet Model",
choices=CMODELS
)
module = gr.Dropdown(
interactive=True,
value="none",
show_label=True,
label="ControlNet Module",
choices=CMODULES
)
seed = gr.Number(label="Seed", value=-1)
with gr.Column():
width = gr.Slider(label="Width", maximum=1024, value=512, step=8)
height = gr.Slider(label="Height", maximum=1024, value=512, step=8)
steps = gr.Slider(label="Sampling Steps", minimum=1, maximum=30, value=25, step=1)
cfg_scale = gr.Slider(label="CFG Scale", minimum=1, maximum=20, value=7, step=1)
resize_mode = gr.Dropdown(label='resize_mode', value="0", choices=["0", "1", "2"])
with gr.Row():
threshold_a = gr.Number(label="threshold_a", value=100)
threshold_b = gr.Number(label="threshold_b", value=200)
text_button = gr.Button("Generate", variant='primary')
with gr.Column(scale=7):
image_output = gr.Image()
text_button.click(controlnet_sd,
inputs=[image_input, model, module, threshold_a, threshold_b, resize_mode, prompt,
negative_prompt, steps, cfg_scale, seed, sampler, width, height],
outputs=image_output)
with gr.Tab("/upscale"):
with gr.Row():
with gr.Column():
image_input = gr.Image(type='filepath')
scale_by = gr.Radio(['2', '4'], label="Scale by")
upscale_btn = gr.Button("Upscale!", variant='primary')
with gr.Column():
image_output = gr.Image()
upscale_btn.click(image_upscale, inputs=[image_input, scale_by], outputs=image_output)
demo.launch(show_api=False)