Update app.py

app.py

@@ -125,12 +125,7 @@ def add_gallery(image, model_str, gallery):
     return gallery
 
 
-CSS="""
-.gradio-container { max-width: 1200px; margin: 0 auto; !important; }
-.output { width=112px; height=112px; max_width=112px; max_height=112px; !important; }
-.gallery { min_width=512px; min_height=512px; max_height=512px; !important; }
-.guide { text-align: center; !important; }
-"""
+
 
 js_func = """
 function refresh() {
@@ -158,108 +153,108 @@ js_AutoSave="""
         
 """
 
+
+CSS="""
+.gradio-container { max-width: 1200px; margin: 0 auto; background: linear-gradient(to bottom, #1a1a1a, #2d2d2d); !important; }
+.output { 
+    width: 112px; 
+    height: 112px; 
+    border-radius: 10px;
+    box-shadow: 0 4px 8px rgba(0,0,0,0.2);
+    transition: transform 0.2s;
+    !important; 
+}
+.output:hover {
+    transform: scale(1.05);
+}
+.gallery { 
+    min-width: 512px;
+    min-height: 512px;
+    max-height: 512px;
+    border-radius: 15px;
+    box-shadow: 0 6px 12px rgba(0,0,0,0.3);
+    !important; 
+}
+.guide { text-align: center; color: #e0e0e0; !important; }
+.primary-btn {
+    background: linear-gradient(45deg, #4a90e2, #357abd);
+    border-radius: 8px;
+    transition: all 0.3s ease;
+}
+.primary-btn:hover {
+    transform: translateY(-2px);
+    box-shadow: 0 5px 15px rgba(74,144,226,0.3);
+}
+"""
+
 with gr.Blocks(theme='NoCrypt/miku@>=1.2.2', fill_width=True, css=CSS) as demo:
-# with gr.Blocks(theme='JohnSmith9982/small_and_pretty', fill_width=True, css=CSS, js=js_func) as demo:
-    gr.HTML("")  
-    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.Column(scale=1):
-            with gr.Group():
+    gr.HTML("""<a href="https://visitorbadge.io/status?path=https%3A%2F%2Fgunship999-SexyImages.hf.space">
+               <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.Group():
+            txt_input = gr.Textbox(
+                label='Your prompt:', 
+                value=preSetPrompt, 
+                lines=3, 
+                autofocus=1,
+                container=False,
+                elem_classes="modern-input"
+            )
+            neg_input = gr.Textbox(
+                label='Negative prompt:', 
+                value=negPreSetPrompt, 
+                lines=1
+            )
+            with gr.Accordion("Advanced Settings", open=False):
                 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) 
-
-        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])
-
-        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)
-
-    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():
+                    width = gr.Slider(label="Width", maximum=1216, step=32, value=0)
+                    height = gr.Slider(label="Height", maximum=1216, step=32, value=0)
                 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.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)
+                    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)
+
+        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="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
+        ) 
+
+demo.launch(show_api=False, max_threads=400)