setup

app.py

@@ -85,23 +85,29 @@ def onLoad(request: gr.Request):
     if x_forwarded_for:
         client_ip = x_forwarded_for
     his_datas = [None for _ in range(10)]
+    info = ''
     try:
         infs = getAllInfs(ApiUrl, OpenId, ApiKey, client_ip)
         print(client_ip, 'history infs: ', len(infs))
-        # print(infs)
-
+        cnt = 0
+        finish_n, fail_n, queue_n = 0, 0, 0
         for i, inf in enumerate(infs):
-            if i>4: continue
-            pose = inf['pose']
-            res = inf['res']
-            # his_datas[i*2] = f"[pose]({pose})"+f"![pose]({pose})"
-            # his_datas[i*2+1] = f"[res]({res})"+f"![res]({res})"  
-            his_datas[i*2] = f"<img src=\"{pose}\" >"
-            his_datas[i*2+1] = f"<img src=\"{res}\" >"
-
+            if inf['state']==2:
+                if cnt>4: continue
+                pose, res = inf['pose'], inf['res']
+                his_datas[cnt*2] = f"<img src=\"{pose}\" >"
+                his_datas[cnt*2+1] = f"<img src=\"{res}\" >"
+                finish_n += 1
+                cnt += 1
+            elif inf['state'] in [-1, -2, 0]:
+                fail_n += 1
+            elif inf['state'] in [1]:
+                queue_n += 1
+        info = f"you have {finish_n} successed tasks, {queue_n} running tasks, {fail_n} failed tasks"
         time.sleep(3)
     except Exception as e:
         print(e)
+    his_datas = his_datas + [info]
     return his_datas
 
 
@@ -145,6 +151,10 @@ with gr.Blocks(css=css) as demo:
                     label='runtime information')
                 
     with gr.Tab('history'):
+
+        with gr.Row():
+            MK02 = gr.Markdown()
+
         with gr.Row():
             his_pose_image1 = gr.HTML()
             his_res_image1 = gr.HTML()
@@ -172,7 +182,7 @@ with gr.Blocks(css=css) as demo:
     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__":
 

utils.py

@@ -57,7 +57,8 @@ def getAllInfs(apiUrl, openId, apiKey, clientIp):
             records = ret.json()['data']['records']
             for record in records:
                 res.append({'pose':OssUrl+record['body_url']+"?thumbnail/768x768>", 
-                    'res':OssUrl+record['showUrl']+"?thumbnail/768x768>"})
+                    'res':OssUrl+record['showUrl']+"?thumbnail/768x768>",
+                    'state':int(record['state'])})
                 # res.append({'pose':OssUrl+record['body_url'], 
                 #     'res':OssUrl+record['showUrl']})
     return res