Update app.py

app.py

@@ -323,8 +323,10 @@ def reset_cache_image_actu(id,cache_image_actu=cache_image_actu):
     cache_image_actu[f"{id}"].clear()
     return 
 def reset_cache_image_all_sessions(cache_image=cache_image):
-    cache_image.clear()
+    for key, list in cache_image:
+        list.clear()
     return 
+    
 def set_session(id):
     if id==0:
         randTemp=randint(1,MAX_SEED)
@@ -333,6 +335,14 @@ def set_session(id):
         return gr.Number(visible=False,value=randTemp)
     else :
         return id
+def print_info_sessions():
+    lenTot=0
+    print("numbre of sessions : "+str(len(cache_image)))
+    for key, list in cache_image:
+        print(key+str(len(list)))
+        lenTot+=len(list)
+    print("images total = "+str(lenTot))
+    return
 
 def disp_models(group_model_choice,nb_rep=nb_rep):
     listTemp=[]
@@ -472,15 +482,20 @@ def make_me():
                     with gr.Column():
                         b21 = gr.Button('Reset Gallery')
                         b22 = gr.Button('Reset Gallery All')
+                        b23 = gr.Button('Reset All Sessions')
+                        b24 = gr.Button('print info sessions')
                     b11.click(load_gallery_actu,[gallery,id_session],gallery)
                     b12.click(load_gallery,[gallery,id_session],gallery)
                     b21.click(reset_gallery,[gallery],gallery)
                     b22.click(reset_cache_image,[id_session],gallery)
+                    b23.click(reset_cache_image_all_sessions,[],[])
+                    b24.click(print_info_sessions,[],[])
                     for m, o in zip(current_models, output):
                         #o.change(add_gallery, [o, m, gallery], [gallery])
                         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], [])
+                gen_button.click(lambda id:gr.Button('Load Galerry All ('+str(len(cache_image_actu[f"{id}"]))+")"), [id_session], [b12])
                 
             with gr.Group():
                 with gr.Row():