Update app.py

app.py

@@ -1,9 +1,22 @@
 import gradio as gr
+# import os
+# import sys
+# from pathlib import Path
 import time
 
-models = [
-    "Yntec/ChickFlick"
+models =[
+    "mann-e/Mann-E_Dreams",
+    "Yntec/ChickFlick",
+    "John6666/ultimate-realistic-mix-v2-sdxl",
+    "Yntec/CrystalReality",
+    "John6666/9527-detail-realistic-xl-v55mix-sdxl",
+    "John6666/epicrealism-xl-v8kiss-sdxl",
+    "John6666/wai-real-mix-v8-sdxl",
+    "John6666/real-vis-xl-v40-sdxl",
+    "Yntec/DegreesOfFreedom",
+
 ]
+    
 
 model_functions = {}
 model_idx = 1
@@ -12,8 +25,9 @@ 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, None, None
-        model_functions[model_idx] = gr.Interface(fn=the_fn, inputs=["text"], outputs=["image", "image", "image"])
+            
+            return None
+        model_functions[model_idx] = gr.Interface(fn=the_fn, inputs=["text"], outputs=["image"])
         
     model_idx+=1
     
@@ -62,15 +76,19 @@ def all_task_start():
 
 def clear_fn():
     nn = len(models)
-    return tuple([None, *[None for _ in range(nn*3)]])
+    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():
             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")
@@ -80,7 +98,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), gr.Image(label=model_path), gr.Image(label=model_path)]
+                        sd_outputs[model_idx] = gr.Image(label=model_path)
                     pass
                 model_idx += 1
                 pass
@@ -103,15 +121,20 @@ 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, *[image for images in sd_outputs.values() for image in images]],
+            [primary_prompt, *list(sd_outputs.values())],
             cancels=[*list(runs_dict.values())])
         tog_box.change(
             clear_it,