Update app.py

app.py

@@ -627,10 +627,10 @@ def updateItemDisplay(seshid, budget=False):
                 scores = jsondata['data']
                 section_scores = [scores[sum(item_lens_ln[:i]):sum(item_lens_ln[:i])+ln] for i,ln in enumerate(item_lens_ln)]
                 section_scores_amax = [np.argmax(s) if s else -1 for s in section_scores]
-                best_idxs = [item_idxs[i][idx] if idx > -1 else -1 for i,idx in enumerate(section_scores_amax)]
+                best_idxs = [item_idxs[i][1][idx] if idx > -1 else -1 for i,idx in enumerate(section_scores_amax)]
                 end = time.time()
                 
-                print('updateDisplay -- item images downloaded: ', str(end-start), section_scores, item_idxs)
+                print('updateDisplay -- item images downloaded: ', str(end-start), best_idxs)
                 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}