search similar images with image search

app.py

@@ -216,6 +216,46 @@ with gr.Blocks() as demo:
                 outputs=[uploaded_selected_idx, uploaded_metadata_output, uploaded_reference_image]
             )
 
+                        # SIMILAR IMAGE SEARCH FOR IMAGE TAB
+            uploaded_similar_gallery = gr.Gallery(label="Similar Images", columns=5, height="auto")
+            uploaded_similar_metadata_state = gr.State([])
+            uploaded_similar_metadata_output = gr.Markdown()
+
+            uploaded_show_similar_button = gr.Button("Show Similar Images")
+
+            def show_similar_uploaded(idx, metadata):
+                if idx is None or not str(idx).isdigit():
+                    return [], []
+                return find_similar(int(idx), metadata)
+
+            uploaded_show_similar_button.click(
+                show_similar_uploaded,
+                inputs=[uploaded_selected_idx, uploaded_metadata_state],
+                outputs=[uploaded_similar_gallery, uploaded_similar_metadata_state]
+            )
+
+            def handle_uploaded_similar_click(evt: gr.SelectData, metadata):
+                idx = evt.index
+                md = show_metadata(idx, metadata)
+                img_path = metadata[idx]["url"]
+                return idx, md, img_path
+
+            uploaded_similar_gallery.select(
+                handle_uploaded_similar_click,
+                inputs=[uploaded_similar_metadata_state],
+                outputs=[uploaded_selected_idx, uploaded_similar_metadata_output, uploaded_reference_image]
+            )
+
+            uploaded_back_button = gr.Button("Back to Initial Uploaded Search")
+
+            def back_to_uploaded_home():
+                return [], "", None
+
+            uploaded_back_button.click(
+                back_to_uploaded_home,
+                outputs=[uploaded_similar_gallery, uploaded_similar_metadata_output, uploaded_reference_image]
+            )
+
     back_button = gr.Button("Back to Home")
 
     def back_to_home():

search.py

@@ -5,6 +5,8 @@ import numpy as np
 # Use a compatible CLIP model
 model = SentenceTransformer("clip-ViT-B-32")
 
+
+
 def search_images_by_text(text, df, embeddings, top_k=30):
     text_emb = model.encode([text])
     filtered_embeddings = embeddings[df.index]