Update app.py

app.py

@@ -228,12 +228,11 @@ with gr.Blocks(css=css) as sdxl:
         """
     )
 
-    with gr.Row():
-        with gr.Column(scale=1):
-            text_prompt = gr.Textbox(label="Prompt", placeholder="a cute cat", lines=1, elem_id="prompt-text-input")
-            negative_prompt = gr.Textbox(label="Negative Prompt", value="text, blurry, fuzziness", lines=1, elem_id="negative-prompt-text-input")
-            image_style = gr.Dropdown(label="Style", choices=["None style", "Cinematic", "Digital Art", "Portrait"], value="None style", allow_custom_value=False)
-            text_button = gr.Button("Generate", variant='primary', elem_id="gen-button")
+    with gr.Row(elem_id="prompt-container"):
+        text_prompt = gr.Textbox(label="Prompt", placeholder="a cute cat", lines=1, elem_id="prompt-text-input")
+        negative_prompt = gr.Textbox(label="Negative Prompt", value="text, blurry, fuzziness", lines=1, elem_id="negative-prompt-text-input")
+        image_style = gr.Dropdown(label="Style", choices=["None style", "Cinematic", "Digital Art", "Portrait"], value="None style", allow_custom_value=False)
+        text_button = gr.Button("Generate", variant='primary', elem_id="gen-button")
     with gr.Row():
         with gr.Column(scale=1):
             image_output = gr.Image(type="pil", label="Output Image", elem_id="gallery")