Update app.py

app.py

@@ -14,6 +14,7 @@ num_points = 20
 touch_points: List[Tuple[float, float]] = [(x, y) for x in range(50, WIDTH-50, int((WIDTH-100)/(num_points-1))) for y in range(50, HEIGHT-50, int((HEIGHT-100)/(num_points-1)))]
 original_points = touch_points.copy()
 velocities: List[Tuple[float, float]] = [(0.0, 0.0)] * len(touch_points)
+is_affected: List[bool] = [False] * len(touch_points)
 
 source = ColumnDataSource(data=dict(x=[x for x, y in touch_points], y=[y for x, y in touch_points]))
 
@@ -28,7 +29,7 @@ st.title("Artificial Touch Simulation")
 chart = st.bokeh_chart(p)
 
 def update_points():
-    global touch_points, velocities
+    global touch_points, velocities, is_affected
     
     # Apply spring force
     for i, (x, y) in enumerate(touch_points):
@@ -45,11 +46,14 @@ def update_points():
         vx, vy = velocities[i]
         touch_points[i] = (x + vx, y + vy)
     
+    # Reset affected flags
+    is_affected = [False] * len(touch_points)
+    
     # Update Bokeh data source
     source.data = dict(x=[x for x, y in touch_points], y=[y for x, y in touch_points])
 
 def on_tap(event):
-    global touch_points, velocities
+    global touch_points, velocities, is_affected
     
     x, y = event.x, event.y
     for i, (tx, ty) in enumerate(touch_points):
@@ -58,6 +62,7 @@ def on_tap(event):
             force_x = (tx - x) / distance
             force_y = (ty - y) / distance
             velocities[i] = (velocities[i][0] - force_x * 10, velocities[i][1] - force_y * 10)
+            is_affected[i] = True
     
     st.write(f"Touch at ({x:.2f}, {y:.2f})")
     update_points()