Update app.py

app.py

@@ -29,9 +29,8 @@ dataset_choice = st.selectbox(
 if dataset_choice == "Sephora Order Complete Dataset":
     list_of_products_to_display = {}
     for itemid in (list(all_products_with_names_sephora.keys())): #for items in the total list of items in all transactions
-        if itemid[0:3]!="600":
-            if itemid in (list(corrected_fp_growth_results_sephora.keys())): #if the item has a result from fp-growth
-                list_of_products_to_display[itemid]= all_products_with_names_sephora[itemid] #show it in the drop down menu
+        if itemid in (list(corrected_fp_growth_results_sephora.keys())): #if the item has a result from fp-growth
+            list_of_products_to_display[itemid]= all_products_with_names_sephora[itemid] #show it in the drop down menu
     
 
     name_to_id = {name: product_id for product_id, name in list_of_products_to_display.items()}    # Reverse the dictionary to map product names to IDs
@@ -48,8 +47,9 @@ if dataset_choice == "Sephora Order Complete Dataset":
 if dataset_choice == "Reliance Digital Order Complete Dataset": 
     list_of_products_to_display = {}
     for itemid in (list(all_products_with_names.keys())): #for items in the total list of items in all transactions
-        if itemid in (list(corrected_fp_growth_results.keys())): #if the item has a result from fp-growth
-            list_of_products_to_display[itemid]= all_products_with_names[itemid] #show it in the drop down menu
+        if itemid[0:3]!="600":
+            if itemid in (list(corrected_fp_growth_results.keys())): #if the item has a result from fp-growth
+                list_of_products_to_display[itemid]= all_products_with_names[itemid] #show it in the drop down menu
 
     name_to_id = {name: product_id for product_id, name in list_of_products_to_display.items()}    # Reverse the dictionary to map product names to IDs
     selected_product_name = st.selectbox('Select A Product:', list(name_to_id.keys()))    # Create a dropdown menu with the product names