Move the token counts to the front

tapas_visualizer.py

@@ -120,9 +120,9 @@ class TapasVisualizer:
                 row_html += f"<{cell_element}>{cell_html}</{cell_element}>"
                 row_token_cnt += len(cur_cell_tokens)
             cumulative_cnt += row_token_cnt
-            # cnt_str = f'{row_token_cnt} | {cumulative_cnt}'
-            row_html += f'<td style="border: none;" align="right">{self.style_span(str(row_token_cnt), ["non-token", "count"])}</td>'
-            row_html += f'<td style="border: none;" align="right">{self.style_span(str(cumulative_cnt), ["non-token", "count"])}</td>'
+            cnt_html = (f'<td style="border: none;" align="right">{self.style_span(str(cumulative_cnt), ["non-token", "count"])}</td>'
+                        f'<td style="border: none;" align="right">{self.style_span(f"<+{row_token_cnt}", ["non-token", "count"])}</td>')
+            row_html = cnt_html + row_html
             table_html += f'<tr>{row_html}</tr>'
 
         return table_html, cumulative_cnt
@@ -141,12 +141,14 @@ class TapasVisualizer:
             if int(segment_id) == 1:
                 cell_tokens[(row_id, col_id)].append(token_text)
 
+        table_html = '<tr><td style="border: none;" colspan="2" align="left">#Tokens</td></tr>'
+
         table_html, cumulative_cnt = self.cells_to_html(cell_vals=[table.columns],
                                                         cell_tokens=cell_tokens,
                                                         row_id_start=0,
                                                         cell_element="th",
                                                         cumulative_cnt=0,
-                                                        table_html="")
+                                                        table_html=table_html)
 
         table_html, cumulative_cnt = self.cells_to_html(cell_vals=table.values,
                                                         cell_tokens=cell_tokens,