Update app.py

app.py

@@ -15,11 +15,27 @@ ELASTICITY = 0.3
 DAMPING = 0.7
 
 # Create sensation map
-sensation_map = np.zeros((HEIGHT, WIDTH))
-for y in range(HEIGHT):
-    for x in range(WIDTH):
-        # Create a complex sensation map with various regions
-        sensation_map[y, x] = np.sin(x/30) * np.cos(y/30) * 5 + np.random.normal(0, 0.5)
+def create_sensation_map(width, height):
+    sensation_map = np.zeros((height, width, 3))  # RGB channels for pain, pleasure, and neutral
+    for y in range(height):
+        for x in range(width):
+            # Base sensation
+            base = np.sin(x/30) * np.cos(y/30) * 0.5 + np.random.normal(0, 0.1)
+            
+            # Pain regions (red channel)
+            pain = np.exp(-((x-150)**2 + (y-150)**2) / 5000) + np.exp(-((x-450)**2 + (y-450)**2) / 5000)
+            
+            # Pleasure regions (green channel)
+            pleasure = np.exp(-((x-300)**2 + (y-300)**2) / 5000) + np.exp(-((x-150)**2 + (y-450)**2) / 5000)
+            
+            # Neutral sensation (blue channel)
+            neutral = 1 - (pain + pleasure)
+            
+            sensation_map[y, x] = [pain, pleasure, neutral]
+    
+    return sensation_map
+
+sensation_map = create_sensation_map(WIDTH, HEIGHT)
 
 # Set up the Hugging Face pipeline
 @st.cache_resource
@@ -36,21 +52,32 @@ touch_container = st.container()
 
 def calculate_sensation(x, y, pressure, duration):
     # Get sensation from the map
-    base_sensation = sensation_map[int(y), int(x)]
+    sensation = sensation_map[int(y), int(x)]
     
     # Modify sensation based on pressure and duration
-    modified_sensation = base_sensation * pressure * (1 + np.log(duration + 1))
+    modified_sensation = sensation * pressure * (1 + np.log(duration + 1))
     
     return modified_sensation
 
 def on_touch(x, y, pressure, duration):
     sensation = calculate_sensation(x, y, pressure, duration)
+    pain = sensation[0]
+    pleasure = sensation[1]
+    neutral = sensation[2]
     
     # Generate a description of the touch
     st.write(f"Touch at ({x:.2f}, {y:.2f}) with pressure {pressure:.2f} for {duration:.2f} seconds")
-    st.write(f"Sensation: {sensation:.2f}")
+    st.write(f"Pain: {pain:.2f}, Pleasure: {pleasure:.2f}, Neutral: {neutral:.2f}")
+    
+    # Determine the dominant sensation
+    if pain > pleasure and pain > neutral:
+        dominant = "pain"
+    elif pleasure > pain and pleasure > neutral:
+        dominant = "pleasure"
+    else:
+        dominant = "neutral"
     
-    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:"
+    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 {dominant} sensation. Pain: {pain:.2f}, Pleasure: {pleasure:.2f}, Neutral: {neutral:.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)
 
@@ -62,7 +89,7 @@ if 'last_touch_position' not in st.session_state:
 
 # Main app logic
 fig, ax = plt.subplots(figsize=(6, 6))
-ax.imshow(sensation_map, cmap='coolwarm', extent=[0, WIDTH, HEIGHT, 0])
+ax.imshow(sensation_map)
 ax.axis('off')
 
 # Convert matplotlib figure to Image
@@ -117,4 +144,4 @@ if canvas_result.json_data is not None:
         st.session_state.last_touch_position = None
 
 st.write("Click and drag on the image to simulate touch. The color represents different sensations.")
-st.write("Red areas are more sensitive (pain or intense pleasure), while blue areas are less sensitive.")
+st.write("Red areas are pain regions, green areas are pleasure regions, and blue areas are neutral.")