import gradio as gr
from predictor import inference

def index_predict(img):
    outputs, preds, heights, bust, waist, hips, description = inference(img, epoch = 7)
    return heights, round(float(bust)), round(float(waist)), round(float(hips)), description[0], description[1]

with gr.Blocks() as demo:
    gr.Markdown(
    """
    # 身材數據評估器 - Body Index Predictor
    ### Input A FACE and get the body index
    """
    )
    with gr.Row():
        image = gr.Image(type="pil")
        with gr.Column(scale=1, min_width=600):
            # 設定輸出元件
            heights = gr.Textbox(label="Heignt(身高)")
            bust = gr.Textbox(label="Bust(胸圍)")
            waist = gr.Textbox(label="Waist(腰圍)")
            hips = gr.Textbox(label="Hips(臀圍)")
            en_des = gr.Textbox(label="English description")
            zh_des = gr.Textbox(label="Chinese description")
    
    #設定按鈕
    submit = gr.Button("Submit")
    #設定按鈕點選事件
    submit.click(fn=index_predict, inputs=image, outputs=[heights, bust, waist, hips, en_des, zh_des])
demo.launch()