Update app.py

app.py

@@ -339,7 +339,29 @@ st.markdown(custom_css, unsafe_allow_html=True)
 
 # Constants
 AVATAR_WIDTH, AVATAR_HEIGHT = 600, 800
+# Create heatmap function
+def create_heatmap(sensation_map):
+    overall_sensitivity = np.mean(sensation_map, axis=2)
+    fig, ax = plt.subplots(figsize=(10, 15))
+    sns.heatmap(overall_sensitivity, cmap='YlOrRd', alpha=0.7, cbar=False, ax=ax)
+    ax.set_axis_off()
+    buf = BytesIO()
+    plt.savefig(buf, format='png', bbox_inches='tight', pad_inches=0)
+    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")
 
@@ -490,21 +512,7 @@ def create_avatar():
        draw.line([start, end], fill=(0, 255, 255, 50), width=1)
   
    return img
-
-# New function to create heatmap
-def create_heatmap(sensation_map):
-    overall_sensitivity = np.mean(sensation_map, axis=2)
-    fig, ax = plt.subplots(figsize=(10, 15))
-    sns.heatmap(overall_sensitivity, cmap='YlOrRd', alpha=0.7, cbar=False, ax=ax)
-    ax.set_axis_off()
-    buf = BytesIO()
-    plt.savefig(buf, format='png', bbox_inches='tight', pad_inches=0)
-    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)
@@ -512,6 +520,9 @@ def create_avatar_with_heatmap():
     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()
+
 
 
 # Streamlit app