Update app.py

app.py

@@ -90,7 +90,7 @@ async def infer(model_str, prompt, nprompt="", height=None, width=None, steps=No
             noise += " "
     task = asyncio.create_task(asyncio.to_thread(models_load[model_str].fn,
                                prompt=f'{prompt} {noise}', negative_prompt=nprompt, **kwargs, token=HF_TOKEN))
-    await asyncio.sleep(0)
+    await asyncio.sleep(3)
     try:
         result = await asyncio.wait_for(task, timeout=timeout)
     except (Exception, asyncio.TimeoutError) as e:
@@ -337,9 +337,14 @@ def reset_gallery(gallery):
     return add_gallery(None,"",[])
 def fonc_load_gallery(id_session,gallery):
     gallery = reset_gallery(gallery)
-    print(cache_image[f"{id_session}"])
     for i in range(len(cache_image[f"{id_session}"])):
-        gallery=add_gallery(cache_image[f"{id_session}"][i],cache_id_image[f"{id_session}"][i]["model"],gallery)
+        gallery=add_gallery(cache_image[f"{id_session}"][i]["image"],cache_image[f"{id_session}"][i]["model"],gallery)
+    return gallery
+def fonc_load_gallery_by_model(id_session,gallery,models,index):
+    gallery = reset_gallery(gallery)
+    for i in range(len(cache_image[f"{id_session}"])):
+        if cache_image[f"{id_session}"][i]["model"]==models[index]:
+            gallery=add_gallery(cache_image[f"{id_session}"][i]["image"],cache_image[f"{id_session}"][i]["model"],gallery)
     return gallery
 
 def fonc_start(id_session,id_module,s,cont):
@@ -383,9 +388,10 @@ def fonc_start(id_session,id_module,s,cont):
             
         task_actu["id_image"]=id_image
         task_actu["model"]=model_actu
-        #cache_image[f"{id_session}"].append(result.value)
-        cache_image[f"{id_session}"].append(result)
-        cache_id_image[f"{id_session}"].append(task_actu)
+        task_actu["image"]=result
+        #cache_image[f"{id_session}"].append(result)
+        #cache_id_image[f"{id_session}"].append(task_actu)
+        cache_image[f"{id_session}"].append(task_actu)
         print("image saved\n")
     else:
         print("fail to generate\n")
@@ -396,17 +402,18 @@ def fonc_start(id_session,id_module,s,cont):
     print(f"\n {num_task_undo} tasks undo\n")
     return result , gr.Textbox(s+"1")
 
-def test_change(s):
-    if s == "11111":
-        return gr.Textbox(s)
-    else:
-        print(s)
-        return gr.Textbox(s+"1")
-    #return None
-def test_start(s):
-    print("go\n")
+def fonc_init(s):
     return gr.Textbox(s+"1")
 
+def index_gallery_first():
+    return gr.Number(0)
+def index_gallery_next(i,list_models):
+    iT=i+1
+    return gr.Number(iT%len(list_models))
+def index_gallery_prev(i,list_models):
+    iT=i-1
+    return gr.Number(iT%len(list_models))
+
 def make_me():
         with gr.Column():
             id_session=gr.Number(visible=False,value=0)
@@ -508,15 +515,26 @@ def make_me():
                         #o.change(test_change,[],[])
                         s.change(fonc_start,[id_session,i,s,cont],[o,s])
                         #button_start.click(lambda : gr.Image(None),[],[o])
-                        gen_event = gr.on(triggers=[button_start.click], fn=test_start,inputs=[s], outputs=[s])
+                        gen_event = gr.on(triggers=[button_start.click], fn=fonc_init,inputs=[s], outputs=[s])
                         
                     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=gr.Button("Load Gallery")
-                    button_load_gallery.click(fonc_load_gallery,[id_session,gallery],[gallery])
-                    
+                    with gr.Column():
+                        button_load_gallery=gr.Button("Load Gallery All")
+                        button_load_gallery.click(fonc_load_gallery,[id_session,gallery],[gallery])
 
+                        index_gallery=gr.Number(0,visible=False)
+                        button_load_gallery_first=gr.Button("Load Gallery first model")
+                        button_load_gallery_first.click(index_gallery_first,[],[index_gallery])
+                        with gr.Row():
+                            button_load_gallery_next=gr.Button("Next model")
+                            button_load_gallery_next.click(index_gallery_next,[index_gallery,list_models_to_gen],[index_gallery])
+                            button_load_gallery_prev=gr.Button("Prev model")
+                            button_load_gallery_prev.click(index_gallery_prev,[index_gallery,list_models_to_gen],[index_gallery])
+                            index_gallery.change(fonc_load_gallery_by_model,[id_session,gallery,list_models_to_gen,index_gallery],[gallery])
+                    
+