Update app.py

app.py

@@ -340,18 +340,25 @@ def fonc_load_gallery(id_session,gallery):
 def fonc_start(id_session,id_module):
     task_actu={}
     model_actu=""
+    print("in fonc\n")
     for model_plus_tasks in cache_list_task[f"{id_session}"]:
         if model_plus_tasks["id_module"]==id_module:
             model_actu=model_plus_tasks["model"]
             task_actu=model_plus_tasks["task"][0]
+            
+            print(f"find model : {model_actu}\n")
+            
     if model_actu=="":
         for model_plus_tasks in cache_list_task[f"{id_session}"]:
             if model_plus_tasks["id_module"]==-1 and model_actu=="":
                 model_actu=model_plus_tasks["model"]
                 task_actu=model_plus_tasks["task"][0]
     if model_actu=="":#crash
+        print("CRASH\n")
         return None,None
+    print("begin gen image\n")
     result=gen_fn(model_actu, task_actu["prompt"], task_actu["nprompt"], task_actu["height"], task_actu["width"], task_actu["steps"], task_actu["cfg"], task_actu["seed"])
+    print("image generate !\n")
     if result!=None:
         result=gr.Image(result)
         id_image=len(cache_image[f"{id_session}"])
@@ -364,6 +371,14 @@ def fonc_start(id_session,id_module):
         task_actu["id_image"]=id_image
         cache_image[f"{id_session}"].append(result.value)
         cache_id_image[f"{id_session}"].append(task_actu)
+        print("image saved\n")
+    else:
+        print("fail to generate\n")
+    num_task_undo=0
+    for model_plus_tasks in cache_list_task[f"{id_session}"]:
+        for task in model_plus_tasks["task"]:
+            num_task_undo+=1
+    print(f"\n {num_task_undo} tasks undo\n")
     return result
             
 def make_me():