Update app.py

app.py

@@ -98,6 +98,11 @@ if upload_file is not None:
                 new_columns = [f'column_{i+1}' for i in range(len(df.columns))]
                 df.columns = new_columns
                 st.dataframe(df, key="csv_dataframe")
+                fig = px.treemap(df, path=[px.Constant("all"), 'column_1', 'column_2'],
+                                 values='column_1', color='column_2')
+                fig.update_layout(margin = dict(t=50, l=25, r=25, b=25))
+                st.subheader("Tree map", divider = "red")
+                st.plotly_chart(fig)
                 
                 
                 st.write("_number of rows_", df.shape[0])