Update app.py

app.py

@@ -625,7 +625,9 @@ def updateItemDisplay(seshid, budget=False):
                 r = requests.post(url, data=json.dumps(valid_data), headers=headers)
                 jsondata = json.loads(r.content)
                 scores = jsondata['data']
-                section_scores = [(item_idxs[i], scores[sum(item_lens_ln[:i]):sum(item_lens_ln[:i])+ln]) for i,ln in enumerate(item_lens_ln)]
+                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)]
                 end = time.time()
                 
                 print('updateDisplay -- item images downloaded: ', str(end-start), section_scores, item_idxs)