Update app.py

app.py

@@ -336,9 +336,12 @@ custom_css = """
 
 # Apply the custom CSS
 st.markdown(custom_css, unsafe_allow_html=True)
+AVATAR_WIDTH = 600
+AVATAR_HEIGHT = 800
+
+# Create sensation map
+avatar_sensation_map = create_sensation_map(AVATAR_WIDTH, AVATAR_HEIGHT)
 
-# Constants
-AVATAR_WIDTH, AVATAR_HEIGHT = 600, 800
 # Create heatmap function
 def create_heatmap(sensation_map):
     overall_sensitivity = np.mean(sensation_map, axis=2)
@@ -350,18 +353,7 @@ def create_heatmap(sensation_map):
     buf.seek(0)
     heatmap_img = Image.open(buf)
     plt.close(fig)
-    return heatmap_img
-
-# Create avatar with heatmap function
-def create_avatar_with_heatmap():
-    avatar_img = create_avatar()
-    heatmap_img = create_heatmap(avatar_sensation_map)
-    heatmap_img = heatmap_img.resize((AVATAR_WIDTH, AVATAR_HEIGHT))
-    combined_img = Image.alpha_composite(avatar_img.convert('RGBA'), heatmap_img.convert('RGBA'))
-    return combined_img
 
-# Create the avatar with heatmap
-avatar_with_heatmap = create_avatar_with_heatmap()
 # Your Streamlit app code goes here
 st.title("NeuraSense AI - Cyberpunk Edition")
 
@@ -512,6 +504,7 @@ def create_avatar():
        draw.line([start, end], fill=(0, 255, 255, 50), width=1)
   
    return img
+
 # Create avatar with heatmap function
 def create_avatar_with_heatmap():
     avatar_img = create_avatar()
@@ -524,7 +517,6 @@ def create_avatar_with_heatmap():
 avatar_with_heatmap = create_avatar_with_heatmap()
 
 
-
 # Streamlit app
 st.title("NeuraSense AI: Advanced Humanoid Techno-Sensory Simulation")
 
@@ -539,8 +531,8 @@ col1, col2 = st.columns([2, 1])
 with col1:
    st.subheader("Humanoid Avatar Interface")
   
-  # Use st_canvas for touch input
-canvas_result = st_canvas(
+ # Use st_canvas for touch input
+    canvas_result = st_canvas(
         fill_color="rgba(0, 255, 255, 0.3)",
         stroke_width=2,
         stroke_color="#00FFFF",