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])
                 st.write("_number of columns_", df.shape[1])
@@ -123,11 +128,7 @@ if upload_file is not None:
                 df.columns = new_columns
                 st.dataframe(df, key="excel_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])
                 st.write("_number of columns_", df.shape[1])
                 st.session_state.df = df