updares app metadata similar images

app.py

@@ -5,6 +5,27 @@ from search import search_images_by_text, get_similar_images
 import requests
 from io import BytesIO
 
+def create_collection_url(row):
+    base_url = "https://www.vogue.com/fashion-shows/"
+    season = str(row["season"]).lower()
+    year = str(row["year"])
+    category = str(row["category"]).lower() if pd.notna(row["category"]) and row["category"] and str(row["category"]).lower() != "nan" else None
+    designer = str(row["designer"]).lower().replace(" ", "-")
+    
+    # Add city if available
+    city = str(row["city"]).lower().replace(" ", "-") if pd.notna(row["city"]) and row["city"] and str(row["city"]).lower() != "nan" else None
+    
+    if pd.isna(category) or category is None or category == "nan":
+        if city:
+            return f"{base_url}{city}-{season}-{year}/{designer}"
+        else:
+            return f"{base_url}{season}-{year}/{designer}"
+    else:
+        if city:
+            return f"{base_url}{city}-{season}-{year}-{category}/{designer}"
+        else:
+            return f"{base_url}{season}-{year}-{category}/{designer}"
+
 import requests
 from io import BytesIO
 #@st.cache_data(show_spinner="Loading FashionDB...")
@@ -18,7 +39,7 @@ def load_data_hf():
     response = requests.get(npy_url)
     response.raise_for_status()  # Raise error if download fails
     embeddings = np.load(BytesIO(response.content))
-
+    df['collection'] = df.apply(create_collection_url, axis=1)
     return df, embeddings
 
 
@@ -96,7 +117,7 @@ with gr.Blocks() as demo:
         max_year = int(df['year'].max())
 
         start_year = gr.Slider(label="Start Year", minimum=min_year, maximum=max_year, value=2000, step=1)
-        end_year = gr.Slider(label="End Year", minimum=min_year, maximum=max_year, value=2025, step=1)
+        end_year = gr.Slider(label="End Year", minimum=min_year, maximum=max_year, value=2024, step=1)
 
     query = gr.Textbox(label="Search", placeholder="e.g., pink dress")
     search_button = gr.Button("Search")
@@ -129,14 +150,30 @@ with gr.Blocks() as demo:
         outputs=[selected_idx, metadata_output]
     )
 
+    # def show_similar(idx, metadata):
+    #     return find_similar(int(idx), metadata)
     def show_similar(idx, metadata):
-        return find_similar(int(idx), metadata)
+        similar_images, similar_metadata = find_similar(int(idx), metadata)
+        return similar_images, similar_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,
         inputs=[selected_idx, metadata_state],
-        outputs=[similar_gallery, metadata_state]
+        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
+    
+    similar_gallery.select(
+    handle_similar_click,
+    inputs=[similar_metadata_state],
+    outputs=[selected_idx, similar_metadata_output]
     )
 
 demo.launch()