add img choice

app.py

@@ -247,7 +247,7 @@ if __name__ == "__main__":
                     ).style(grid=[2], height="auto")
 
                     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")
+                        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),

ui_functions.py

@@ -89,6 +89,12 @@ def copy_img_to_input(img, idx):
         # print(img)
         # print("=============")
         # print("The img type is:{}".format(type(img[0])))
+        idx_map = {
+            "图片1(img1)":0,
+            "图片2(img2)":1,
+            "图片3(img4)":2,
+            "图片4(img5)":3,
+        }
         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')