✨ Feat(streamlit_reportview.py): Display dataframes in streamlit apps with st-aggrid to add searching and filtering features. Fixes #62

Author: sayalaruano

pyproject.toml

@@ -13,6 +13,8 @@ version = "0.1.0"
 [tool.poetry.dependencies]
 python = ">=3.9,<3.9.7 || >3.9.7,<4.0"
 streamlit = "1.39.0"
+streamlit-aggrid = "1.1.0"
+quarto-cli = "*"
 plotly = "5.15.0"
 pyvis = "^0.3.2"
 pandas = {extras = ["parquet"], version = "^2.2.3"}
@@ -28,6 +30,7 @@ vl-convert-python = "^1.7.0"
 dataframe-image = "^0.2.6"
 strenum = { version = "^0.4.15", python = "<3.11" }
 pyyaml = "^6.0.2"
+
 # optional doc depencencies, follow approach as described here:
 # https://github.com/python-poetry/poetry/issues/2567#issuecomment-646766059
 sphinx = {version="*", optional=true}
@@ -36,15 +39,12 @@ myst-nb = {version="*", optional=true}
 ipywidgets = {version="*", optional=true}
 sphinx-new-tab-link = {version = "!=0.2.2", optional=true}
 jupytext = {version="*", optional=true}
-# quarto
-quarto-cli = "*"
 
 [tool.poetry.group.dev.dependencies]
 ipykernel = {version="^6.29.5", optional=true}
 
 [tool.poetry.requires-plugins]
 poetry-dynamic-versioning = { version = ">=1.0.0,<2.0.0", extras = ["plugin"] }
-
 
 [tool.poetry-dynamic-versioning]
 enable = true

src/vuegen/streamlit_reportview.py

@@ -466,9 +466,26 @@ def _generate_dataframe_content(self, dataframe) -> List[str]:
             read_function = read_function_mapping[file_extension]
             dataframe_content.append(f"""df = pd.{read_function.__name__}('{dataframe.file_path}')""")
 
-            # Display the dataframe
-            dataframe_content.append("st.dataframe(df, use_container_width=True)")
-        
+            # Displays a DataFrame using AgGrid with configurable options.
+            dataframe_content.append("""
+# Displays a DataFrame using AgGrid with configurable options.
+grid_builder = GridOptionsBuilder.from_dataframe(df)
+grid_builder.configure_default_column(editable=True, groupable=True)
+grid_builder.configure_side_bar(filters_panel=True, columns_panel=True)
+grid_builder.configure_selection(selection_mode="multiple")
+grid_builder.configure_pagination(enabled=True, paginationAutoPageSize=False, paginationPageSize=20)
+grid_options = grid_builder.build()
+
+AgGrid(df, gridOptions=grid_options)
+
+# Button to download the df
+df_csv = utils.convert_df(df)
+st.download_button(
+    label=f"Download dataframe as CSV",
+    data=df,
+    file_name=f"dataframe.csv",
+    mime='text/csv',
+)""")
         except Exception as e:
             self.report.logger.error(f"Error generating content for DataFrame: {dataframe.title}. Error: {str(e)}")
             raise