Segment Subsets by Color and Resize Data Points

app.py

@@ -2,6 +2,8 @@ import streamlit as st
 import pandas as pd
 from bokeh.plotting import figure
 from bokeh.models import ColumnDataSource, HoverTool
+from bokeh.palettes import Category10
+from bokeh.transform import factor_cmap
 
 # --- Define Styles ---
 st.markdown(
@@ -46,6 +48,10 @@ source = ColumnDataSource(data=dict(
     img=df['img']
 ))
 
+unique_labels = df['label'].unique().tolist()
+palette = Category10[len(unique_labels)] if len(unique_labels) <= 10 else Category10[10]
+
+
 # Configure figure
 TOOLTIPS = """
 <div>
@@ -60,6 +66,6 @@ TOOLTIPS = """
 
 
 p = figure(width=400, height=400, tooltips=TOOLTIPS, title="")
-p.scatter('x', 'y', size=20, source=source)
+p.scatter('x', 'y', size=8, source=source, color=factor_cmap('label', palette=palette, factors=unique_labels))
 
 st.bokeh_chart(p)