fix some bugs

app.py

@@ -246,21 +246,29 @@ 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(
                                 margin=False,
                                 rounded=(True, True, True, True),
                             )
 
-                    go_to_img2img_btn = gr.Button("转至图生图(Go to img2img)").style(
-                                margin=False,
-                                rounded=(True, True, True, True),
-                            )
+                    # 并没有什么软用，转跳还是很卡，而且只能转跳一次
+                    # go_to_img2img_btn = gr.Button("转至图生图(Go to img2img)").style(
+                    #             margin=False,
+                    #             rounded=(True, True, True, True),
+                    #         )
 
 
                     gr.Examples(examples=examples, fn=request_images, inputs=text, outputs=gallery, examples_per_page=100)
                     text.submit(request_images, inputs=[text, class_draw, style_draw, sample_size, sr_option], outputs=gallery)
                     btn.click(request_images, inputs=[text, class_draw, style_draw, sample_size, sr_option], outputs=gallery)
 
+                    sample_size.change(
+                        fun=uifn.change_img_choices,
+                        inputs=[sample_size],
+                        outputs=[img_choices]
+                    )
+
             with gr.TabItem("图生图(Img-to-Img)", id="img2img_tab"):
                 with gr.Row(elem_id="prompt_row"):
                     img2img_prompt = gr.Textbox(label="Prompt",
@@ -428,13 +436,14 @@ if __name__ == "__main__":
                 output_txt2img_copy_to_input_btn.click(
                     uifn.copy_img_to_input,
                     [gallery],
-                    [img2img_image_editor, img2img_image_mask],
-                )
-                go_to_img2img_btn.click(
-                    uifn.go_to_img2img,
-                    [],
-                    [tabs],
+                    [tabs, img2img_image_editor, img2img_image_mask],
                 )
+
+                # go_to_img2img_btn.click(
+                #     uifn.go_to_img2img,
+                #     [],
+                #     [tabs],
+                # )
                 
                 # 下面这几个函数现在都没什么用
                 output_img2img_copy_to_input_btn.click(

ui_functions.py

@@ -5,6 +5,13 @@ from io import BytesIO
 import base64
 import re
 
+def change_img_choices(sample_size):
+    choices = ['图片1(img1)']
+    for i in range(sample_size):
+        choices.append(
+            '图片{}(img_{})'.format(i+1,i+1)
+        )
+    return choices
 
 def change_image_editor_mode(choice, cropped_image, masked_image, resize_mode, width, height):
     if choice == "Mask":
@@ -85,12 +92,9 @@ def copy_img_to_input(img):
         processed_image = Image.open(BytesIO(base64.b64decode(image_data)))
         tab_update = gr.update(selected='img2img_tab')
         img_update = gr.update(value=processed_image)
-        return processed_image, processed_image
+        return tab_update,processed_image, processed_image
     except IndexError:
         return [None, None]
-def go_to_img2img():
-    tab_update = gr.update(selected='img2img_tab')
-    return tab_update
 
 def copy_img_to_edit(img):
     try: