home botton and recursive similarity search

app.py

@@ -43,26 +43,26 @@ def load_data_hf():
     return df, embeddings
 
 
-from huggingface_hub import hf_hub_download
-def load_data1():
-# Login using e.g. `huggingface-cli login` to access this dataset
-    path = hf_hub_download(
-        repo_id="traopia/fashion_show_data_all_embeddings",
-        filename="fashion_show_data_all_embeddings.json"
-    )
-    df = pd.read_json(path, lines = True)
-
-    #df = pd.read_json("hf://datasets/traopia/fashion_show_data_all_embeddings.json/fashion_show_data_all_embeddings.json", lines=True)
-    df["fashion_clip_image"] = df["fashion_clip_image"].apply(lambda x: x[0] if isinstance(x, list) else x)
-    df["image_urls"] = df["image_urls"].apply(lambda x: x[0] if x is not None else None)
-    df = df.rename(columns={"fashion_house":"designer", "image_urls":"url", "URL":"collection"})
+# from huggingface_hub import hf_hub_download
+# def load_data1():
+# # Login using e.g. `huggingface-cli login` to access this dataset
+#     path = hf_hub_download(
+#         repo_id="traopia/fashion_show_data_all_embeddings",
+#         filename="fashion_show_data_all_embeddings.json"
+#     )
+#     df = pd.read_json(path, lines = True)
+
+#     #df = pd.read_json("hf://datasets/traopia/fashion_show_data_all_embeddings.json/fashion_show_data_all_embeddings.json", lines=True)
+#     df["fashion_clip_image"] = df["fashion_clip_image"].apply(lambda x: x[0] if isinstance(x, list) else x)
+#     df["image_urls"] = df["image_urls"].apply(lambda x: x[0] if x is not None else None)
+#     df = df.rename(columns={"fashion_house":"designer", "image_urls":"url", "URL":"collection"})
     
-    df = df.dropna(subset="fashion_clip_image")
-    df = df.reset_index(drop=True)
-    df["key"] = df.index
-    embeddings = np.vstack(df["fashion_clip_image"].values)
+#     df = df.dropna(subset="fashion_clip_image")
+#     df = df.reset_index(drop=True)
+#     df["key"] = df.index
+#     embeddings = np.vstack(df["fashion_clip_image"].values)
     
-    return df, embeddings
+#     return df, embeddings
 
 df, embeddings = load_data_hf()
 
@@ -99,10 +99,14 @@ def show_metadata(idx, metadata):
     return out
 
 def find_similar(idx, metadata):
+    if not isinstance(idx, int) or idx >= len(metadata) or idx < 0:
+        return []  # or gr.update(visible=False)
     key = metadata[idx]["key"]
     similar_df = get_similar_images(df, key, embeddings, top_k=5)
     return similar_df["url"].tolist(), similar_df.to_dict(orient="records")
 
+
+
 # Gradio UI
 with gr.Blocks() as demo:
     gr.Markdown("# 👗 FashionDB Explorer")
@@ -124,6 +128,7 @@ with gr.Blocks() as demo:
     
     result_gallery = gr.Gallery(label="Search Results", columns=5, height="auto")
     metadata_output = gr.Markdown()
+    reference_image = gr.Image(label="Reference Image", interactive=False)
     similar_gallery = gr.Gallery(label="Similar Images", columns = 5, height="auto")
 
     metadata_state = gr.State([])
@@ -139,25 +144,38 @@ with gr.Blocks() as demo:
         outputs=[result_gallery, metadata_state, metadata_output, similar_gallery]
     )
 
+    # def handle_click(evt: gr.SelectData, metadata):
+    #     idx = evt.index
+    #     md = show_metadata(idx, metadata)
+    #     return idx, md
     def handle_click(evt: gr.SelectData, metadata):
         idx = evt.index
         md = show_metadata(idx, metadata)
-        return idx, md
+        img_path = metadata[idx]["url"]
+        return idx, md, img_path
+
+    # result_gallery.select(
+    #     handle_click,
+    #     inputs=[metadata_state],
+    #     outputs=[selected_idx, metadata_output]
+    # )
 
     result_gallery.select(
         handle_click,
         inputs=[metadata_state],
-        outputs=[selected_idx, metadata_output]
+        outputs=[selected_idx, metadata_output, reference_image]
     )
 
     # def show_similar(idx, metadata):
     #     return find_similar(int(idx), metadata)
     def show_similar(idx, metadata):
-        similar_images, similar_metadata = find_similar(int(idx), metadata)
-        return similar_images, similar_metadata
+        if idx is None or not str(idx).isdigit():
+            return [],[]  # safe fallback
+        return find_similar(int(idx), metadata)
 
     similar_metadata_state = gr.State()
     similar_metadata_output = gr.Markdown()
+    
     show_similar_button = gr.Button("Show Similar Images")
     show_similar_button.click(
         show_similar,
@@ -165,15 +183,36 @@ with gr.Blocks() as demo:
         outputs=[similar_gallery, similar_metadata_state]
     )
 
+    # def handle_similar_click(evt: gr.SelectData, metadata):
+    #     idx = evt.index
+    #     md = show_metadata(idx, metadata)
+    #     return idx, md
+
     def handle_similar_click(evt: gr.SelectData, metadata):
         idx = evt.index
         md = show_metadata(idx, metadata)
-        return idx, md
+        img_path = metadata[idx]["url"]
+        return idx, md, img_path
     
+    # similar_gallery.select(
+    # handle_similar_click,
+    # inputs=[similar_metadata_state],
+    # outputs=[selected_idx, similar_metadata_output]
+    # )
     similar_gallery.select(
     handle_similar_click,
     inputs=[similar_metadata_state],
-    outputs=[selected_idx, similar_metadata_output]
+    outputs=[selected_idx, similar_metadata_output, reference_image]
+    )
+
+    back_button = gr.Button("Back to Home")
+
+    def back_to_home():
+        return [], "", None  # clear similar_gallery, metadata_output, reference image
+
+    back_button.click(
+        back_to_home,
+        outputs=[similar_gallery, similar_metadata_output, reference_image]
     )
 
 demo.launch()