Update app.py

app.py

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