Update app.py

app.py

@@ -18,16 +18,13 @@ if uploaded_file is not None:
     # Generate a treemap or sunburst plot based on data types
     numerical_cols = df.select_dtypes(include=["float", "int"]).columns
     categorical_cols = df.select_dtypes(include=["object"]).columns
-
-    fig = px.treemap(df, path=categorical_cols)
-    st.plotly_chart(fig)
         
-    # if len(numerical_cols) >= 2:
-    #     fig = px.scatter_matrix(df, dimensions=numerical_cols)
-    #     st.plotly_chart(fig)
-    # elif len(categorical_cols) >= 2:
-    #     fig = px.treemap(df, path=categorical_cols)
-    #     st.plotly_chart(fig)
-    # else:
-    #     fig = px.sunburst(df, path=categorical_cols + numerical_cols)
-    #     st.plotly_chart(fig)
+    if len(numerical_cols) >= 2:
+        fig = px.scatter_matrix(df, dimensions=numerical_cols)
+        st.plotly_chart(fig)
+    elif len(categorical_cols) >= 2:
+        fig = px.treemap(df, path=categorical_cols)
+        st.plotly_chart(fig)
+    else:
+        fig = px.sunburst(df, path=categorical_cols + numerical_cols)
+        st.plotly_chart(fig)