Update app.py

app.py

@@ -102,14 +102,13 @@ if upload_file is not None:
                 st.subheader("Select columns for the Tree Map", divider="blue")
                 parent_column = st.selectbox("Select the parent column:", all_columns)
                 value_column = st.selectbox("Select the value column:", all_columns)
-                color_column = st.selectbox("Select the color column (optional):", ["None"] + all_columns)
+                
                 if parent_column and value_column:
                     path_columns = [px.Constant("all"), parent_column]
-                    color_param = value_column if color_column == "None" else color_column
+                    
                     fig = px.treemap(df,
                              path=path_columns,
-                             values=value_column,
-                             color=color_param)
+                             values=value_column)
                     fig.update_layout(margin=dict(t=50, l=25, r=25, b=25))
                     st.subheader("Tree map", divider="red")
                     st.plotly_chart(fig)