Update app.py

app.py

@@ -484,8 +484,15 @@ def generate_ai_response(keypoints, sensation_map):
     
     return response
 
-# Streamlit app
-st.title("NeuraSense AI - Humanoid Touch Point Detection")
+# Create and display avatar with heatmap
+st.subheader("Avatar with Sensation Heatmap")
+
+# You need to define sensation_map and sensation_type before this
+sensation_map = np.random.rand(AVATAR_HEIGHT 600, AVATAR_WIDTH, 300)  # Example random sensation map
+sensation_type = 0  # Example sensation type (0 for Pain)
+
+avatar_with_heatmap = create_avatar_with_heatmap(sensation_map, sensation_type)
+st.image(avatar_with_heatmap, use_column_width=True)
 
 uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])