Update app.py

app.py

@@ -8,6 +8,7 @@ import plotly.express as px
 import base64
 import plotly.figure_factory as ff
 import plotly.graph_objects as go
+from scipy.spatial import ConvexHull
 
 st.set_page_config(layout="wide")
 
@@ -120,12 +121,14 @@ with tab1:
     # For tab1
     fig = go.Figure()
     
-    # Add shaded regions
+    # Add shaded regions using convex hull
     for cluster in np.unique(y_kmeans):
         cluster_data = df[df['cluster'] == cluster]
-        x_data = cluster_data[df.columns[0]]
-        y_data = cluster_data[df.columns[1]]
-        fig.add_trace(go.Histogram2dContour(x=x_data, y=y_data, colorscale=px.colors.qualitative.Set1, showscale=False, opacity=0.25))
+        x_data = cluster_data[df.columns[0]].values
+        y_data = cluster_data[df.columns[1]].values
+        if len(cluster_data) > 2:  # ConvexHull requires at least 3 points
+            hull = ConvexHull(cluster_data[[df.columns[0], df.columns[1]]])
+            fig.add_trace(go.Polygon(x=x_data[hull.vertices], y=y_data[hull.vertices], fillcolor=px.colors.qualitative.Set1[cluster], opacity=0.5))
     
     # Add scatter plot
     fig.add_trace(go.Scatter(x=df[df.columns[0]], y=df[df.columns[1]], mode='markers', marker=dict(color=y_kmeans, colorscale=px.colors.qualitative.Set1), showlegend=False))
@@ -146,8 +149,6 @@ with tab1:
     # Update layout
     fig.update_layout(width=1200, height=500)
 
-
-
     st.write("""
     ### Visualizing Groups
     ##### Here are the groups from our tidying method. Each color has a number at its center, representing its group.