🐛 Fix(streamlit_reportview.py): add aggird imports and fix issues in the code to loaf the dfs

Author: sayalaruano

src/vuegen/streamlit_reportview.py

@@ -468,6 +468,7 @@ def _generate_dataframe_content(self, dataframe) -> List[str]:
 
             # Displays a DataFrame using AgGrid with configurable options.
             dataframe_content.append("""
+df_index = 1
 # Displays a DataFrame using AgGrid with configurable options.
 grid_builder = GridOptionsBuilder.from_dataframe(df)
 grid_builder.configure_default_column(editable=True, groupable=True)
@@ -479,13 +480,14 @@ def _generate_dataframe_content(self, dataframe) -> List[str]:
 AgGrid(df, gridOptions=grid_options)
 
 # Button to download the df
-df_csv = utils.convert_df(df)
+df_csv = df.to_csv(sep=',', header=True, index=False).encode('utf-8')
 st.download_button(
-    label=f"Download dataframe as CSV",
-    data=df,
-    file_name=f"dataframe.csv",
+    label="Download dataframe {df_index} as CSV",
+    data=df_csv,
+    file_name="dataframe_{df_index}.csv",
     mime='text/csv',
-)""")
+    key="download_button_{df_index}")
+df_index += 1""")
         except Exception as e:
             self.report.logger.error(f"Error generating content for DataFrame: {dataframe.title}. Error: {str(e)}")
             raise
@@ -724,7 +726,7 @@ def _generate_component_imports(self, component: r.Component) -> List[str]:
                 r.PlotType.PLOTLY: ['import json', 'import requests'],
                 r.PlotType.INTERACTIVE_NETWORK: ['import requests']
             },
-            'dataframe': ['import pandas as pd'],
+            'dataframe': ['import pandas as pd', 'from st_aggrid import AgGrid, GridOptionsBuilder'],
             'markdown': ['import requests'],
             'chatbot': ['import time', 'import json', 'import requests']
         }