setup

app.py

@@ -2,10 +2,32 @@
 from utils import *
 
 
+
+cloth_examples = get_cloth_examples()
+pose_examples = get_pose_examples()
+
+# Description
+title = r"""
+<h1 align="center">Outfit Anyone in the Wild: Get rid of Annoying Restrictions for Virtual Try-on Task</h1>
+"""
+
+description = r"""
+<b>Official 🤗 Gradio demo</b> for <a href='https://github.com/selfitcamera/Outfit-Anyone-in-the-Wild' target='_blank'><b>Outfit Anyone in the Wild: Get rid of Annoying Restrictions for Virtual Try-on Task</b></a>.<br>
+1. Clothing models are fixed in this demo, but you can create your own in SelfitCamera WeChat applet (for Chainese users).
+2. You can upload your own pose photo, then click the run button and wait for 3~5 minutes to see the results.
+3. After submitting the task, feel free to leave this page. Everytime you refresh this page, completed tasks will be displayed on the history tab (bind with your ip address).
+4. Share your try-on photo with your friends and enjoy! 😊"""
+
+css = """
+.gradio-container {width: 85% !important}
+"""
+
+mk_guide = "If image does not display successfully after button clicked in your browser(mostly Mac+Chrome), try [this demo](https://openxlab.org.cn/apps/detail/jiangxiaoguo/OutfitAnyone-in-the-Wild) please"
+
+
 def onUpload():
     return ""
 
-
 def onClick(cloth_id, pose_image, pose_id, size, request: gr.Request):
     if pose_image is None:
         return None, "no pose image found !"
@@ -16,7 +38,7 @@ def onClick(cloth_id, pose_image, pose_id, size, request: gr.Request):
         # print(res)
         assert os.path.exists(res), res
         # res = cv2.imread(res)
-        return res, "Done! Use the pre-run results directly, the cloth size does not take effect "
+        return res, "Done! Use the pre-run results directly, the cloth size does not take effect ", mk_guide
     else:
         try:
             client_ip = request.client.host
@@ -26,15 +48,15 @@ def onClick(cloth_id, pose_image, pose_id, size, request: gr.Request):
             timeId = int(  str(time.time()).replace(".", "")  )+random.randint(1000, 9999)
             isUpload = upload_pose_img(ApiUrl, OpenId, ApiKey, client_ip, timeId, pose_image)
             if isUpload==0:
-                return None, "fail to upload"
+                return None, "fail to upload", ""
             elif isUpload==-1:
-                return None, "There is a running task already, please wait and check the history tab"
+                return None, "There is a running task already, please wait and check the history tab", ""
             elif isUpload==-2:
-                return None, "can not creat task, you have exhausted free trial quota"
+                return None, "can not creat task, you have exhausted free trial quota", ""
 
             taskId = publicClothSwap(ApiUrl, OpenId, ApiKey, client_ip, cloth_id, timeId, size)
             if taskId==0:
-                return None, "fail to public you task"
+                return None, "fail to public you task", ""
 
             max_try = 1
             wait_s = 30
@@ -42,20 +64,20 @@ def onClick(cloth_id, pose_image, pose_id, size, request: gr.Request):
                 time.sleep(wait_s)
                 state = getInfRes(ApiUrl, OpenId, ApiKey, client_ip, timeId)
                 if state=='stateIs-1':
-                    return None, "task failed, it may be that no human was detected, or there may be illegal content, etc. "
+                    return None, "task failed, it may be that no human was detected, or there may be illegal content, etc. ", ""
                 elif state=='stateIs0':
-                    return None, "task not public success"
+                    return None, "task not public success", ""
                 elif len(state)>20:
-                    return state, "task finished"
+                    return state, "task finished", ""
                 elif (not state.startswith('stateIs')):
                     # return None, 'task is in queue, position is '+str(state)
                     pass
                 else:
-                    return None, state
-            return None, "task has been created successfully, you can refresh the page 5~15 mins latter, and check the following history tab"
+                    return None, state, ""
+            return None, "task has been created successfully, you can refresh the page 5~15 mins latter, and check the following history tab", ""
         except Exception as e:
             print(e)
-            return None, "fail to create task"
+            return None, "fail to create task", ""
 
 def onLoad(request: gr.Request):
     client_ip = request.client.host
@@ -83,26 +105,6 @@ def onLoad(request: gr.Request):
     return his_datas
 
 
-cloth_examples = get_cloth_examples()
-pose_examples = get_pose_examples()
-
-
-# Description
-title = r"""
-<h1 align="center">Outfit Anyone in the Wild: Get rid of Annoying Restrictions for Virtual Try-on Task</h1>
-"""
-
-description = r"""
-<b>Official 🤗 Gradio demo</b> for <a href='https://github.com/selfitcamera/Outfit-Anyone-in-the-Wild' target='_blank'><b>Outfit Anyone in the Wild: Get rid of Annoying Restrictions for Virtual Try-on Task</b></a>.<br>
-1. Clothing models are fixed in this demo, but you can create your own in SelfitCamera WeChat applet (for Chainese users).
-2. You can upload your own pose photo, then click the run button and wait for 3~5 minutes to see the results.
-3. After submitting the task, feel free to leave this page. Everytime you refresh this page, completed tasks will be displayed on the history tab (bind with your ip address).
-4. Share your try-on photo with your friends and enjoy! 😊"""
-
-
-css = """
-.gradio-container {width: 85% !important}
-"""
 with gr.Blocks(css=css) as demo:
     # description
     gr.Markdown(title)
@@ -138,7 +140,7 @@ with gr.Blocks(css=css) as demo:
                 # res_image = gr.Image(label="result image", value=None, type="numpy")
                 # res_image = gr.Image(label="result image", value=cv2.imread(init_res), 
                 #     type="numpy")
-                gr.Markdown("If image does not display successfully after button clicked in your browser(mostly Mac+Chrome), try [this demo](https://openxlab.org.cn/apps/detail/jiangxiaoguo/OutfitAnyone-in-the-Wild) please")
+                MK01 = gr.Markdown()
                 info_text = gr.Textbox(value="", interactive=False, 
                     label='runtime information')
                 
@@ -164,7 +166,7 @@ with gr.Blocks(css=css) as demo:
             his_res_image5 = gr.HTML()            
 
     run_button.click(fn=onClick, inputs=[cloth_id, pose_image, pose_id, size_slider], 
-                     outputs=[res_image, info_text], concurrency_limit=50)
+                     outputs=[res_image, info_text, MK01], concurrency_limit=50)
 
     pose_image.upload(fn=onUpload, inputs=[], outputs=[pose_id],)
     demo.load(onLoad, inputs=[], outputs=[his_pose_image1, his_res_image1,