Update app.py

app.py

@@ -14,6 +14,8 @@ cross_sell_data = read_pickle_files("fynd.cross_sell_recommendations-00000000000
 upsell_data = read_pickle_files("fynd.up_sell_recommendations_000000000000000000000002s.pkl")
 uid_name_pairs = read_pickle_files("uid_name_pairs.pkl")
 uid_image_html_pairs = read_pickle_files("uid_image_html_pairs.pkl")
+item_costs_sephora_data = read_pickle_files("wavg_item_costs_sephora.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()}
@@ -61,13 +63,18 @@ if selected_product_id:
     if selected_product_id in uid_image_html_pairs:
         image_url = uid_image_html_pairs[selected_product_id]
         st.image(image_url, use_column_width=False, width=450)  # Set width to make image smaller
+        st.write("Cost of chosen item:", str(round(item_costs_sephora_data[selected_product_id], 2)))
+
 
 # Display recommendations for the selected product
 if selected_product_id:
     recommendations = get_recommendations(selected_product_id, recommendations_data)
     reccomendation_names = []
     reccomendation_images = []
-    
+    reccomendation_costs = []
+    # reccomendation_ids = [recommendations.get("product_id","item missing") for item in recommendations]
+    # item_costs_sephora = [item_costs_sephora_data.get(item, "cost missing") for item in reccomendation_ids]
+
     if recommendations:
         #st.subheader(f"Recommendations:")
         if len(recommendations)>10:
@@ -78,6 +85,7 @@ if selected_product_id:
         for recommendation in recommendations:
             product_name = recommendation.get('product_name')
             recommended_product_id = recommendation.get('product_id')
+            recommended_product_cost = item_costs_sephora_data.get(recommended_product_id, "item missing")
 
             # Display the image of each recommended product using the image URL
             if recommended_product_id in uid_image_html_pairs:
@@ -86,6 +94,7 @@ if selected_product_id:
 
             reccomendation_names.append(product_name)
             reccomendation_images.append(recommended_image_url)
+            reccomendation_costs.append(recommended_product_cost)
 
             # Display the product name
             #st.write(f"Product Name: {product_name}")
@@ -99,7 +108,7 @@ if selected_product_id:
         mid_section += f"""<div class="item">
                             <div id="image-container"><img src='{reccomendation_images[index]}' /></div>
                             <p style="font-size: 16px; font-weight: bold; white-space: normal; word-wrap: break-word;">{str(reccomendation_names[index])}</p>
+                            <p>{round(reccomendation_costs[index],2)}</p>
                         </div>"""
     mid_html = html2 + mid_section + """</div></div></body>"""
-    st.markdown(mid_html, unsafe_allow_html=True)
-
+    st.markdown(mid_html, unsafe_allow_html=True)