Update app.py

app.py

@@ -176,103 +176,105 @@ with gr.Blocks(theme='NoCrypt/miku@>=1.2.2', fill_width=True, css=CSS) as demo:
                <img src="https://api.visitorbadge.io/api/visitors?path=https%3A%2F%2Fgunship999-SexyImages.hf.space&countColor=%23263759" />
                </a>""")
     
-    with gr.Column(scale=2):
-        with gr.Accordion("Model Selection", open=True):
-            model_choice = gr.CheckboxGroup(
-                models,
-                label=f'Choose up to {int(num_models)} models',
-                value=default_models,
-                interactive=True
-            )
+    with gr.Tab('6 Models'):
+        with gr.Column(scale=2):
+            with gr.Group():
+                txt_input = gr.Textbox(label='Your prompt:', value=preSetPrompt, lines=3, autofocus=1)
+                neg_input = gr.Textbox(label='Negative prompt:', value=negPreSetPrompt, lines=1)
+                with gr.Accordion("Advanced", open=False, visible=True):
+                    with gr.Row():
+                        width = gr.Slider(label="Width", info="If 0, the default value is used.", maximum=1216, step=32, value=0)
+                        height = gr.Slider(label="Height", info="If 0, the default value is used.", maximum=1216, step=32, value=0)
+                    with gr.Row():
+                        steps = gr.Slider(label="Number of inference steps", info="If 0, the default value is used.", maximum=100, step=1, value=0)
+                        cfg = gr.Slider(label="Guidance scale", info="If 0, the default value is used.", maximum=30.0, step=0.1, value=0)
+                        seed = gr.Slider(label="Seed", info="Randomize Seed if -1.", minimum=-1, maximum=MAX_SEED, step=1, value=-1)
+                        seed_rand = gr.Button("Randomize Seed 🎲", size="sm", variant="secondary")
+                        seed_rand.click(randomize_seed, None, [seed], queue=False)
+            with gr.Row():
+                gen_button = gr.Button(f'Generate up to {int(num_models)} images', variant='primary', scale=3)
+                random_button = gr.Button(f'Randomize Models', variant='secondary', scale=1)
+                #stop_button = gr.Button('Stop', variant='stop', interactive=False, scale=1)
+                #gen_button.click(lambda: gr.update(interactive=True), None, stop_button)
+            gr.Markdown("", elem_classes="guide")
 
-        with gr.Group():
-            txt_input = gr.Textbox(
-                label='Your prompt:', 
-                value=preSetPrompt, 
-                lines=3, 
-                autofocus=1
-            )
-            neg_input = gr.Textbox(
-                label='Negative prompt:', 
-                value=negPreSetPrompt, 
-                lines=1
-            )
-            with gr.Accordion("Advanced Settings", open=False):
-                with gr.Row():
-                    width = gr.Slider(label="Width", maximum=1216, step=32, value=0)
-                    height = gr.Slider(label="Height", maximum=1216, step=32, value=0)
+        with gr.Column(scale=1):
+            with gr.Group():
                 with gr.Row():
-                    steps = gr.Slider(label="Steps", maximum=100, step=1, value=0)
-                    cfg = gr.Slider(label="Guidance Scale", maximum=30.0, step=0.1, value=0)
-                    seed = gr.Slider(label="Seed", minimum=-1, maximum=MAX_SEED, step=1, value=-1)
-                    seed_rand = gr.Button("🎲", size="sm", elem_classes="primary-btn")
-                    seed_rand.click(randomize_seed, None, [seed], queue=False)
+                    output = [gr.Image(label=m, show_download_button=True, elem_classes="output",
+                              interactive=False, width=112, height=112, show_share_button=False, format="png",
+                              visible=True) for m in default_models]
+                    current_models = [gr.Textbox(m, visible=False) for m in default_models]
 
-        with gr.Row():
-            gen_button = gr.Button(
-                f'Generate {int(num_models)} Images', 
-                variant='primary', 
-                scale=3,
-                elem_classes="primary-btn"
-            )
-            random_button = gr.Button(
-                'Randomize Models', 
-                variant='secondary', 
-                scale=1
-            )
-
-    with gr.Column(scale=1):
-        with gr.Group():
-            with gr.Row():
-                output = [gr.Image(label=m, show_download_button=True, 
-                          elem_classes="output",
-                          interactive=False, width=112, height=112, 
-                          show_share_button=False, format="png",
-                          visible=True) for m in default_models]
-                current_models = [gr.Textbox(m, visible=False) 
-                                for m in default_models]
+        with gr.Column(scale=2):
+            gallery = gr.Gallery(label="Output", show_download_button=True, elem_classes="gallery",
+                                interactive=False, show_share_button=False, container=True, format="png",
+                                preview=True, object_fit="cover", columns=2, rows=2) 
 
-    with gr.Column(scale=2):
-        gallery = gr.Gallery(
-            label="Generated Images", 
-            show_download_button=True, 
-            elem_classes="gallery",
-            interactive=False, 
-            show_share_button=False, 
-            container=True, 
-            format="png",
-            preview=True, 
-            object_fit="cover", 
-            columns=2, 
-            rows=2
-        ) 
+        for m, o in zip(current_models, output):
+            gen_event = gr.on(triggers=[gen_button.click, txt_input.submit], fn=gen_fn,
+                              inputs=[m, txt_input, neg_input, height, width, steps, cfg, seed], outputs=[o],
+                              concurrency_limit=None, queue=False) # Be sure to delete ", queue=False" when activating the stop button
+            o.change(add_gallery, [o, m, gallery], [gallery])
+            #stop_button.click(lambda: gr.update(interactive=False), None, stop_button, cancels=[gen_event])
 
-    model_choice.change(update_imgbox, model_choice, output)
-    model_choice.change(extend_choices, model_choice, current_models)
-    random_button.click(random_choices, None, model_choice)
+        with gr.Column(scale=4):
+            with gr.Accordion('Model selection'):
+                model_choice = gr.CheckboxGroup(models, label = f'Choose up to {int(num_models)} different models from the {len(models)} available!', value=default_models, interactive=True)
+                model_choice.change(update_imgbox, model_choice, output)
+                model_choice.change(extend_choices, model_choice, current_models)
+                random_button.click(random_choices, None, model_choice)
 
-    for m, o in zip(current_models, output):
-        gen_event = gr.on(
-            triggers=[gen_button.click, txt_input.submit],
-            # 수정: 입력값을 실제 텍스트로 처리
-            fn=lambda txt, neg, h, w, s, c, seed, m=m: gen_fn(
-                m, 
-                str(txt) if txt is not None else "", 
-                str(neg) if neg is not None else "", 
-                h, w, s, c, seed
-            ) if m != 'NA' else None,
-            inputs=[txt_input, neg_input, height, width, steps, cfg, seed],
-            outputs=[o],
-            concurrency_limit=None,
-            queue=False
-        )
-        o.change(
-            fn=lambda img, g, m=m: add_gallery(img, m, g) if m != 'NA' else g,
-            inputs=[o, gallery],
-            outputs=[gallery]
-        )
+    with gr.Tab('Single model'):
+        with gr.Column(scale=2):
+            model_choice2 = gr.Dropdown(models, label='Choose model', value=models[0])
+            with gr.Group():
+                # global preSetPrompt
+                # global negPreSetPrompt
+                txt_input2 = gr.Textbox(label='Your prompt:', value = preSetPrompt, lines=3, autofocus=1)
+                neg_input2 = gr.Textbox(label='Negative prompt:', value=negPreSetPrompt, lines=1)
+                with gr.Accordion("Advanced", open=False, visible=True):
+                    with gr.Row():
+                        width2 = gr.Slider(label="Width", info="If 0, the default value is used.", maximum=1216, step=32, value=0)
+                        height2 = gr.Slider(label="Height", info="If 0, the default value is used.", maximum=1216, step=32, value=0)
+                    with gr.Row():
+                        steps2 = gr.Slider(label="Number of inference steps", info="If 0, the default value is used.", maximum=100, step=1, value=0)
+                        cfg2 = gr.Slider(label="Guidance scale", info="If 0, the default value is used.", maximum=30.0, step=0.1, value=0)
+                        seed2 = gr.Slider(label="Seed", info="Randomize Seed if -1.", minimum=-1, maximum=MAX_SEED, step=1, value=-1)
+                        seed_rand2 = gr.Button("Randomize Seed", size="sm", variant="secondary")
+                        seed_rand2.click(randomize_seed, None, [seed2], queue=False)
+            num_images = gr.Slider(1, max_images, value=max_images, step=1, label='Number of images')
+            with gr.Row():
+                gen_button2 = gr.Button('Let the machine halucinate', variant='primary', scale=2)
+                #stop_button2 = gr.Button('Stop', variant='stop', interactive=False, scale=1)
+                #gen_button2.click(lambda: gr.update(interactive=True), None, stop_button2)
 
+        with gr.Column(scale=1):
+            with gr.Group():
+                with gr.Row():
+                    output2 = [gr.Image(label='', show_download_button=True, elem_classes="output",
+                               interactive=False, width=112, height=112, visible=True, format="png",
+                               show_share_button=False, show_label=False) for _ in range(max_images)]
 
+        with gr.Column(scale=2):
+            gallery2 = gr.Gallery(label="Output", show_download_button=True, elem_classes="gallery",
+                                interactive=False, show_share_button=True, container=True, format="png",
+                                preview=True, object_fit="cover", columns=2, rows=2) 
 
+        for i, o in enumerate(output2):
+            img_i = gr.Number(i, visible=False)
+            num_images.change(lambda i, n: gr.update(visible = (i < n)), [img_i, num_images], o, queue=False)
+            gen_event2 = gr.on(triggers=[gen_button2.click, txt_input2.submit],
+                               fn=lambda i, n, m, t1, t2, n1, n2, n3, n4, n5: gen_fn(m, t1, t2, n1, n2, n3, n4, n5) if (i < n) else None,
+                               inputs=[img_i, num_images, model_choice2, txt_input2, neg_input2,
+                                       height2, width2, steps2, cfg2, seed2], outputs=[o],
+                                       concurrency_limit=None, queue=False)  # Be sure to delete ", queue=False" when activating the stop button
+            o.change(add_gallery, [o, model_choice2, gallery2], [gallery2])
+            #stop_button2.click(lambda: gr.update(interactive=False), None, stop_button2, cancels=[gen_event2])
+    
+    # gr.Markdown(js_AutoSave)
+    gr.Markdown("")
 
-demo.launch(show_api=False, max_threads=400)
+# demo.queue(default_concurrency_limit=200, max_size=200)
+demo.launch(show_api=False, max_threads=400)
+# demo.launch(show_api=False, max_threads=400, js=js_AutoSave)