Update app.py

app.py

@@ -674,8 +674,11 @@ class TinyAgent:
             meta['source']='analytic'
         else:
             chosen_id=cnn_action_idx+1
-            meta={'source':'cnn','probs':probs.tolist(),
-                  'candidates':candidates[:4] if candidates else []}
+            # Package read_board result for display even in CNN fallback
+            cnn_cands=[(r, cand_answer, cand_conf)] if cand_answer is not None else []
+            meta={'source':'cnn','probs':probs.tolist(),'candidates':cnn_cands,
+                  'hypothesis':cand_reasoning[0][:60] if cand_reasoning else 'none',
+                  'conf':cand_conf,'n_diffs':len(pixel_diff(grid,cand_answer)) if cand_answer is not None else 0}
 
         self.prev_feat=feat; self.prev_action=cnn_action_idx
         self.step_count+=1
@@ -787,22 +790,29 @@ def pull_frame():
 
     grid=data['grid']; meta=data['meta']; step=data['step']
     levels=data['levels']; state=data['state']; action=data['action']
-    candidates=meta.get('candidates',[])
     source=meta.get('source','cnn')
+    hyp_str=meta.get('hypothesis','none')
+    cand_conf=meta.get('conf',0.0)
+    n_diffs=meta.get('n_diffs',0)
+
+    # Build display candidates list: [(label, grid, conf)]
+    raw_cands=meta.get('candidates',[])
+    # raw_cands entries are (reasoning_str, grid, conf)
+    # Normalise to (short_label, grid, conf)
+    disp_cands=[]
+    for entry in raw_cands:
+        if len(entry)==3:
+            label=str(entry[0])[:40]; cgrid=entry[1]; cconf=entry[2]
+            if isinstance(cgrid, np.ndarray): disp_cands.append((label,cgrid,cconf))
 
     # Determine what to highlight
     mark_cell=None; highlight=None
-    if source=='analytic' and 'cell' in meta:
-        r,c,v=meta['cell']
-        best_cand_name=meta.get('hypothesis','?')
-        best_conf=meta.get('conf',0)
-        # Compute Re-side diff for candidate
-        cand_name,cand_grid,cand_conf=(candidates[0] if candidates
-                                        else (best_cand_name,grid,best_conf))
+    if source=='analytic' and 'cell' in meta and disp_cands:
+        _,cand_grid,_=disp_cands[0]
         if cand_grid.shape==grid.shape:
             all_diffs=pixel_diff(grid,cand_grid)
-            highlight=all_diffs[:20]  # show up to 20 wrong cells in red
-            mark_cell=meta['cell']    # cyan star on the cell we're clicking
+            highlight=all_diffs[:20]
+            mark_cell=meta['cell']
 
     source_emoji='🧠' if source=='analytic' else '🎲'
     _latest['grid_img']=render_grid(
@@ -811,16 +821,20 @@ def pull_frame():
         highlight=highlight,
         mark_cell=mark_cell)
 
-    # Im side: hypothesis ranking
-    _latest['hyp_img']=render_hypothesis_panel(candidates)
+    # Im side: hypothesis panel — show read_board reasoning as bar
+    if disp_cands:
+        # Convert to format render_hypothesis_panel expects: [(name,grid,conf)]
+        _latest['hyp_img']=render_hypothesis_panel(disp_cands)
+    else:
+        _latest['hyp_img']=None
 
-    # Re side: candidate grid (what Im thinks the answer looks like)
-    if candidates and candidates[0][1].shape==grid.shape:
-        cname,cgrid,cconf=candidates[0]
+    # Re side: candidate grid
+    if disp_cands and disp_cands[0][1].shape==grid.shape:
+        cname,cgrid,cconf2=disp_cands[0]
         diffs=pixel_diff(grid,cgrid)
         _latest['cand_img']=render_grid(
             cgrid,
-            title=f"Im candidate: {cname} (conf={cconf:.2f}) — {len(diffs)} cells differ",
+            title=f"Im answer: {cname[:35]} (conf={cconf2:.2f}) — {len(diffs)} cells differ",
             highlight=diffs[:20])
     else:
         _latest['cand_img']=None