Update app.py

app.py

@@ -2,61 +2,122 @@ import gradio as gr
 import time
 
 
-model1 = gr.Interface.load("models/pimpilikipilapi1/NSFW_master", preprocess=True, postprocess=False)
-model2 = gr.Interface.load("models/prashanth970/flux-lora-uncensored", preprocess=True, postprocess=False)
-model3 = gr.Interface.load("models/DiegoJR1973/NSFW-TrioHMH-Flux", preprocess=True, postprocess=False)
+models =[
+    "pimpilikipilapi1/NSFW_master",
+    "DiegoJR1973/NSFW-TrioHMH-Flux",
+    "prashanth970/flux-lora-uncensored",
+]
+
+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
 
-models = {
-    1: model1,
-    2: model2,
-    3: model3
-}
 
 def send_it_idx(idx):
     def send_it_fn(prompt):
-        model = models.get(idx)
-        if model:
-            return model(prompt)
-        return None
+        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 + 360
+    et = time.time()
+    if et > to and t_stamp != 0:
+        d = gr.update(value=0)
+        tog = gr.update(value=1)
+    else:
+        if cnt != 0:
+            d = gr.update(value=et)
+        else:
+            d = gr.update(value=0)
+        tog = gr.update(value=0)
+        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():
-    return None, None, None
+    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():
-            primary_prompt = gr.Textbox(label="Prompt", value="")
-            with gr.Row():
-                run = gr.Button("Run", variant="primary")
-                clear_btn = gr.Button("Clear")
-
-        sd_outputs = {}
-        for idx, model in models.items():
-            with gr.Column(scale=3, min_width=200):
-                with gr.Box():
-                    sd_outputs[idx] = gr.Image(label=f"Model {idx}")
+            with gr.Row(scale=6):
+                primary_prompt=gr.Textbox(label="Prompt", value="")
+            with gr.Row(scale=6):
+                with gr.Row():
+                    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.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)
 
-        def all_task_start():
-            t_stamp = time.time()
-            return gr.update(value=t_stamp), gr.update(value=t_stamp), gr.update(value=0)
+        start_box.change(
+            all_task_end,
+            [start_box, end_box],
+            [start_box, tog_box],
+            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])
-
-        for idx, model in models.items():
-            run.click(model, inputs=[primary_prompt], outputs=[sd_outputs[idx]])
+        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
 
         clear_btn.click(
             clear_fn,
             None,
-            [primary_prompt, *list(sd_outputs.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())])
 
 my_interface.queue(concurrency_count=100, status_update_rate=1)
-my_interface.launch(inline=True, show_api=False)
+my_interface.launch(inline=True, show_api=False)