Update app.py

app.py

@@ -70,6 +70,7 @@ def all_task_end(cnt,t_stamp):
     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)
@@ -77,20 +78,20 @@ def all_task_start():
 
 def clear_fn():
     nn = len(models)
-    return tuple([None for _ in range(nn)])
+    return tuple([None, *[None for _ in range(nn)]])
 
 
 
 with gr.Blocks() 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():
+        #     gr.Markdown("""- Primary prompt: 你想画的内容(英文单词，如 a cat, 加英文逗号效果更好；点 Improve 按钮进行完善)\n- Real prompt: 完善后的提示词，出现后再点右边的 Run 按钮开始运行""")
         with gr.Row():
             with gr.Column(scale=6):
-                primary_prompt=gr.Textbox(label="Primary prompt")
-                real_prompt=gr.Textbox(label="Real prompt")
+                primary_prompt=gr.Textbox(label="Prompt")
+                # real_prompt=gr.Textbox(label="Real prompt")
             with gr.Column(scale=6):
-                improve_prompts_btn=gr.Button("Improve")
+                # improve_prompts_btn=gr.Button("Improve")
                 with gr.Row():
                     run=gr.Button("Run")
                     clear_btn=gr.Button("Clear")
@@ -122,26 +123,26 @@ with gr.Blocks() as my_interface:
         runs_dict = {}
         model_idx = 1
         for model_path in models:
-            runs_dict[model_idx] = run.click(interfaces[model_idx], inputs=[real_prompt], outputs=[sd_outputs[model_idx]])
+            runs_dict[model_idx] = run.click(interfaces[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=[real_prompt],
-            cancels=list(runs_dict.values()))
+        # 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, real_prompt, *list(sd_outputs.values())],
-            cancels=[improve_prompts_btn_clicked, *list(runs_dict.values())])
+            [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()), improve_prompts_btn_clicked])
+            cancels=[*list(runs_dict.values())])
 
 my_interface.queue(concurrency_count=600, status_update_rate=1)
 my_interface.launch(inline=True, show_api=False)