Update app.py

app.py

@@ -556,7 +556,7 @@ def isJSONSerialiazable(obj):
     except:
         return False
 
-def updateItemDisplay(seshid, budget=False):
+def updateItemDisplay(seshid, budget=False, pallete_change=False):
     if True:
         keys = ['filtered_items_room', 'current_img_signature', 'items', 'filtered_items_style', 'room_style', 'room_type', 'bedsize', 'price', 'filtered_items_colors', 'Palette_ids', 'item_data', 'item_data_', 'display_items', 'item_data_keys', 'Palettes_displayed', 'room_img_loaded', 'Paletteid', 'img_id', 'seshid']
         url = 'https://dreamdemo.pythonanywhere.com/sesh'
@@ -585,7 +585,7 @@ def updateItemDisplay(seshid, budget=False):
 
             sesh.product_means[sesh.current_img_signature] = {item:sum([float(pd['price'].replace('$', '').replace(',', '')) for pd in products])/len(products) if products else 0 for item,products in sesh.item_data[sesh.current_img_signature][sesh.room_type][sesh.room_style][sesh.price][sesh.Palette_ids[sesh.current_img_signature]].items() if item not in ['Palette', 'Living Room', 'Bedroom']}
 
-            if hasattr(sesh, 'budget_number') and sesh.budget_number:
+            if hasattr(sesh, 'budget_number') and sesh.budget_number and not pallete_change:
                 print('updateItemDisplay -- sesh.budget_number: ', sesh.budget_number)
                 #product_lens = {item:len(products) for item,products in sesh.item_data[sesh.current_img_signature][sesh.room_type][sesh.room_style][sesh.price][sesh.Palette_ids[sesh.current_img_signature]].items() if item not in ['Palette', 'Living Room', 'Bedroom']}
                 #print('updateItemDisplay -- product_lens: ', product_lens)
@@ -803,7 +803,7 @@ def updateColors(evt: gr.SelectData, seshid):
     sesh.Palette_imgs[sesh.img_id] = sesh.Palette_img
 
     if hasattr(sesh, 'room_img_loaded') and sesh.room_img_loaded:
-        display = updateItemDisplay(seshid)
+        display = updateItemDisplay(seshid, pallete_change=True)
         return display
     else:
         return sesh.updated_display