Update app.py

app.py

@@ -297,7 +297,6 @@ def crea_list_task(id_session,list_param,list_models_to_gen,nb_images_by_prompt)
             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):
@@ -323,6 +322,50 @@ def aff_models_perso(txt_list_perso,models=models):
             if start==-1:
                 t1=False
     return gr.Dropdown(choices=[["",list_perso]], value=list_perso )
+
+def add_gallery(image, model_str, gallery):
+    if gallery is None: gallery = []
+    #with lock:
+    if image is not None: gallery.append((image, model_str))
+    return gallery
+
+def reset_gallery(gallery):
+    return add_gallery(None,"",[])
+
+def load_gallery(gallery,id):
+    gallery = reset_gallery(gallery)
+    for c in cache_image[f"{id}"]:
+        gallery=add_gallery(c[0],c[1],gallery)
+    return gallery
+
+def fonc_start(id_session,id_module):
+    task_actu={}
+    model_actu=""
+    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]
+    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
+        return None,None
+    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"])
+    if result!=None:
+        result=gr.Image(result)
+        id_image=len(cache_image[f"{id_session}"])
+        for model_plus_tasks in cache_list_task[f"{id_session}"]:
+            if model_plus_tasks["id_module"]==id_module:
+                model_plus_tasks["task"].remove(task_actu)
+            if len(model_plus_tasks["task"])==0:
+                cache_list_task[f"{id_session}"].remove(model_plus_tasks)
+            
+        task_actu["id_image"]=id_image
+        cache_image[f"{id_session}"].append(result.value)
+        cache_id_image[f"{id_session}"].append(task_actu)
+    return result
             
 def make_me():
         with gr.Column():
@@ -394,7 +437,7 @@ 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],[])#################
+            list_models_to_gen.change(crea_list_task,[id_session,list_param,list_models_to_gen,nb_images_by_prompt],[])
             
 
             
@@ -405,17 +448,24 @@ def make_me():
 
                 with gr.Row():
                     outputs=[]
-                    outputs_name=[]
+                    id_modules=[]
                     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))
+                        id_modules.append(gr.Number(i,interactive=False,render=False))
+                    for o,i in zip(outputs,id_modules):
+                        o.change(fonc_start,[id_session,i],[o])
+                    gallery = gr.Gallery(label="Output", show_download_button=True, elem_classes="gallery",
+                                    interactive=False, show_share_button=True, container=True, format="png",
+                                    preview=True, object_fit="cover",columns=4,rows=4) 
+                    button_load_gallery=("Load Gallery")
+                    button_load_gallery.click(fonc_load_gallery,[id_session,gallery],[])
                     
-
-
-
-
-
-
+def fonc_load_gallery(id_session,gallery):
+    gallery = reset_gallery(gallery)
+    for i in range(len(cache_image[f"{id_session}"])):
+        gallery=add_gallery(cache_image[f"{id_session}"][i],cache_image[f"{id_session}"][i]["model"],gallery)
+    return gallery
+