Update app.py

app.py

@@ -1,10 +1,17 @@
 import gradio as gr
+# import os
+# import sys
+# from pathlib import Path
 import time
 
-models = [
+models =[
+    "enhanceaiteam/Flux-uncensored",
+    "CultriX/flux-nsfw-highress",
     "pimpilikipilapi1/NSFW_master",
     "DiegoJR1973/NSFW-TrioHMH-Flux",
+    "lustlyai/Flux_Lustly.ai_Uncensored_nsfw_v1",
     "prashanth970/flux-lora-uncensored",
+    "black-forest-labs/FLUX.1-dev",
 ]
 
 model_functions = {}
@@ -16,7 +23,7 @@ for model_path in models:
         def the_fn(txt):
             return None
         model_functions[model_idx] = gr.Interface(fn=the_fn, inputs=["text"], outputs=["image"])
-    model_idx += 1
+    model_idx+=1
 
 
 def send_it_idx(idx):
@@ -42,12 +49,14 @@ def all_task_end(cnt,t_stamp):
     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
 
@@ -62,36 +71,44 @@ 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():
+    with gr.Column(scale=12):
+        # with gr.Row():
+        #     gr.Markdown("""- Primary prompt: 你想画的内容(英文单词，如 a cat, 加英文逗号效果更好；点 Improve 按钮进行完善)\n- Real prompt: 完善后的提示词，出现后再点右边的 Run 按钮开始运行""")
         with gr.Row():
-            with gr.Column():
-                primary_prompt = gr.Textbox(label="Prompt", value="")
-            with gr.Column():
+            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")
+                    run=gr.Button("Run",variant="primary")
+                    clear_btn=gr.Button("Clear")
         with gr.Row():
             sd_outputs = {}
             model_idx = 1
             for model_path in models:
                 with gr.Column(scale=3, min_width=200):
-                    with gr.Group():
+                    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=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],
-            show_progress=True)  # Removed 'every=1'
+            every=1,
+            show_progress=True)
 
         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])
@@ -100,7 +117,14 @@ with gr.Blocks(title="SD Models") as my_interface:
         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,
@@ -112,4 +136,5 @@ with gr.Blocks(title="SD Models") as my_interface:
             tog_box,
             cancels=[*list(runs_dict.values())])
 
-my_interface.launch(inline=True, show_api=False)
+my_interface.queue(concurrency_count=100, status_update_rate=1)
+my_interface.launch(inline=True, show_api=False)