Spaces:
Runtime error
Runtime error
File size: 3,060 Bytes
eb87c56 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 |
import pickle
import streamlit as st
from html_information import html
import random
random.seed(42)
def load_pickle_file(file_name):
with open(file_name, 'rb') as f:
return pickle.load(f)
def streamlit_carousel(header_name: str, rec_item_url: list,
rec_item_name: list) -> None:
st.header(header_name)
mid_section = ""
for index, value in enumerate(rec_item_url):
mid_section += """<div class="item"><div id="image-container"><img src='""" + str(value) + """' /></div><p>""" + str(rec_item_name[index]) + """</p></div>"""
mid_html = html + mid_section + """</div></div></body>"""
st.markdown(mid_html, unsafe_allow_html=True)
companies_list = ["JMD", "JMP", "FRESHPIK"]
st.set_page_config(page_title="My App", page_icon=":guardsman:", layout="wide", initial_sidebar_state="auto")
st.header("Similar Products, Bought Together Recommendations")
st.subheader("Choose a Company")
company_name = st.selectbox("Company List", companies_list)
if company_name == "JMD":
id_url = load_pickle_file("jmd_uid_image_url_map.pkl")
bt_document = load_pickle_file("jmd_bt_rec.pkl")
sim_document = load_pickle_file("jmd_sim_rec.pkl")
elif company_name == "JMP":
id_url = load_pickle_file("jmp_uid_image_url_map.pkl")
bt_document = load_pickle_file("jmp_bt_rec.pkl")
sim_document = load_pickle_file("jmp_sim_rec.pkl")
else:
id_url = load_pickle_file("freshpik_uid_image_url_map.pkl")
bt_document = load_pickle_file("freshpik_bt_rec.pkl")
sim_document = load_pickle_file("freshpik_sim_rec.pkl")
product_names = [item["product_name"] for item in bt_document]
st.subheader("Choose a Product")
index = st.selectbox("Product List", range(len(product_names)), format_func=lambda x: product_names[x])
query_uid = bt_document[index]["product_id"]
print(f"Query ID: {query_uid}")
print()
query_url = id_url[query_uid]
st.image(query_url, width=200)
#sim_recommendations = sim_document[index]["recommendations"]
for val in sim_document:
if val["product_id"] == query_uid:
sim_recommendations = val["recommendations"]
print("Similar Product Recommendations")
print(sim_recommendations)
print()
sim_recommendations_uid = [item["product_id"] for item in sim_recommendations]
sim_recommendations_name = [item["product_name"] for item in sim_recommendations]
sim_recommendations_url = [id_url[item] for item in sim_recommendations_uid]
streamlit_carousel("Similar Product Recommendations", sim_recommendations_url, sim_recommendations_name)
for val in bt_document:
if val["product_id"] == query_uid:
bt_recommendations = val["recommendations"]
print("Bought Together Recommendations")
print(bt_recommendations)
print()
bt_recommendations_uid = [item["product_id"] for item in bt_recommendations]
bt_recommendations_name = [item["product_name"] for item in bt_recommendations]
bt_recommendations_url = [id_url[item] for item in bt_recommendations_uid]
streamlit_carousel("Bought Together Recommendations", bt_recommendations_url, bt_recommendations_name)
|