Update app.py

app.py

@@ -1,106 +1,148 @@
 import gradio as gr
-import os
-import sys
-from pathlib import Path
-
-models = [
-    {"name": "Claudfuen 1", "url": "sd-concepts-library/ahx-model-6"},
-    {"name": "Deliberate", "url": "Masagin/Deliberate"},
-    {"name": "Seek Art Mega", "url": "coreco/seek.art_MEGA"},
-    {"name": "Realistic Vision 1.4", "url": "SG161222/Realistic_Vision_V1.4"},
-    {"name": "Dreamshaper", "url": "Lykon/DreamShaper"},
+# import os
+# import sys
+# from pathlib import Path
+import time
+
+models =[
+    "stablediffusionapi/all-526-animated",
+    "kubanemil/AnyLORA",
+    "andite/anything-v4.0",
+    "stablediffusionapi/anything-v5",
+    "gsdf/Counterfeit-V2.0",
+    "dreamlike-art/dreamlike-anime-1.0",
+    "eimiss/EimisAnimeDiffusion_1.0v",
+    "DucHaiten/DucHaitenAnime",
+    "xiaolxl/GuoFeng3",
+    "andite/pastel-mix",
+    "prompthero/openjourney-v4",
+    "aipicasso/picasso-diffusion-1-1",
+    "stablediffusionapi/rev-anim",
+    "coreml/coreml-ReV-Animated",
 ]
 
-current_model = models[0]
 
-text_gen = gr.Interface.load("spaces/Omnibus/MagicPrompt-Stable-Diffusion_link") 
-
-models2 = []
-for model in models:
-    model_url = f"models/{model['url']}"
-    loaded_model = gr.Interface.load(model_url, live=True, preprocess=True)
-    models2.append(loaded_model)
-
-
-def text_it(inputs, text_gen=text_gen):
-    return text_gen(inputs)
-
-
-def set_model(current_model_index):
-    global current_model
-    current_model = models[current_model_index]
-    return gr.update(value=f"{current_model['name']}")
-
-
-def send_it(inputs, model_choice):
-    proc = models2[model_choice]
-    return proc(inputs)
-
-
-with gr.Blocks() as myface:
-    gr.HTML(
-
-    )
-
-    with gr.Row():
+model_functions = {}
+model_idx = 1
+for model_path in models:
+    try:
+        model_functions[model_idx] = gr.Interface.load(f"models/{model_path}", live=False, preprocess=True, postprocess=False)
+    except Exception as error:
+        def the_fn(txt):
+            return None
+        model_functions[model_idx] = gr.Interface(fn=the_fn, inputs=["text"], outputs=["image"])
+    model_idx+=1
+
+
+def send_it_idx(idx):
+    def send_it_fn(prompt):
+        output = (model_functions.get(str(idx)) or model_functions.get(str(1)))(prompt)
+        return output
+    return send_it_fn
+
+def get_prompts(prompt_text):
+    return prompt_text
+
+def clear_it(val):
+    if int(val) != 0:
+        val = 0
+    else:
+        val = 0
+        pass
+    return val
+
+def all_task_end(cnt,t_stamp):
+    to = t_stamp + 60
+    et = time.time()
+    if et > to and t_stamp != 0:
+        d = gr.update(value=0)
+        tog = gr.update(value=1)
+        #print(f'to: {to}  et: {et}')
+    else:
+        if cnt != 0:
+            d = gr.update(value=et)
+        else:
+            d = gr.update(value=0)
+        tog = gr.update(value=0)
+        #print (f'passing:  to: {to}  et: {et}')
+        pass
+    return d, tog
+
+def all_task_start():
+    print("\n\n\n\n\n\n\n")
+    t = time.gmtime()
+    t_stamp = time.time()
+    current_time = time.strftime("%H:%M:%S", t)
+    return gr.update(value=t_stamp), gr.update(value=t_stamp), gr.update(value=0)
+
+def clear_fn():
+    nn = len(models)
+    return tuple([None, *[None for _ in range(nn)]])
+
+
+
+with gr.Blocks(title="SD Models") as my_interface:
+    with gr.Column(scale=12):
+        # with gr.Row():
+        #     gr.Markdown("""- Primary prompt: 你想画的内容(英文单词，如 a cat, 加英文逗号效果更好；点 Improve 按钮进行完善)\n- Real prompt: 完善后的提示词，出现后再点右边的 Run 按钮开始运行""")
         with gr.Row():
-            input_text = gr.Textbox(label="Prompt idea", lines=1)
-            # Model selection dropdown
-            model_name1 = gr.Dropdown(
-                label="Choose Model",
-                choices=[m["name"] for m in models],
-                type="index",
-                value=current_model["name"],
-                interactive=True,
-            )
+            with gr.Row(scale=6):
+                primary_prompt=gr.Textbox(label="Prompt", value="")
+                # real_prompt=gr.Textbox(label="Real prompt")
+            with gr.Row(scale=6):
+                # improve_prompts_btn=gr.Button("Improve")
+                with gr.Row():
+                    run=gr.Button("Run",variant="primary")
+                    clear_btn=gr.Button("Clear")
         with gr.Row():
-            see_prompts = gr.Button("Generate Prompts")
-            run = gr.Button("Generate Images", variant="primary")
-    
-    with gr.Row():
-        output1 = gr.Image(label="")
-        output2 = gr.Image(label="")
-    with gr.Row():
-        magic1 = gr.Textbox(lines=2)
-        magic2 = gr.Textbox(lines=2)
-    with gr.Row():            
-        output3 = gr.Image(label="")
-        output4 = gr.Image(label="")
-    with gr.Row():
-        magic3 = gr.Textbox(lines=2)
-        magic4 = gr.Textbox(lines=2)
-    with gr.Row():
-        output5 = gr.Image(label="")
-        output6 = gr.Image(label="")
-    with gr.Row():
-        magic5 = gr.Textbox(lines=2)
-        magic6 = gr.Textbox(lines=2)
-    with gr.Row():   
-        output7 = gr.Image(label="")
-        output8 = gr.Image(label="")
-    with gr.Row():
-        magic7 = gr.Textbox(lines=2)
-        magic8 = gr.Textbox(lines=2)
-
-    model_name1.change(set_model, inputs=model_name1, outputs=[output1, output2, output3, output4, output5, output6, output7, output8])
-
-    run.click(send_it, inputs=[magic1, model_name1], outputs=[output1])
-    run.click(send_it, inputs=[magic2, model_name1], outputs=[output2])
-    run.click(send_it, inputs=[magic3, model_name1], outputs=[output3])
-    run.click(send_it, inputs=[magic4, model_name1], outputs=[output4])
-    run.click(send_it, inputs=[magic5, model_name1], outputs=[output5])
-    run.click(send_it, inputs=[magic6, model_name1], outputs=[output6])
-    run.click(send_it, inputs=[magic7, model_name1], outputs=[output7])
-    run.click(send_it, inputs=[magic8, model_name1], outputs=[output8])
-
-    see_prompts.click(text_it, inputs=[input_text], outputs=[magic1])
-    see_prompts.click(text_it, inputs=[input_text], outputs=[magic2])
-    see_prompts.click(text_it, inputs=[input_text], outputs=[magic3])
-    see_prompts.click(text_it, inputs=[input_text], outputs=[magic4])
-    see_prompts.click(text_it, inputs=[input_text], outputs=[magic5])
-    see_prompts.click(text_it, inputs=[input_text], outputs=[magic6])
-    see_prompts.click(text_it, inputs=[input_text], outputs=[magic7])
-    see_prompts.click(text_it, inputs=[input_text], outputs=[magic8])
-
-myface.queue(concurrency_count=200)
-myface.launch(inline=True, show_api=False, max_threads=400)
+            sd_outputs = {}
+            model_idx = 1
+            for model_path in models:
+                with gr.Column(scale=3, min_width=320):
+                    with gr.Box():
+                        sd_outputs[model_idx] = gr.Image(label=model_path)
+                    pass
+                model_idx += 1
+                pass
+            pass
+
+        with gr.Row(visible=False):
+            start_box=gr.Number(interactive=False)
+            end_box=gr.Number(interactive=False)
+            tog_box=gr.Textbox(value=0,interactive=False)
+
+        start_box.change(
+            all_task_end,
+            [start_box, end_box],
+            [start_box, tog_box],
+            every=1,
+            show_progress=False)
+
+        primary_prompt.submit(all_task_start, None, [start_box, end_box, tog_box])
+        run.click(all_task_start, None, [start_box, end_box, tog_box])
+        runs_dict = {}
+        model_idx = 1
+        for model_path in models:
+            runs_dict[model_idx] = run.click(model_functions[model_idx], inputs=[primary_prompt], outputs=[sd_outputs[model_idx]])
+            model_idx += 1
+            pass
+        pass
+
+        # improve_prompts_btn_clicked=improve_prompts_btn.click(
+        #     get_prompts,
+        #     inputs=[primary_prompt],
+        #     outputs=[primary_prompt],
+        #     cancels=list(runs_dict.values()))
+        clear_btn.click(
+            clear_fn,
+            None,
+            [primary_prompt, *list(sd_outputs.values())],
+            cancels=[*list(runs_dict.values())])
+        tog_box.change(
+            clear_it,
+            tog_box,
+            tog_box,
+            cancels=[*list(runs_dict.values())])
+
+my_interface.queue(concurrency_count=600, status_update_rate=1)
+my_interface.launch(inline=True, show_api=False)