Update app.py

app.py

@@ -23,6 +23,8 @@ nb_mod_dif=20
 nb_models=nb_mod_dif*nb_rep
 
 cache_image={}
+cache_id_image={}
+cache_list_task={}
 
 
 def load_fn(models):
@@ -156,6 +158,9 @@ def set_session(id):
     if id==0:
         randTemp=randint(1,MAX_SEED)
         cache_image[f"{randTemp}"]=[]
+        cache_id_image[f"{randTemp}"]=[]
+        
+        cache_list_task[f"{randTemp}"]=[]
         return gr.Number(visible=False,value=randTemp)
     else :
         return id
@@ -281,7 +286,18 @@ def fonc_load_info_custom(nb_of_models_to_gen,list_model_custom,index_first_mode
         str_temp+=f"\"{list_model_custom[i+index_first_model]}\",\n"
     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):
+    cache_list_task[f"{id_session}"]=[]
+    dict_temp={}
+    for m in list_models_to_gen:
+        dict_temp["model"]=m
+        dict_temp["id_module"]=-1
+        dict_temp["task"]=[]
+        for p in list_param:
+            for i in range(nb_images_by_prompt):
+                dict_temp["task"].append({"txt_input":p[0],"neg_input":p[1],"width":p[2],"height":p[3],"steps":p[4],"cfg":p[5],"seed":p[6]})
+        cache_list_task[f"{id_session}"].append(dict_temp)
+    print(cache_list_task[f"{id_session}"])###########################################
 
     
 def cutStrg(longStrg,start,end):
@@ -310,6 +326,7 @@ def aff_models_perso(txt_list_perso,models=models):
             
 def make_me():
         with gr.Column():
+            id_session=gr.Number(visible=False,value=0)
             with gr.Group():
                 with gr.Row():
                     with gr.Column(scale=4):
@@ -344,6 +361,7 @@ def make_me():
                 outputs=[txt_input,neg_input,width,height,steps,cfg,seed],
             )
             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():
@@ -375,17 +393,22 @@ 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],[])#################
+            
 
+            
             with gr.Column():
                 with gr.Row():
-                    id_session=gr.Number(visible=False,value=0)
                     nb_req_simult=gr.Number(50,"Number simultaneous request")
                     nb_req_simult_max_by_mod=gr.Number(2,"Number max simultaneous request by model")
 
                 with gr.Row():
                     outputs=[]
+                    outputs_name=[]
                     for i in range(nb_req_simult.value):
                         outputs.append(gr.Image(None,interactive=False,render=False))
+                        outputs_name.append(gr.Number(i,interactive=False,render=False))