add img choice

app.py

@@ -246,8 +246,9 @@ if __name__ == "__main__":
                         label="Generated images", show_label=False, elem_id="gallery"
                     ).style(grid=[2], height="auto")
 
-                    img_choices = gr.Dropdown(["图片1(img1)"],label='请选择一张图片发送到图生图',show_label=True,value="图片1(img1)")
-                    output_txt2img_copy_to_input_btn = gr.Button("发送图片到图生图(Sent the image to img2img)").style(
+                    with gr.Row().style(mobile_collapse=False, equal_height=True):
+                        img_choices = gr.Dropdown(["图片1(img1)"],label='请选择一张图片发送到图生图',show_label=True,value="图片1(img1)", type="index")
+                        output_txt2img_copy_to_input_btn = gr.Button("发送图片到图生图(Sent the image to img2img)").style(
                                 margin=False,
                                 rounded=(True, True, True, True),
                             )
@@ -435,7 +436,7 @@ if __name__ == "__main__":
                 # )
                 output_txt2img_copy_to_input_btn.click(
                     uifn.copy_img_to_input,
-                    [gallery],
+                    [gallery, img_choices],
                     [tabs, img2img_image_editor, img2img_image_mask],
                 )
 

ui_functions.py

@@ -83,12 +83,12 @@ def copy_img_params_to_lab(params):
         return prompt,seed,steps,cfg_scale,sampler
     except IndexError:
         return [None, None]
-def copy_img_to_input(img):
+def copy_img_to_input(img, idx):
     try:
         # print(img)
         # print("=============")
         # print("The img type is:{}".format(type(img[0])))
-        image_data = re.sub('^data:image/.+;base64,', '', img[0])
+        image_data = re.sub('^data:image/.+;base64,', '', img[idx])
         processed_image = Image.open(BytesIO(base64.b64decode(image_data)))
         tab_update = gr.update(selected='img2img_tab')
         img_update = gr.update(value=processed_image)