Update app.py

app.py

@@ -13,7 +13,7 @@ lock = RLock()
 HF_TOKEN = os.environ.get("HF_TOKEN") if os.environ.get("HF_TOKEN") else None # If private or gated models aren't used, ENV setting is unnecessary.
 
 
-nb_req_simult=50 ########
+nb_req_simult=80 ########
 nb_gallery_model=5
 
 now2 = 0
@@ -29,6 +29,7 @@ cache_image={}
 cache_id_image={}
 cache_list_task={}
 cache_text_actu={}
+from_reload={}
 
 
 def load_fn(models):
@@ -169,19 +170,40 @@ def set_session(id):
         
         cache_list_task[f"{randTemp}"]=[]
         cache_text_actu[f"{randTemp}"]={}
+        from_reload[f"{randTemp}"]=False
         return gr.Number(visible=False,value=randTemp)
     else :
         return id
+
+def fonc_restore_session(id):
+    from_reload[f"{id}"]=True
+    list_param=[]
+    list_models=[]
+    for m in cache_list_task[f"{id}"]:
+        if m["model"] not in list_models:
+            list_models.append(m["model"])
+        for t in m["task"]:
+            if [t["prompt"],t["nprompt"],t["width"],t["height"],t["steps"],t["cfg"],t["seed"]] not in list_param:
+                list_param.append([t["prompt"],t["nprompt"],t["width"],t["height"],t["steps"],t["cfg"],t["seed"]])
+    for t in cache_image[f"{id}"]:
+        if t["model"] not in list_models :
+            list_models.append(t["model"])
+        if [t["prompt"],t["nprompt"],t["width"],t["height"],t["steps"],t["cfg"],t["seed"]] not in list_param:
+            list_param.append([t["prompt"],t["nprompt"],t["width"],t["height"],t["steps"],t["cfg"],t["seed"]])
+
+    cache_text_actu[f"{id}"]["nb_modules_use"]=nb_req_simult
+    cache_text_actu[f"{id}"]["stop"]=False
+    return gr.Dropdown(choices=[["a",list_param]], value=list_param) ,gr.Dataset(samples=list_param), list_models , len(list_models)          
+
 def print_info_sessions():
     lenTot=0
-    print("###################################")
-    print("number of sessions : "+str(len(cache_image)))
+    s=""
+    s+="number of sessions : "+str(len(cache_image))+"\n"
     for key, listT in cache_image.items():
-        print("session "+key+" : "+str(len(listT)))
+        s+="session "+key+" : "+str(len(listT))+"\n"
         lenTot+=len(listT)
-    print("images total = "+str(lenTot))
-    print("###################################")
-    return
+    s+="images total = "+str(lenTot)+"\n"
+    return s
 
 def disp_models(group_model_choice,nb_rep=nb_rep):
     listTemp=[]
@@ -298,6 +320,9 @@ def fonc_load_info_custom(nb_of_models_to_gen,list_model_custom,index_first_mode
     return nb_of_models_to_gen,gr.Textbox(str_temp),gr.Dropdown(choices=[["",list_models_temp]], value=list_models_temp )
 
 def crea_list_task(id_session,list_param,list_models_to_gen,nb_images_by_prompt):
+    if from_reload[f"{id_session}"]==True:
+        from_reload[f"{id_session}"]=False
+        return
     cache_list_task[f"{id_session}"]=[]
     dict_temp={}
     list_progress=[]
@@ -436,7 +461,7 @@ def fonc_start(id_session,id_module,s,cont,list_models_to_gen):
                 i=0
                 for model in list_models_to_gen:
                     if model_actu==model:
-                        cache_text_actu[f"{id_session}"]['progress'][i]=int(((1-(len(model_plus_tasks["task"])/cache_text_actu[f"{id_session}"]["nb_tasks_by_model"]))*8)//1)+1
+                        cache_text_actu[f"{id_session}"]['progress'][i]=int(((1-(len(model_plus_tasks["task"])/cache_text_actu[f"{id_session}"]["nb_tasks_by_model"]))*7)//1)+2
                     i+=1
             if len(model_plus_tasks["task"])==0:
                 cache_list_task[f"{id_session}"].remove(model_plus_tasks)
@@ -519,7 +544,6 @@ def make_me():
         with gr.Tab(" Sort ",visible=False) as tab_p:   
             button_test_pass.click(test_pass_aff,[test_pass],[tab_p]) 
             with gr.Column():
-                id_session=gr.Number(visible=False,value=0)
                 with gr.Group():
                     with gr.Row():
                         with gr.Column(scale=4):
@@ -553,10 +577,24 @@ def make_me():
                     inputs=[txt_input,neg_input,width,height,steps,cfg,seed],
                     outputs=[txt_input,neg_input,width,height,steps,cfg,seed],
                 )
+
+
+                
+                with gr.Accordion("Restore Session",open=False) :
+                    with gr.Row():
+                        text_info_session=gr.Textbox()
+                        with gr.Column():
+                            button_info_session=gr.Button("Get infos sessions")
+                            button_info_session.click(print_info_sessions,[],[text_info_session])
+                            id_session=gr.Number(0,interactive = True,label="ID session",show_label=True)
+                            button_restore_session=gr.Button("Restore Session")
+                    
+
+                
                 add_param.click(fonc_add_param,[list_param,txt_input,neg_input,width,height,steps,cfg,seed],[disp_param.dataset,list_param])
                 add_param.click(set_session,[id_session],[id_session])
                 del_param.click(fonc_del_param,[list_param,txt_input,neg_input,width,height,steps,cfg,seed],[disp_param.dataset,list_param])
-    
+                
                 with gr.Row():
                     list_models_to_gen=gr.Dropdown(choices=[["",[]]], value=[], visible=False)
                     disp_info=gr.Textbox(label="Info")
@@ -586,7 +624,6 @@ def make_me():
                             
                             load_model_custom=gr.Button("Load Models Custom")
                             load_model_custom.click(fonc_load_info_custom,[nb_of_models_to_gen,list_model_custom,index_first_model_custom],[nb_of_models_to_gen,disp_info,list_models_to_gen])
-                
                 list_models_to_gen.change(crea_list_task,[id_session,list_param,list_models_to_gen,nb_images_by_prompt],[])
                 
     
@@ -720,8 +757,11 @@ def make_me():
                                 
     
 
+            button_restore_session.click(fonc_restore_session,[id_session],[list_param,disp_param.dataset,list_models_to_gen,nb_of_models_to_gen])
 
 
+    
+
 js_code = """
     
     console.log('ghgh');