import gradio as gr

sex_value = ""
age_value = 0


def prompt(user_upload, name, sex_value, age_value):

    if sex_value == "male":
        sex = "man"
    elif sex_value == "female":
        sex = "women"
    else:
        sex = "nutral person"

    processed_output=[
    "output/demo1.jpg", 
    "output/demo2.jpg",
    "output/demo3.jpg",
    "output/demo4.jpg",
    "output/demo5.jpg"
    ]

    return f"RAW photo, VAR TRIGGERWORD, photo a {sex} ,wearing a suit, {age_value} years old,standing on a huge stage, energetic pose, wide shot, huge american flag in background, stage lighting --n (deformed iris, deformed pupils, semi-realistic, cgi, 3d, render, sketch, cartoon, drawing, anime, mutated hands and fingers:1.4), (deformed, distorted, disfigured:1.3), poorly drawn, bad anatomy, wrong anatomy, extra limb, missing limb, floating limbs, disconnected limbs, mutation, mutated, ugly, disgusting, amputation", name, processed_output


demo = gr.Interface(
    fn=prompt,
    inputs=[
        gr.Files(label="upload 10 pics", file_count="directory", file_types=["image", ".jpg"]),
        #gr.Button(value="load demo files"),
        gr.Textbox(label="name"),
        gr.Radio(["male", "female", "none binary"], label="sex"),
        gr.Slider(18, 99, step=1, label="age")
    ],
    outputs=[
        gr.Textbox(label="prompt"),
        gr.Textbox(label="name"),
        gr.Gallery(label="processed images", columns=[3])
        #gr.Textbox(label="file list")
    ]
)

demo.launch(share=True)