Update app.py

app.py

@@ -9,8 +9,8 @@ def read_pickle_files(pickle_file):
 # Load the necessary pickle files
 cross_sell_data = read_pickle_files("fynd.cross_sell_recommendations-000000000000000000000001s.pkl")
 upsell_data = read_pickle_files("fynd.up_sell_recommendations_000000000000000000000002s.pkl")
-catalog_data = read_pickle_files("clickstream_data_sephora_sephora_products_1724404940894(1).pkl")
 uid_name_pairs = read_pickle_files("uid_name_pairs.pkl")
+uid_image_html_pairs = read_pickle_files("uid_image_html_pairs.pkl")
 
 # Create a mapping from product_id to product name for dropdown
 product_name_to_id = {name: uid for name, uid in uid_name_pairs.items()}
@@ -18,16 +18,7 @@ product_name_to_id = {name: uid for name, uid in uid_name_pairs.items()}
 # Create a reverse mapping from product_id to product_name for display purposes
 product_id_to_name = {uid: name for name, uid in uid_name_pairs.items()}
 
-# Function to get product image URL from catalog
-def get_product_image_url(product_id, catalog_data):
-    for product in catalog_data:
-        if product['uid'] == product_id:
-            medias = product.get('medias', [])
-            if medias and 'url' in medias[0]:
-                return medias[0]['url']
-    return None
-
-# Extract product names from recommendation data
+# Function to extract product list from recommendation data
 def extract_product_list(recommendation_data):
     product_ids = [entry['product_id'] for entry in recommendation_data]
     # Map the product IDs to names for the dropdown
@@ -43,7 +34,7 @@ def get_recommendations(product_id, recommendation_data):
 # Streamlit App Layout
 st.title("Product Recommendations")
 
-# Dropdown for selecting type (cross-sell or up-sell)
+# Dropdown for selecting recommendation type
 recommendation_type = st.selectbox("Select recommendation type:", ["Cross-sell", "Up-sell"])
 
 # Choose the appropriate data based on recommendation type
@@ -61,6 +52,15 @@ selected_product_name = st.selectbox("Select a product:", product_list)
 # Get the selected product's ID using the name
 selected_product_id = product_name_to_id.get(selected_product_name)
 
+# Display the image of the selected product using the image URL
+if selected_product_id:
+    st.subheader(f"Selected Product: {selected_product_name}")
+    
+    # Check if the product's ID has an associated image HTML and use the image URL
+    if selected_product_id in uid_image_html_pairs:
+        image_url = uid_image_html_pairs[selected_product_id]
+        st.image(image_url, caption=selected_product_name, use_column_width=False, width=150)  # Set width to make image smaller
+
 # Display recommendations for the selected product
 if selected_product_id:
     recommendations = get_recommendations(selected_product_id, recommendations_data)
@@ -69,10 +69,14 @@ if selected_product_id:
         st.subheader(f"Recommendations for {selected_product_name}")
         for recommendation in recommendations:
             product_name = recommendation.get('product_name')
-            product_image_url = get_product_image_url(recommendation['product_id'], catalog_data)
+            recommended_product_id = recommendation.get('product_id')
+
+            # Display the image of each recommended product using the image URL
+            if recommended_product_id in uid_image_html_pairs:
+                recommended_image_url = uid_image_html_pairs[recommended_product_id]
+                st.image(recommended_image_url, caption=product_name, use_column_width=False, width=150)  # Set width to make images smaller
             
-            if product_image_url:
-                st.image(product_image_url, width=100)  # Display image
-            st.write(f"{product_name}")
+            # Display the product name
+            st.write(f"Product Name: {product_name}")
     else:
         st.write("No recommendations found for this product.")