Proposed changes

app.py

@@ -8,10 +8,10 @@ import plotly.graph_objects as go
 def plot_decision(
         clf: SVC,
         X: np.ndarray,
+        y: np.array,
         x_range: np.array,
         y_range: np.array,
         weights: np.array,
-        colors: list[str],
         title: str
     ):
     # plot the decision function
@@ -28,7 +28,7 @@ def plot_decision(
             x=x_range,
             y=y_range,
             z=Z,
-            colorscale="gray",
+            colorscale="Viridis",
             opacity=0.75,        
             showscale=False,
         )
@@ -40,7 +40,8 @@ def plot_decision(
             y=X[:, 1],
             mode="markers",
             marker=dict(
-                color=colors,
+                color=y,
+                colorscale="viridis",
                 size=(weights + 5) * 2
             ),
         )
@@ -76,25 +77,24 @@ def app_fn(seed: int, weight_1: int, weight_2: int):
 
     # Plotting
     x_range = np.arange(-4, 5, 0.1)
-    colors = np.where(np.array(y)==1, "white", "black")
 
     fig_no_weights = plot_decision(
         clf_no_weights, 
         X,
+        y,
         x_range, 
         x_range, 
-        sample_weight_constant, 
-        colors, 
+        sample_weight_constant,  
         "SVM without Weights"
     )
 
     fig_weights = plot_decision(
         clf_weights,
         X,
+        y,
         x_range,
         x_range,
         sample_weight_last_ten,
-        colors,
         "SVM with Weights"
     )
 
@@ -122,12 +122,14 @@ with gr.Blocks(title=title) as demo:
         seed = gr.inputs.Slider(0, 100, 1, default=0, label="Seed")
         weight_1 = gr.inputs.Slider(0, 20, 1, default=5, label="Weight for last 5 Samples")
         weight_2 = gr.inputs.Slider(0, 20, 1, default=15, label="Weight for Sample 10")
-    btn = gr.Button("Run")
+    # btn = gr.Button("Run")
     with gr.Row():
         fig_no_weights = gr.Plot(label="SVM without Weights")
         fig_weights = gr.Plot(label="SVM with Weights")
         
-    btn.click(fn=app_fn, outputs=[fig_no_weights, fig_weights], inputs=[seed, weight_1, weight_2])
+    seed.change(fn=app_fn, outputs=[fig_no_weights, fig_weights], inputs=[seed, weight_1, weight_2])
+    weight_1.change(fn=app_fn, outputs=[fig_no_weights, fig_weights], inputs=[seed, weight_1, weight_2])
+    weight_2.change(fn=app_fn, outputs=[fig_no_weights, fig_weights], inputs=[seed, weight_1, weight_2])
     demo.load(fn=app_fn, outputs=[fig_no_weights, fig_weights], inputs=[seed, weight_1, weight_2])
 
 demo.launch()