Update app.py

app.py

@@ -77,10 +77,11 @@ def similarity_search(
     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]
     
     # return closest_class, similar_images
 
@@ -91,7 +92,7 @@ def similarity_search(
         preview=True,
         visible=True,
     )
-    return closest_class, image_gallery
+    return closest_class, image_gallery, similar_images_paths
 
 
 # Gradio Application
@@ -112,6 +113,9 @@ 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
@@ -141,14 +145,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__':