Fixing multiselect to preserve user input

app.py

@@ -8,8 +8,8 @@ from contentBased_model import *
 if "user_preferences" not in st.session_state:
     st.session_state["user_preferences"] = {}
 
-if "user_titles" not in st.session_state:
-    st.session_state["user_titles"] = []
+# if "user_titles" not in st.session_state:
+#     st.session_state["user_titles"] = []
 
 data = pd.read_pickle("data/2k_titles.pkl")
 
@@ -74,16 +74,24 @@ def spr_sidebar():
 # Home page. One of the page in Main Header
 def home_page():
     st.title("Steam Recommendation System")
+    # st.session_state
 
-    # st.session_state['user_title'] = st.session_state['input_title']
-    preferences = st.multiselect(
-        label="Input games you like:",
-        options=list(data),
-        key="user_titles")
-
-    user_input = generate_app_gamebox(preferences)
+    if "user_preferences" in st.session_state:
+        ids_list = ids_to_titles(st.session_state["user_preferences"])
+        preferences = st.multiselect(
+            label="Input games you like:",
+            options=list(data),
+            default=ids_list,
+            key="user_titles")
+        user_input = generate_app_gamebox(preferences)
+    else:
+        preferences = st.multiselect(
+            label="Input games you like:",
+            options=list(data),
+            key="user_titles")
+        user_input = generate_app_gamebox(preferences)
 
-    state = st.button("Get state")
+    state = st.button("Get recommendation")
 
     if state:
         st.session_state["user_preferences"] = user_input
@@ -91,10 +99,6 @@ def home_page():
         st.markdown("---")
         games_recomm(st.session_state["user_preferences"])
 
-        # st.session_state
-
-
-
 # Result page
 # Show the list of predictions for active user
 def result_page():

easeBased_model.py

@@ -85,8 +85,8 @@ class EASE:
             output_preds.append([self.item_map[id] for id in top_items.indices.tolist()])            
             score_preds.append( top_items.values.tolist() )
         
-        print("'\nOutput preds:", output_preds)
-        print("Score preds:", score_preds)
+        # print("'\nOutput preds:", output_preds)
+        # print("Score preds:", score_preds)
         pred_df['predicted_items'] = output_preds
         pred_df['predicted_score'] = score_preds
         

func.py

@@ -8,12 +8,18 @@ from PIL import Image
 data_titles = pd.read_pickle("data/2k_titles.pkl")
 data_ids = pd.read_pickle("data/2k_ids.pkl")
 
+def ids_to_titles(ids_list):
+    return [data_ids[title] for title in ids_list]
+
+def titles_to_ids(titles_list):
+    return [data_titles[id] for id in titles_list]
+
 def generate_app_gamebox(titles):
-    titles_id = []
-    # selectboxes_id = {}
+    """
+        Placeholder
+    """
 
-    for title in titles:
-        titles_id.append(data_titles[title])
+    titles_id = titles_to_ids(titles)
 
     for id in titles_id:
         url = f"https://cdn.cloudflare.steamstatic.com/steam/apps/{id}/header.jpg"
@@ -33,10 +39,11 @@ def generate_app_gamebox(titles):
     return titles_id
 
 def generate_res_gamebox(ids):
-    # titles_list = []
+    """
+        Placeholder
+    """
 
     for id in ids:
-        # titles_list.append(data_ids[id])
         url_page = f"https://store.steampowered.com/app/{id}/"
         url_img = f"https://cdn.cloudflare.steamstatic.com/steam/apps/{id}/header.jpg"
         resp = requests.get(url_img)
@@ -48,6 +55,9 @@ def generate_res_gamebox(ids):
             
             st.divider()
 
+def combine_hybrid_result():
+    pass
+
     # for title in titles_list:
     #     # url = f"https://store.steampowered.com/app/"
     #     with st.container():