Update app.py

app.py

@@ -58,22 +58,21 @@ def generate_images(
 # Create the Gradio blocks
 with gr.Blocks(css="style.css", theme='derekzen/stardust') as demo:
     with gr.Row(equal_height=False):
-        with gr.Column():
-            with gr.Group():
-                model_dropdown = gr.Dropdown(choices=list(models.keys()), value=model_list[0] if model_list else None, label="Model")
-                prompt = gr.Textbox(label="Prompt")
-                generate_btn = gr.Button("Generate Image")
-            with gr.Accordion("Advanced", open=False):
-                negative_prompt = gr.Textbox(label="Negative Prompt", value="lowres, (bad), text, error, fewer, extra, missing, worst quality, jpeg artifacts, low quality, watermark, unfinished, displeasing, oldest, early, chromatic aberration, signature, extra digits, artistic error, username, scan, [abstract]")
-                num_inference_steps = gr.Slider(minimum=10, maximum=50, step=1, value=25, label="Number of Inference Steps")
-                guidance_scale = gr.Slider(minimum=1, maximum=20, step=0.5, value=7.5, label="Guidance Scale")
-                height = gr.Slider(minimum=1024, maximum=2048, step=256, value=1024, label="Height")
-                width = gr.Slider(minimum=1024, maximum=2048, step=256, value=1024, label="Width")
-                num_images = gr.Slider(minimum=1, maximum=4, step=1, value=4, label="Number of Images")
-        with gr.Column():
-            output_gallery = gr.Gallery(label="Generated Images", height=480, scale=1)
+        with gr.Column(elem_id="input_column"):
+            with gr.Group(elem_id="input_group"):
+                model_dropdown = gr.Dropdown(choices=list(models.keys()), value=model_list[0] if model_list else None, label="Model", elem_id="model_dropdown")
+                prompt = gr.Textbox(label="Prompt", elem_id="prompt_textbox")
+                generate_btn = gr.Button("Generate Image", elem_id="generate_button")
+            with gr.Accordion("Advanced", open=False, elem_id="advanced_accordion"):
+                negative_prompt = gr.Textbox(label="Negative Prompt", value="lowres, (bad), text, error, fewer, extra, missing, worst quality, jpeg artifacts, low quality, watermark, unfinished, displeasing, oldest, early, chromatic aberration, signature, extra digits, artistic error, username, scan, [abstract]", elem_id="negative_prompt_textbox")
+                num_inference_steps = gr.Slider(minimum=10, maximum=50, step=1, value=25, label="Number of Inference Steps", elem_id="num_inference_steps_slider")
+                guidance_scale = gr.Slider(minimum=1, maximum=20, step=0.5, value=7.5, label="Guidance Scale", elem_id="guidance_scale_slider")
+                height = gr.Slider(minimum=1024, maximum=2048, step=256, value=1024, label="Height", elem_id="height_slider")
+                width = gr.Slider(minimum=1024, maximum=2048, step=256, value=1024, label="Width", elem_id="width_slider")
+                num_images = gr.Slider(minimum=1, maximum=4, step=1, value=4, label="Number of Images", elem_id="num_images_slider")
+        with gr.Column(elem_id="output_column"):
+            output_gallery = gr.Gallery(label="Generated Images", height=480, scale=1, elem_id="output_gallery")
         
         generate_btn.click(generate_images, inputs=[model_dropdown, prompt, negative_prompt, num_inference_steps, guidance_scale, height, width, num_images], outputs=output_gallery)
 
-demo.launch()
-            
+demo.launch()