made use the same settings checkbox UI prettier

app.py

@@ -84,16 +84,11 @@ def generate_single_image(
 @spaces.GPU(duration=80)
 def generate_arena_images(
     prompt,
-    negative_prompt_A,
-    negative_prompt_B,
-    num_inference_steps_A,
-    num_inference_steps_B,
-    height_A,
-    height_B,
-    width_A,
-    width_B,
-    guidance_scale_A,
-    guidance_scale_B,
+    negative_prompt,
+    num_inference_steps,
+    height,
+    width,
+    guidance_scale,
     seed,
     num_images_per_prompt,
     model_choice_A,
@@ -106,22 +101,14 @@ def generate_arena_images(
 
     generator = torch.Generator().manual_seed(seed)
 
-    # Apply settings based on use_same_settings
-    if use_same_settings:
-        num_inference_steps_B = num_inference_steps_A
-        height_B = height_A
-        width_B = width_A
-        guidance_scale_B = guidance_scale_A
-        negative_prompt_B = negative_prompt_A
-
     # Generate images for both models
     images_A = generate_single_image(
         prompt,
-        negative_prompt_A,
-        num_inference_steps_A,
-        height_A,
-        width_A,
-        guidance_scale_A,
+        negative_prompt,
+        num_inference_steps,
+        height,
+        width,
+        guidance_scale,
         seed,
         num_images_per_prompt,
         model_choice_A,
@@ -129,11 +116,11 @@ def generate_arena_images(
     )
     images_B = generate_single_image(
         prompt,
-        negative_prompt_B,
-        num_inference_steps_B,
-        height_B,
-        width_B,
-        guidance_scale_B,
+        negative_prompt,
+        num_inference_steps,
+        height,
+        width,
+        guidance_scale,
         seed,
         num_images_per_prompt,
         model_choice_B,
@@ -228,7 +215,69 @@ with gr.Blocks(css=css) as demo:
                 result_B = gr.Gallery(label="Generated Images (Model B)", elem_id="gallery_B")
             with gr.Accordion("Advanced options", open=False):
                 use_same_settings = gr.Checkbox(label='Use same settings for both models', value=True)
-                with gr.Row():
+
+                # Conditional UI elements based on use_same_settings
+                with gr.Row(visible=True):
+                    negative_prompt = gr.Textbox(
+                        label="Negative Prompt",
+                        info="Describe what you don't want in the image",
+                        value="deformed, distorted, disfigured, poorly drawn, bad anatomy, incorrect anatomy, extra limb, missing limb, floating limbs, mutated hands and fingers, disconnected limbs, mutation, mutated, ugly, disgusting, blurry, amputation",
+                        placeholder="Ugly, bad anatomy...",
+                    )
+                with gr.Row(visible=True):
+                    num_inference_steps = gr.Slider(
+                        label="Number of Inference Steps",
+                        info="The number of denoising steps of the image. More denoising steps usually lead to a higher quality image at the cost of slower inference",
+                        minimum=1,
+                        maximum=50,
+                        value=25,
+                        step=1,
+                    )
+                    guidance_scale = gr.Slider(
+                        label="Guidance Scale",
+                        info="Controls how much the image generation process follows the text prompt. Higher values make the image stick more closely to the input text.",
+                        minimum=0.0,
+                        maximum=10.0,
+                        value=7.5,
+                        step=0.1,
+                    )
+                with gr.Row(visible=True):
+                    width = gr.Slider(
+                        label="Width",
+                        info="Width of the Image",
+                        minimum=256,
+                        maximum=1344,
+                        step=32,
+                        value=1024,
+                    )
+                    height = gr.Slider(
+                        label="Height",
+                        info="Height of the Image",
+                        minimum=256,
+                        maximum=1344,
+                        step=32,
+                        value=1024,
+                    )
+                with gr.Row(visible=True):
+                    seed = gr.Slider(
+                        value=42,
+                        minimum=0,
+                        maximum=MAX_SEED,
+                        step=1,
+                        label="Seed",
+                        info="A starting point to initiate the generation process, put 0 for a random one",
+                    )
+                    num_images_per_prompt = gr.Slider(
+                        label="Images Per Prompt",
+                        info="Number of Images to generate with the settings",
+                        minimum=1,
+                        maximum=4,
+                        step=1,
+                        value=2,
+                    )
+
+                # Conditional UI elements based on use_same_settings
+                with gr.Row(visible=False):
                     negative_prompt_A = gr.Textbox(
                         label="Negative Prompt (Model A)",
                         info="Describe what you don't want in the image",
@@ -241,7 +290,7 @@ with gr.Blocks(css=css) as demo:
                         value="deformed, distorted, disfigured, poorly drawn, bad anatomy, incorrect anatomy, extra limb, missing limb, floating limbs, mutated hands and fingers, disconnected limbs, mutation, mutated, ugly, disgusting, blurry, amputation",
                         placeholder="Ugly, bad anatomy...",
                     )
-                with gr.Row():
+                with gr.Row(visible=False):
                     num_inference_steps_A = gr.Slider(
                         label="Number of Inference Steps (Model A)",
                         info="The number of denoising steps of the image. More denoising steps usually lead to a higher quality image at the cost of slower inference",
@@ -258,7 +307,7 @@ with gr.Blocks(css=css) as demo:
                         value=25,
                         step=1,
                     )
-                with gr.Row():
+                with gr.Row(visible=False):
                     width_A = gr.Slider(
                         label="Width (Model A)",
                         info="Width of the Image",
@@ -275,7 +324,7 @@ with gr.Blocks(css=css) as demo:
                         step=32,
                         value=1024,
                     )
-                with gr.Row():
+                with gr.Row(visible=False):
                     height_A = gr.Slider(
                         label="Height (Model A)",
                         info="Height of the Image",
@@ -292,7 +341,7 @@ with gr.Blocks(css=css) as demo:
                         step=32,
                         value=1024,
                     )
-                with gr.Row():
+                with gr.Row(visible=False):
                     guidance_scale_A = gr.Slider(
                         label="Guidance Scale (Model A)",
                         info="Controls how much the image generation process follows the text prompt. Higher values make the image stick more closely to the input text.",
@@ -309,17 +358,34 @@ with gr.Blocks(css=css) as demo:
                         value=7.5,
                         step=0.1,
                     )
-                with gr.Row():
-                    seed = gr.Slider(
+                with gr.Row(visible=False):
+                    seed_A = gr.Slider(
                         value=42,
                         minimum=0,
                         maximum=MAX_SEED,
                         step=1,
-                        label="Seed",
+                        label="Seed (Model A)",
                         info="A starting point to initiate the generation process, put 0 for a random one",
                     )
-                    num_images_per_prompt = gr.Slider(
-                        label="Images Per Prompt",
+                    seed_B = gr.Slider(
+                        value=42,
+                        minimum=0,
+                        maximum=MAX_SEED,
+                        step=1,
+                        label="Seed (Model B)",
+                        info="A starting point to initiate the generation process, put 0 for a random one",
+                    )
+                with gr.Row(visible=False):
+                    num_images_per_prompt_A = gr.Slider(
+                        label="Images Per Prompt (Model A)",
+                        info="Number of Images to generate with the settings",
+                        minimum=1,
+                        maximum=4,
+                        step=1,
+                        value=2,
+                    )
+                    num_images_per_prompt_B = gr.Slider(
+                        label="Images Per Prompt (Model B)",
                         info="Number of Images to generate with the settings",
                         minimum=1,
                         maximum=4,
@@ -342,21 +408,16 @@ with gr.Blocks(css=css) as demo:
                 fn=generate_arena_images,
                 inputs=[
                     prompt,
-                    negative_prompt_A,
-                    negative_prompt_B,
-                    num_inference_steps_A,
-                    num_inference_steps_B,
-                    height_A,
-                    height_B,
-                    width_A,
-                    width_B,
-                    guidance_scale_A,
-                    guidance_scale_B,
+                    negative_prompt,
+                    num_inference_steps,
+                    height,
+                    width,
+                    guidance_scale,
                     seed,
                     num_images_per_prompt,
                     model_choice_A,
                     model_choice_B,
-                    use_same_settings
+                    use_same_settings,
                 ],
                 outputs=[result_A, result_B],
             )