Update app.py

app.py

@@ -110,6 +110,21 @@ css = """
         #component-16{border-top-width: 1px!important;margin-top: 1em}
         .image_duplication{position: absolute; width: 100px; left: 50px}
         .tabitem{border: 0 !important}.style(mobile_collapse=False, equal_height=True).style(mobile_collapse=False, equal_height=True).style(mobile_collapse=False, equal_height=True).style(mobile_collapse=False, equal_height=True
+        #gen-button{
+        border-top-left-radius:0;
+        border-bottom-left-radius:0;
+        }
+        #gallery {
+            min-height: 22rem;
+            margin-bottom: 15px;
+            margin-left: auto;
+            margin-right: auto;
+            border-bottom-right-radius: .5rem !important;
+            border-bottom-left-radius: .5rem !important;
+        }
+        #gallery>div>.h-full {
+            min-height: 20rem;
+        }
 """
 
 with gr.Blocks(css=css) as demo:
@@ -164,14 +179,14 @@ with gr.Blocks(css=css) as demo:
     with gr.Row(elem_id="prompt-container").style(mobile_collapse=False, equal_height=True):    
         with gr.Column(scale=6, min_width=600):  
             prompt = gr.Textbox(label="Prompt", placeholder="a cute cat, 8k", show_label=True, lines=1, elem_id="prompt-text-input")
-            text_button = gr.Button("Generate", variant='primary', elem_id="generate")
+            text_button = gr.Button("Generate", variant='primary', elem_id="gen-button")
     with gr.Row():    
         with gr.Column(scale=1):
-            image_output = gr.Image()
+            image_output = gr.Image(elem_id="gallery")
     with gr.Row():
         with gr.Accordion("Additionals inputs", open=False):
             with gr.Column(scale=1):
-                negative_prompt = gr.Textbox(label="Negative Prompt", value="text, blurry", placeholder="What you don't want to see in the image", show_label=True, lines=1)
+                negative_prompt = gr.Textbox(label="Negative Prompt", value="text, blurry", placeholder="What you don't want to see in the image", show_label=True, lines=1, elem_id="negative-prompt-text-input")
             with gr.Column(scale=1):
                 steps = gr.Slider(label="Sampling Steps", minimum=1, maximum=30, value=25, step=1)
                 cfg_scale = gr.Slider(label="CFG Scale", minimum=1, maximum=20, value=7, step=1)