[feat] 上传图片区域也改为gallery

algorithm.py

@@ -91,17 +91,17 @@ class ImageSearch():
 
         return zip(product_cropped_images, product_names)
 
-    def upload_wait2search_image(self, image_files):
+    def upload_wait2search_image(self, image_infos):
 
         wait2search_image_list = []
         wait2search_image_hashes = []
         wait2search_image_names = []
 
-        for image_file in image_files:
-
+        for image_info in image_infos:
+            image_file, image_label = image_info
             wait2search_image = Image.open(image_file)
 
-            wait2search_image_names.append(image_file.name.split('/')[-1])
+            wait2search_image_names.append(image_file.split('/')[-1])
             wait2search_image_list.append(wait2search_image)
             wait2search_image_hashes.append(hashlib.md5(wait2search_image.tobytes()).hexdigest())
 

app.py

@@ -31,14 +31,14 @@ with gr.Blocks() as demo:
     upload_pdf_btn.click(SearchImageTask.upload_products_pdf_file, input_pdf, gallery_pdf)
     with gr.Row():
         with gr.Column(scale=1):
-            input_img = gr.Files(label="Upload Image", height=300)
+            gallery_input_image = gr.Gallery(label="Upload Images", elem_id="gallery", height=400)
         with gr.Column(scale=1):
-            gallery_image = gr.Gallery(label="Matched Images", elem_id="gallery", height=300)
+            gallery_output_image = gr.Gallery(label="Matched Images", elem_id="gallery", height=400)
 
     with gr.Row():
         upload_img_btn = gr.Button("上传测试图片，匹配作品登记证书")
 
-    upload_img_btn.click(SearchImageTask.upload_wait2search_image, input_img, gallery_image)
+    upload_img_btn.click(SearchImageTask.upload_wait2search_image, gallery_input_image, gallery_output_image)
 
 
 if __name__ == "__main__":