Update app.py

app.py

@@ -1,16 +1,9 @@
 import gradio as gr
-# import os
-# import sys
-# from pathlib import Path
 import time
 
-models =[
-
+models = [
     "John6666/ultimate-realistic-mix-v2-sdxl"
-
-
 ]
-    
 
 model_functions = {}
 model_idx = 1
@@ -19,9 +12,8 @@ for model_path in models:
         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"])
+            return None, None, None
+        model_functions[model_idx] = gr.Interface(fn=the_fn, inputs=["text"], outputs=["image", "image", "image"])
         
     model_idx+=1
     
@@ -70,19 +62,15 @@ def all_task_start():
 
 def clear_fn():
     nn = len(models)
-    return tuple([None, *[None for _ in range(nn)]])
+    return tuple([None, *[None for _ in range(nn*3)]])
 
 
 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():
             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")
@@ -92,7 +80,7 @@ with gr.Blocks(title="SD Models") as my_interface:
             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)
+                        sd_outputs[model_idx] = [gr.Image(label=model_path), gr.Image(label=model_path), gr.Image(label=model_path)]
                     pass
                 model_idx += 1
                 pass
@@ -115,20 +103,15 @@ with gr.Blocks(title="SD Models") as my_interface:
         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]])
+            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())],
+            [primary_prompt, *[image for images in sd_outputs.values() for image in images]],
             cancels=[*list(runs_dict.values())])
         tog_box.change(
             clear_it,