Update app.py

app.py

@@ -1,7 +1,9 @@
 import streamlit as st
+import numpy as np
 from typing import List, Tuple
 from transformers import pipeline
 import matplotlib.pyplot as plt
+import time
 
 # Constants
 WIDTH, HEIGHT = 600, 600
@@ -15,6 +17,10 @@ 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)
 
+# Create pain and pleasure points
+pain_points = [(100, 100), (500, 500)]
+pleasure_points = [(300, 300), (200, 400)]
+
 # Set up the Hugging Face pipeline
 @st.cache_resource
 def load_model():
@@ -23,7 +29,7 @@ def load_model():
 text_generator = load_model()
 
 # Streamlit app
-st.title("Artificial Touch Simulation")
+st.title("Advanced Artificial Touch Simulation")
 
 # Create a Streamlit container for the touch simulation
 touch_container = st.container()
@@ -49,7 +55,19 @@ def update_points():
     # Reset affected flags
     is_affected = [False] * len(touch_points)
 
-def on_tap(x, y):
+def calculate_sensation(x, y, pressure, duration):
+    sensation = 0
+    for px, py in pain_points:
+        distance = np.sqrt((x - px)**2 + (y - py)**2)
+        if distance < 50:
+            sensation -= (50 - distance) * pressure * (duration ** 1.5) / 50
+    for px, py in pleasure_points:
+        distance = np.sqrt((x - px)**2 + (y - py)**2)
+        if distance < 50:
+            sensation += (50 - distance) * pressure / 50
+    return sensation
+
+def on_tap(x, y, pressure, duration):
     global touch_points, velocities, is_affected
     
     for i, (tx, ty) in enumerate(touch_points):
@@ -57,12 +75,17 @@ def on_tap(x, y):
         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)
+            velocities[i] = (velocities[i][0] - force_x * 10 * pressure, velocities[i][1] - force_y * 10 * pressure)
             is_affected[i] = True
     
+    sensation = calculate_sensation(x, y, pressure, duration)
+    
     # Generate a description of the touch
-    st.write(f"Touch at ({x:.2f}, {y:.2f})")
-    text = text_generator(f"The user touched the screen at ({x:.2f}, {y:.2f}).", max_length=100, num_return_sequences=1, do_sample=True, top_k=50, top_p=0.95, num_beams=1)[0]['generated_text']
+    st.write(f"Touch at ({x:.2f}, {y:.2f}) with pressure {pressure:.2f} for {duration:.2f} seconds")
+    st.write(f"Sensation: {sensation:.2f}")
+    
+    prompt = f"The user touched the screen at ({x:.2f}, {y:.2f}) with a pressure of {pressure:.2f} for {duration:.2f} seconds, resulting in a sensation of {sensation:.2f}. Describe the experience:"
+    text = text_generator(prompt, max_length=100, num_return_sequences=1, do_sample=True, top_k=50, top_p=0.95, num_beams=1)[0]['generated_text']
     st.write(text)
     
     update_points()
@@ -72,6 +95,12 @@ if 'x' not in st.session_state:
     st.session_state.x = 0
 if 'y' not in st.session_state:
     st.session_state.y = 0
+if 'pressure' not in st.session_state:
+    st.session_state.pressure = 1.0
+if 'duration' not in st.session_state:
+    st.session_state.duration = 0.0
+if 'touch_start_time' not in st.session_state:
+    st.session_state.touch_start_time = None
 
 # Main app logic
 fig, ax = plt.subplots(figsize=(6, 6))
@@ -82,12 +111,29 @@ for i, (x, y) in enumerate(touch_points):
     color = "red" if is_affected[i] else "navy"
     ax.add_artist(plt.Circle((x, y), 5, color=color, alpha=0.5))
 
+for x, y in pain_points:
+    ax.add_artist(plt.Circle((x, y), 10, color="red", alpha=0.3))
+
+for x, y in pleasure_points:
+    ax.add_artist(plt.Circle((x, y), 10, color="green", alpha=0.3))
+
 touch_container.pyplot(fig)
 
-if touch_container.button("Tap the screen"):
-    on_tap(st.session_state.x, st.session_state.y)
+col1, col2 = st.columns(2)
+with col1:
+    st.session_state.x = st.slider("X coordinate", 0, WIDTH, st.session_state.x)
+    st.session_state.y = st.slider("Y coordinate", 0, HEIGHT, st.session_state.y)
+with col2:
+    st.session_state.pressure = st.slider("Pressure", 0.1, 2.0, st.session_state.pressure)
+
+if touch_container.button("Start Touch"):
+    st.session_state.touch_start_time = time.time()
 
-st.session_state.x = st.slider("X coordinate", 0, WIDTH, st.session_state.x)
-st.session_state.y = st.slider("Y coordinate", 0, HEIGHT, st.session_state.y)
+if touch_container.button("End Touch"):
+    if st.session_state.touch_start_time is not None:
+        st.session_state.duration = time.time() - st.session_state.touch_start_time
+        on_tap(st.session_state.x, st.session_state.y, st.session_state.pressure, st.session_state.duration)
+        st.session_state.touch_start_time = None
+        st.session_state.duration = 0.0
 
 update_points()