Update app.py

app.py

@@ -192,7 +192,7 @@ def append_token(token_id: int):
     text_rx.set(text_rx.value + decoded)
     # lock highlight to the clicked token and keep it after re-predict
     preview_token(int(token_id))
-    on_predict()  # refresh next tokens but do NOT override highlight anymore
+    on_predict()  # refresh next tokens but preserve highlight
 
 # ------------------ Auto-predict on typing (debounced) ------------------
 @solara.component
@@ -221,7 +221,7 @@ def AutoPredictWatcher():
     solara.use_effect(effect, [text, auto])
     return solara.Text("", style={"display": "none"})
 
-# ------------------ UI: clickable + hoverable list ------------------
+# ------------------ UI: clickable + HOVERABLE rows ------------------
 @solara.component
 def PredictionsList():
     df = preds_rx.value
@@ -240,21 +240,17 @@ def PredictionsList():
             tok_disp = display_token_from_id(tid)
             label = fmt_row(i, prob, tid, tok_disp)
 
-            # Wrap a Button (click) inside a Div (hover). Div events pass (event, ...),
-            # Button click is reliable across Solara versions.
-            with solara.Div(
-                attributes={"tabindex": "0", "role": "button"},
-                on_mouse_enter=lambda e=None, tid=tid: preview_token(tid),
-                on_mouse_over=lambda e=None, tid=tid: preview_token(tid),
-                on_mouse_move=lambda e=None, tid=tid: preview_token(tid),
-                on_focus=lambda e=None, tid=tid: preview_token(tid),
-            ):
-                solara.Button(
-                    label,
-                    on_click=lambda tid=tid: append_token(tid),
-                    classes=["rowbtn"],
-                    style={"justifyContent": "flex-start", "width": "100%"},
-                )
+            # Put ALL hover handlers directly on the Button.
+            solara.Button(
+                label,
+                on_click=lambda tid=tid: append_token(tid),                 # click to append
+                on_mouse_enter=lambda e=None, tid=tid: preview_token(tid),  # hover preview
+                on_mouse_over=lambda e=None, tid=tid: preview_token(tid),   # extra safety
+                on_mouse_move=lambda e=None, tid=tid: preview_token(tid),   # tracks movement
+                on_focus=lambda e=None, tid=tid: preview_token(tid),        # keyboard focus
+                classes=["rowbtn"],
+                style={"justifyContent": "flex-start", "width": "100%"},
+            )
 
 # ------------------ Page ------------------
 @solara.component
@@ -298,4 +294,4 @@ def Page():
 
 # Seed initial predictions and mount
 on_predict()
-Page()
+Page()