Adjust the layout

app.py

@@ -268,7 +268,7 @@ if __name__ == "__main__":
 
                     gallery = gr.Gallery(
                         label="Generated images", show_label=False, elem_id="gallery"
-                    ).style(grid=[2], height="auto")
+                    ).style(grid=[2,2])
 
                     with gr.Row().style(mobile_collapse=False, equal_height=True):
                         img_choices = gr.Dropdown(["图片1(img1)"],label='请选择一张图片发送到图生图或者超分',show_label=True,value="图片1(img1)")
@@ -281,13 +281,17 @@ if __name__ == "__main__":
                                 margin=False,
                                 rounded=(True, True, True, True),
                             )
+
+                    sr_gallery = gr.Gallery(
+                        label="SR images", show_label=True, elem_id="sr_gallery"
+                    ).style(grid=[2,2])
+
+                    with gr.Row():
+                        prompt = gr.Markdown("提示(Prompt)：", visible=False)
                     with gr.Row():
                         move_prompt_zh = gr.Markdown("请移至图生图部分进行编辑(拉到顶部)", visible=False)
                     with gr.Row():
                         move_prompt_en = gr.Markdown("Please move to the img2img section for editing(Pull to the top)", visible=False)
-                    sr_gallery = gr.Gallery(
-                        label="SR images", show_label=True, elem_id="sr_gallery"
-                    ).style(grid=[2], height="auto")
 
                         
 
@@ -344,11 +348,7 @@ if __name__ == "__main__":
                             visible=False,
                             image_mode="RGBA",
                             elem_id="img2img_editor"
-                        )
-                        with gr.Row():
-                            gr.Markdown('请选择一张图像掩盖掉一部分区域,并输入文本描述')
-                        with gr.Row():
-                            gr.Markdown('Please select an image to cover up a part of the area and enter a text description.')
+                        ).style(grid=[4,4,4])
                         with gr.Tabs(visible=False):
                             with gr.TabItem("编辑设置"):
                                 with gr.Row():
@@ -405,7 +405,10 @@ if __name__ == "__main__":
                                             inputs=output_img2img_seed, outputs=[],
                                             _js=call_JS("gradioInputToClipboard"), fn=None, show_progress=False)
                                     output_img2img_stats = gr.HTML(label='Stats')
-
+                    with gr.Row():
+                        gr.Markdown('请选择一张图像掩盖掉一部分区域,并输入文本描述')
+                    with gr.Row():
+                        gr.Markdown('Please select an image to cover up a part of the area and enter a text description.')
                 gr.Markdown('# 编辑设置',visible=False)
 
                 with gr.Row(visible=False):
@@ -480,7 +483,7 @@ if __name__ == "__main__":
                 output_txt2img_copy_to_input_btn.click(
                     uifn.copy_img_to_input,
                     [gallery, img_choices],
-                    [tabs, img2img_image_editor, img2img_image_mask, move_prompt_zh, move_prompt_en],
+                    [tabs, img2img_image_editor, img2img_image_mask, move_prompt_zh, move_prompt_en, prompt],
                 )
 
                 # go_to_img2img_btn.click(

ui_functions.py

@@ -102,7 +102,8 @@ def copy_img_to_input(img, idx):
         img_update = gr.update(value=processed_image)
         move_prompt_zh_update = gr.update(visible=True)
         move_prompt_en_update = gr.update(visible=True)
-        return tab_update,processed_image, processed_image, move_prompt_zh_update, move_prompt_en_update
+        prompt_update = gr.update(visible=True)
+        return tab_update,processed_image, processed_image, move_prompt_zh_update, prompt_update, 
     except IndexError:
         return [None, None]