Update app.py

app.py

@@ -293,15 +293,15 @@ def add_gallery(image, model_str, gallery):
 def reset_gallery(gallery):
     return add_gallery(None,"",[])
 
-def load_gallery(gallery):
+def load_gallery(gallery,id):
     gallery = reset_gallery(gallery)
-    for c in cache_image:
+    for c in cache_image[f"{id}"]:
         gallery=add_gallery(c[0],c[1],gallery)
     return gallery
-def load_gallery_actu(gallery):
+def load_gallery_actu(gallery,id):
     gallery = reset_gallery(gallery)
     #for c in cache_image_actu:
-    for c in sorted(cache_image_actu, key=itemgetter(1)):
+    for c in sorted(cache_image[f"{id}"], key=itemgetter(1)):
         gallery=add_gallery(c[0],c[1],gallery)
     return gallery
 
@@ -439,15 +439,15 @@ def make_me():
                     with gr.Column():
                         b21 = gr.Button('Reset Gallery')
                         b22 = gr.Button('Reset Gallery All')
-                    b11.click(load_gallery_actu,gallery,gallery)
-                    b12.click(load_gallery,gallery,gallery)
-                    b21.click(reset_gallery,gallery,gallery)
-                    b22.click(reset_cache_image,[],gallery)
+                    b11.click(load_gallery_actu,[gallery,id_session],gallery)
+                    b12.click(load_gallery,[gallery,id_session],gallery)
+                    b21.click(reset_gallery,[gallery,id_session],gallery)
+                    b22.click(reset_cache_image,[id_session],gallery)
                     for m, o in zip(current_models_g, output_g):
                         #o.change(add_gallery, [o, m, gallery], [gallery])
-                        o.change(add_cache_image,[o,m],[])
-                        o.change(add_cache_image_actu,[o,m],[])
-                gen_button.click(reset_cache_image_actu, [], [])
+                        o.change(add_cache_image,[o,m,id_session],[])
+                        o.change(add_cache_image_actu,[o,m,id_session],[])
+                gen_button.click(reset_cache_image_actu, [id_session], [])
                 
             with gr.Group():
                 with gr.Row():