Update app.py

app.py

@@ -532,6 +532,10 @@ def updateItemDisplay(seshid, budget=False):
         return sesh.updated_display 
 
 def getMostUsedColorsRGB(img):
+    try:
+        img = np.array(Image.open(io.BytesIO(r)))
+    except:
+        return []
     imgq = img // 48 * 48
     clr_counts = {}
     for i in range(0, imgq.shape[0], 7):
@@ -611,7 +615,7 @@ def updateItemDisplay(seshid, budget=False):
                 urls = [sesh.item_data[sesh.current_img_signature][sesh.room_type][sesh.room_style][sesh.price][sesh.Palette_ids[sesh.current_img_signature]][pair[0]][idx]['img'] for i,pair in enumerate(item_idxs) if pair[1] for idx in pair[1]]
                 start = time.time()
                 responses = asyncio.run(main(urls))
-                bestclr_keys = [getMostUsedColorsRGB(np.array(Image.open(io.BytesIO(r)))) for r in responses]
+                bestclr_keys = [getMostUsedColorsRGB(r) for r in responses]
                 end = time.time()
                 print('updateDisplay -- item images downloaded: ', str(end-start), bestclr_keys)
                 sesh.display_items_ = {img:[(-1 if item in ['Palette', 'Living Room', 'Bedroom'] or not sesh.item_data[img][sesh.item_data_keys[img]['room_type']][sesh.item_data_keys[img]['room_style']][sesh.item_data_keys[img]['price']][sesh.item_data_keys[img]['Paletteid']][item] else (random.sample(list(range(len(sesh.item_data[img][sesh.item_data_keys[img]['room_type']][sesh.item_data_keys[img]['room_style']][sesh.item_data_keys[img]['price']][sesh.item_data_keys[img]['Paletteid']][item]))), 1)[0]), item) for itemi,item in enumerate(sesh.items_layouts[img])] for img in sesh.item_data_keys}