Update app.py

app.py

@@ -252,18 +252,22 @@ elif tab == "Supervised SHAP Clustering":
 
     # Initialize an empty figure
     fig = go.Figure()
-
+    
+    # Define color mapping for clusters
+    cluster_colors = ['red', 'blue', 'purple']
+    
     # Plot clustered genes based on PCA components
-    for cluster in df_for_plot['Cluster'].unique():
+    for i, cluster in enumerate(df_for_plot['Cluster'].unique()):
         filtered_df = df_for_plot[(df_for_plot['Cluster'] == cluster) & (df_for_plot['SpecialGroup'] == 'None')]
         fig.add_trace(go.Scatter(
             x=filtered_df['PCA_1'], y=filtered_df['PCA_2'],
             mode='markers',
             name=f'Cluster {cluster}',
             text=filtered_df['Gene'],
+            marker=dict(color=cluster_colors[i]),
             hoverinfo="text+x+y",
         ))
-
+    
     # Overlay "Most Likely Training Gene"
     filtered_df = df_for_plot[df_for_plot['SpecialGroup'] == 'Most Likely Training Gene']
     fig.add_trace(go.Scatter(
@@ -271,10 +275,10 @@ elif tab == "Supervised SHAP Clustering":
         mode='markers',
         name='Most Likely Training Gene',
         text=filtered_df['Gene'],
-        marker=dict(color='rgba(255, 0, 0, .9)'),
+        marker=dict(color='black'),
         hoverinfo="text+x+y",
     ))
-
+    
     # Overlay "User Input Gene"
     filtered_df = df_for_plot[df_for_plot['SpecialGroup'] == 'User Input Gene']
     fig.add_trace(go.Scatter(
@@ -282,10 +286,10 @@ elif tab == "Supervised SHAP Clustering":
         mode='markers',
         name='User Input Gene',
         text=filtered_df['Gene'],
-        marker=dict(color='rgba(0, 255, 0, .9)'),
+        marker=dict(color='orange'),
         hoverinfo="text+x+y",
     ))
-
+    
     # Customize layout
     fig.update_layout(
         title='Supervised SHAP Clustering with PCA',
@@ -294,5 +298,5 @@ elif tab == "Supervised SHAP Clustering":
         showlegend=True,
         legend_title_text='Gene Category',
     )
-
+    
     st.plotly_chart(fig, use_container_width=True)