import gradio as gr

dictionary = {
    "a playful man": "A young man in his mid-20s to early 30s, Has a playful and energetic demeanor, with a big smile always on his face, Has a lean and athletic build, Has messy, yet stylish hair that is always styled in a way that makes it look like he just got out of bed, Has a playful twinkle in his eye, and a hint of mischief in his expression, Wears casual, comfortable clothing that is both stylish and practical,Often seen with a bag or backpack, as if he's always ready for an adventure",
    "a calm woman": "A reserved and collected woman with a soft spoken voice and gentle demeanor. She carries herself with grace and poise, always appearing calm and in control, even in stressful situations. Her appearance is polished and put together, with a preference for understated elegance. She is kind and empathetic, and her presence brings a sense of peace to those around her.",
    "a boy who is bullied": "todo"
    }

def chatgpt(query):
    try:
        output = dictionary[query]
    except:
        output = "Please try: https://chat.openai.com"
    return output

def generateimage1(query):
    try:
        output = dictionary[query]
    except:
        output = "Please try: https://chat.openai.com"
    return output

def generateimage2(query):
    try:
        output = dictionary[query]
    except:
        output = "Please try: https://chat.openai.com"
    return output

with gr.Blocks() as demo:
    gr.Markdown("Ask **ChatGPT** to generate detailed descriptbions.")
    gr.Markdown("Example: Act as a professional character designer. Please generate prompts include detail appearance descriptions that related to the personality. Here's your first question: 'a gentle man':")
    with gr.Row():
        inpprompt = gr.Dropdown(list(dictionary.keys()))
        outprompt = gr.Textbox()
    btn = gr.Button("Generate mode detailed description.")
    btn.click(fn=chatgpt, inputs=inpprompt, outputs=outprompt)

    
    gr.Markdown("Ask **Stable Diffusion** to generate images.")
    with gr.Row():
        out1image = gr.Image()
        out2image = gr.Image()
    with gr.Row():
        btn1 = gr.Button("Generate image with origial prompt.")
        btn2 = gr.Button("Generate image with ChatGPT's prompt.")
        
    btn1.click(fn=generateimage1, inputs=inpprompt, outputs=out1image)
    btn2.click(fn=generateimage2, inputs=outprompt, outputs=out2image)
    

demo.launch()