add img choice

ui_functions.py

@@ -92,8 +92,8 @@ def copy_img_to_input(img, idx):
         idx_map = {
             "图片1(img1)":0,
             "图片2(img2)":1,
-            "图片3(img4)":2,
-            "图片4(img5)":3,
+            "图片3(img3)":2,
+            "图片4(img4)":3,
         }
         idx = idx_map[idx]
         image_data = re.sub('^data:image/.+;base64,', '', img[idx])