Update app.py

app.py

@@ -637,14 +637,25 @@ def create_avatar():
         draw.line([start, end], fill=(0, 255, 255, 50), width=1)
     
     return img
- # Adjust alpha channel of heatmap
+def create_avatar_with_heatmap(sensation_map, sensation_type):
+    # Create the base avatar
+    avatar_img = create_avatar()
+    
+    # Create a heatmap
+    heatmap_img = create_heatmap(sensation_map, sensation_type)
+    
+    # Resize heatmap to match avatar size
+    heatmap_img = heatmap_img.resize((AVATAR_WIDTH, AVATAR_HEIGHT))
+    
+    # Adjust alpha channel of heatmap
     data = np.array(heatmap_img)
     if data.shape[2] == 3:  # If RGB, add an alpha channel
         data = np.concatenate([data, np.full((data.shape[0], data.shape[1], 1), 255, dtype=np.uint8)], axis=2)
     data[:, :, 3] = data[:, :, 3] * 0.5  # Reduce opacity to 50%
     heatmap_img = Image.fromarray(data)
     
-    combined_img = Image.alpha_composite(avatar_img.convert('RGBA'), heatmap_img)
+    # Combine avatar and heatmap
+    combined_img = Image.alpha_composite(avatar_img.convert('RGBA'), heatmap_img.convert('RGBA'))
     return combined_img
 
 # Streamlit app
@@ -652,7 +663,6 @@ st.title("NeuraSense AI: Advanced Humanoid Techno-Sensory Simulation")
 
 
 
-# Create two columns
 
 # Create three columns
 col1, col2, col3 = st.columns(3)