[Feature Request] Show index and enable styling in data table

[cod3licious]

With other frameworks (e.g., streamlit) it is possible to show pandas DataFrames with their style attributes, for example, to color the individual cells based on the values contained in them to highlight high or low values.

The rx.data_table unfortunately does not seem to support pandas Styler objects. One workaround is to use the pandas to_html() function and embed the html directly, however, then of course you're losing the ability to sort, pagination, etc. and the table looks quite ugly by default.

Code example:

from copy import deepcopy

import numpy as np
import pandas as pd
import reflex as rx

class Data(rx.State):
    df: pd.DataFrame = pd.DataFrame(
        np.random.randn(10, 5), columns=[f"col_{i}" for i in range(5)], index=[f"idx_{i}" for i in range(10)]
    )
    df_styled: str = deepcopy(df).style.format(precision=2).background_gradient(axis=0, cmap="coolwarm").to_html()

def show_df():
    return rx.vstack(
        rx.data_table(
            data=Data.df,
            pagination=True,
            search=True,
            sort=True,
        ),
        rx.html(Data.df_styled),
    )

app = rx.App()
app.add_page(show_df)

This gives:

Ideally, I'd want to use reflex's data_table but with the same styling as the html table below it.

Additionally, it would also be great if data_table would support showing the index of the dataframe, since sometimes this contains actual names instead of just row indices.

[SeqCrafter]

The feature is supported in Ag-grid, which is a component in the Reflex Enterprise package