Added image processing

app.py

@@ -167,14 +167,16 @@ with gr.Blocks() as demo:
     with gr.Row():
         gr.Markdown("### Step 1: Upload or select a reference facial image for enrollment.")  
     with gr.Row():
+        image_input_enroll = gr.Image(type="pil", visible=False)
         with gr.Column():
-            image_input_enroll = gr.Image(label="Upload a reference facial image.", type="pil", sources="upload")
-            image_input_enroll.change(fn=crop_face_to_112x112, inputs=image_input_enroll, outputs=image_input_enroll)
+            image_upload_enroll = gr.Image(label="Upload a reference facial image.", type="pil", sources="upload")
+            image_upload_enroll.change(fn=crop_face_to_112x112, inputs=image_upload_enroll, outputs=image_input_enroll)
         with gr.Column():
             example_gallery = gr.Gallery(value=example_images, columns=3)
+            example_gallery.select(fn=get_selected_image, inputs=None, outputs=image_input_enroll)
         with gr.Column():
-            image_output_enroll = gr.Image(label="Reference facial image", sources="upload")
-            image_input_enroll.change(fn=display_image, inputs=image_input_enroll, outputs=image_output_enroll)
+            selectedImage = gr.Image(type="pil", label="Reference facial image", interactive=False)
+            image_input_enroll.change(fn=lambda img: img, inputs=image_input_enroll, outputs=selectedImage)
             
     with gr.Row():
         gr.Markdown("### Step 2: Generate reference embedding.") 
@@ -244,14 +246,17 @@ with gr.Blocks() as demo:
     with gr.Row():
         gr.Markdown("### Step 1: Upload or select a probe facial image for authentication.")  
     with gr.Row():
+        selectedImagePath_auth = gr.State()
+        image_input_auth = gr.Image(type="pil", visible=False)
         with gr.Column():
-            image_input_auth = gr.Image(label="Upload a facial image.", type="pil", sources="upload")
-            image_input_auth.change(fn=crop_face_to_112x112, inputs=image_input_auth, outputs=image_input_auth)
+            image_upload_auth = gr.Image(label="Upload a facial image.", type="pil", sources="upload")
+            image_upload_auth.change(fn=crop_face_to_112x112, inputs=image_upload_auth, outputs=image_input_auth)
         with gr.Column():
             example_gallery = gr.Gallery(value=example_images_auth, columns=3)
+            example_gallery.select(fn=get_selected_image_auth, inputs=None, outputs=image_input_auth)
         with gr.Column():
-            image_output_auth = gr.Image(label="Probe facial image", sources="upload")
-            image_input_auth.change(fn=display_image, inputs=image_input_auth, outputs=image_output_auth)
+            selectedImage = gr.Image(type="pil", label="Probe facial image", interactive=False)
+            image_input_auth.change(fn=lambda img: img, inputs=image_input_auth, outputs=selectedImage)
     
     with gr.Row():
         gr.Markdown("### Step 2: Generate probe facial embedding.")