app.py

from utils import *


cloth_examples = get_cloth_examples(hr=0)
cloth_hr_examples = get_cloth_examples(hr=1)
pose_examples = get_pose_examples()
tip1, tip2 = get_tips()

# Description
title = r"""
<h1 align="center">Outfit Anyway: Best customer try-on experience ever</h1>
"""

description = r"""
<b>Test results on </b> <a href='https://selfit-deploy-1256039085.cos.accelerate.myqcloud.com/ClothData/Publics/Shows/shows/man_v2/man_v2.html' target='_blank'><b>man01</b></a>.<br>
<b>Test results on </b> <a href='https://selfit-deploy-1256039085.cos.accelerate.myqcloud.com/ClothData/Publics/Shows/shows/woman_v2/woman_v2.html' target='_blank'><b>woman01</b></a>.<br>
"""

mk_guide = ""


def onClick(cloth_image, pose_image, high_resolution):
    if pose_image is None:
        yield None, "no pose image found !", ""
        return None, "no pose image found !", ""
    if cloth_image is None:
        yield None, "no cloth image found !", ""
        return None, "no cloth image found !", ""

    pose_id = os.path.basename(pose_image).split(".")[0]
    cloth_id = int(os.path.basename(cloth_image).split(".")[0])
            
    try:
        client_ip = '8.8.8.8'
        yield None, "begin to upload ", ""

        timeId = int(  str(time.time()).replace(".", "")  )+random.randint(1000, 9999)
        upload_url = upload_pose_img(ApiUrl, OpenId, ApiKey, client_ip, timeId, pose_image)
        # exit(0)
        yield None, "begin to public task ", ""
        # return None, "begin to public task ", ""
        
        if len(upload_url)==0:
            yield None, "fail to upload", ""
            return None, "fail to upload", ""

        if high_resolution:
            public_res = publicClothSwap(upload_url, cloth_id, is_hr=1)
        else:
            public_res = publicClothSwap(upload_url, cloth_id, is_hr=0)
        if public_res is None:
            yield None, "fail to public you task", ""
            return None, "fail to public you task", ""

        yield public_res['mid_result'], f"task is processing, task id: {public_res['id']}, {public_res['msg']}", ""

        max_try = 120
        wait_s = 1
        for i in range(max_try):
            time.sleep(wait_s)
            state = getInfRes(public_res['id'])
            timestamp = int(time.time() * 1000)
            if state is None:
                yield public_res['mid_result'] + f"?t={timestamp}", "task query failed,", ""
            elif state['status']=='PROCESSING':
                yield public_res['mid_result'] + f"?t={timestamp}", f"task is processing, query {i}", ""
            elif state['status']=='SUCCEED':
                yield state['output1'] + f"?t={timestamp}", f"task finished, {state['msg']}", ""
                return state['output1'] + f"?t={timestamp}", f"task finished, {state['msg']}", ""
            elif state['status']=='FAILED':
                yield None, f"task failed, {state['msg']}", ""
                return None, f"task failed, {state['msg']}", ""
            else:
                yield public_res['mid_result'] + f"?t={timestamp}", f"task is on processing, query {i}", ""
        return None, "no machine...", ""
    except Exception as e:
        print(e)
        raise e
        return None, "fail to create task", ""

with gr.Blocks() as demo:
    gr.Markdown(title)
    gr.Markdown(description)

    with gr.Accordion('upload tips', open=False):
        with gr.Row():
            gr.HTML(f"<img src=\"{tip1}\" >")
            gr.HTML(f"<img src=\"{tip2}\" >")
                    
    with gr.Row():
        with gr.Column():
            cloth_image = gr.Image(value=None, interactive=False, type="filepath", label="choose a clothing")
            example = gr.Examples(inputs=cloth_image,examples_per_page=20,examples=cloth_examples, label="clothing")
            hr_example = gr.Examples(inputs=cloth_image,examples_per_page=9,examples=cloth_hr_examples, label="invalid clothing")
            
        with gr.Column():
            pose_image = gr.Image(value=None, type="filepath", label="choose/upload a photo")
            example_pose = gr.Examples(inputs=pose_image,
                                      examples_per_page=20,
                                      examples=pose_examples)
            
        with gr.Column():
            with gr.Column():
                # size_slider = gr.Slider(-3, 3, value=1, interactive=True, label="clothes size")
                high_resolution = gr.Checkbox(value=False, label="high resolution", interactive=True)
                
                run_button = gr.Button(value="Run")
                info_text = gr.Textbox(value="", interactive=False, 
                    label='runtime information')                
                res_image = gr.Image(label="result image", value=None, type="filepath")
                MK01 = gr.Markdown()

                
    # with gr.Tab('history'):

    #     with gr.Row():  # 用 Row 包裹按钮
    #         with gr.Column(scale=0.5):  # Button 占用 Row 的一半
    #             refresh_button = gr.Button("Refresh History", size="small")
    #     MK02 = gr.Markdown(value="")  # 示例 Markdown 内容

    #     # with gr.Row():
    #     #     MK02 = gr.Markdown()

    #     with gr.Row():
    #         his_pose_image1 = gr.HTML()
    #         his_res_image1 = gr.HTML()

    #     with gr.Row():
    #         his_pose_image2 = gr.HTML()
    #         his_res_image2 = gr.HTML()

    #     with gr.Row():
    #         his_pose_image3 = gr.HTML()
    #         his_res_image3 = gr.HTML()            

    #     with gr.Row():
    #         his_pose_image4 = gr.HTML()
    #         his_res_image4 = gr.HTML()            

    #     with gr.Row():
    #         his_pose_image5 = gr.HTML()
    #         his_res_image5 = gr.HTML()            

    run_button.click(fn=onClick, inputs=[cloth_image, pose_image, high_resolution], 
                     outputs=[res_image, info_text, MK01])

    # refresh_button.click(fn=onLoad, inputs=[], outputs=[his_pose_image1, his_res_image1,
    #     his_pose_image2, his_res_image2, his_pose_image3, his_res_image3,
    #     his_pose_image4, his_res_image4, his_pose_image5, his_res_image5,
    #     MK02])

    # demo.load(onLoad, inputs=[], outputs=[his_pose_image1, his_res_image1,
    #     his_pose_image2, his_res_image2, his_pose_image3, his_res_image3,
    #     his_pose_image4, his_res_image4, his_pose_image5, his_res_image5,
    #     MK02])

if __name__ == "__main__":

    demo.queue(max_size=50)
    # demo.queue(concurrency_count=60)
    # demo.launch(server_name='0.0.0.0', server_port=225)
    demo.launch(server_name='0.0.0.0')