import streamlit as st
from typing import List, Tuple
from bokeh.plotting import figure
from bokeh.models import ColumnDataSource
from bokeh.events import Tap

# Constants
WIDTH, HEIGHT = 600, 600
ELASTICITY = 0.3
DAMPING = 0.7

# Create touch points
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]))

# Set up the Bokeh plot
p = figure(width=WIDTH, height=HEIGHT, tools="", toolbar_location=None)
p.circle('x', 'y', size=5, color="navy", alpha=0.5, source=source)

# Streamlit app
st.title("Artificial Touch Simulation")

# Create a Streamlit container for the Bokeh plot
chart = st.bokeh_chart(p)

def update_points():
    global touch_points, velocities, is_affected
    
    # Apply spring force
    for i, (x, y) in enumerate(touch_points):
        force_x = (original_points[i][0] - x) * ELASTICITY
        force_y = (original_points[i][1] - y) * ELASTICITY
        velocities[i] = (velocities[i][0] + force_x, velocities[i][1] + force_y)
    
    # Apply damping
    for i, (vx, vy) in enumerate(velocities):
        velocities[i] = (vx * DAMPING, vy * DAMPING)
    
    # Update position
    for i, (x, y) in enumerate(touch_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, is_affected
    
    x, y = event.x, event.y
    for i, (tx, ty) in enumerate(touch_points):
        distance = ((tx - x)**2 + (ty - y)**2)**0.5
        if distance < 30:
            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()

p.on_event(Tap, on_tap)

while True:
    update_points()
    chart.bokeh_chart(p)
    st.experimental_rerun()