Update app.py

app.py

@@ -59,10 +59,12 @@ def similarity_search(query_image, num_images=5, *_, feature_extractor=feature_e
     closest_class = metadata.class_name.values[0]
 
     # Similar Images
-    similar_images = [
-        load_image(os.path.join(database_path, class_name, file_name))
+    similar_images_paths = [
+        os.path.join(database_path, class_name, file_name)
         for class_name, file_name in zip(metadata.class_name.values, metadata.file_name.values)
     ]
+    similar_images = [load_image(img) for img in similar_images_paths]
+    
     image_gallery = gr.Gallery(
         value=similar_images,
         label='Similar Images',
@@ -70,7 +72,7 @@ def similarity_search(query_image, num_images=5, *_, feature_extractor=feature_e
         preview=True,
         visible=True,
     )
-    return closest_class, image_gallery
+    return closest_class, image_gallery, similar_images
 
 
 # Gradio Application
@@ -91,7 +93,10 @@ with gr.Blocks(theme='soft') as app:
 
         # Output Gallery Display
         output_gallery = gr.Gallery(visible=False)
-
+        
+    # Hidden output for similar images paths
+    similar_paths_output = gr.Textbox(visible=False)
+    
     with gr.Row(equal_height=True):
 
         # Predicted Class
@@ -121,14 +126,14 @@ with gr.Blocks(theme='soft') as app:
     query_image.change(
         fn=similarity_search,
         inputs=[query_image, n_images],
-        outputs=[pred_class, output_gallery]
+        outputs=[pred_class, output_gallery, similar_paths_output]
     )
 
     # Search - On Click
     search_btn.click(
         fn=similarity_search,
         inputs=[query_image, n_images],
-        outputs=[pred_class, output_gallery]
+        outputs=[pred_class, output_gallery, similar_paths_output]
     )
 
 if __name__ == '__main__':