Update app.py

app.py

@@ -360,8 +360,11 @@ with gr.Blocks() as demo:
         else:
             cost_time = 0
         status, video_url = get_result_with_task_id(task_id)
-        task_status[task_id]["status"] = status
-        task_status[task_id]["url"] = video_url
+        if task_status not in task_status:
+            task_status[task_id] = {"status": status, "url": video_url, "time": time.time()}
+        else:
+            task_status[task_id]["status"] = status
+            task_status[task_id]["url"] = video_url
         waiting_time = get_waiting_time(task_id)
         value = task_status.get(task_id, {"value": 100})["value"]
         value = min(value, int(cost_time*100/waiting_time))