import gradio as gr

title = "Character-Aware Stable Diffusion (CHARRED)"

description = "This is a demo of the CHARRED character-aware stable diffusion model for you to enjoy at your own leisure, risk and peril"

def infer_charred(prompt):
    # your inference function for charr stable difusion control 
    return None

with gr.Blocks(theme='gradio/soft') as demo:
 
    gr.Markdown("# Character aware stable difusion")

    with gr.Tab("Journal"):
        gr.Markdown("## On How Four Crazy Fellow Embarked on Training a U-Net from Scratch in Five Days and Almost Died in the End.")

    with gr.Tab("☢️ DEMO ☢️"):
        prompt_input_charr = gr.Textbox(label="Prompt")
        charr_output = gr.Image(label="Output Image")
        submit_btn = gr.Button(value = "Submit")
        charr_inputs = [prompt_input_charr]
        submit_btn.click(fn=infer_charred, inputs=charr_inputs, outputs=[charr_output])
    #examples = [["postage stamp from california", "low quality", "charr_output.png", "charr_output.png" ]]
    #gr.Examples(fn = infer_sd, inputs = ["text", "text", "image", "image"], examples=examples, cache_examples=True)


demo.launch()