Update app.py

app.py

@@ -468,12 +468,9 @@ def create_heatmap(sensation_map, sensation_type):
     buf = io.BytesIO()
     plt.savefig(buf, format='png')
     buf.seek(0)
-    
-    data = base64.b64encode(buf.getvalue()).decode('utf-8')
-    
     plt.close()
     
-    return f'data:image/png;base64,{data}'
+    return Image.open(buf)
 
 def generate_ai_response(keypoints, sensation_map):
     num_keypoints = len(keypoints)
@@ -580,13 +577,6 @@ 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()
-    heatmap_img = create_heatmap(avatar_sensation_map)
-    heatmap_img = heatmap_img.resize((AVATAR_WIDTH, AVATAR_HEIGHT))
-    heatmap_img = heatmap_img.convert('RGBA')
     
     # Adjust alpha channel of heatmap
     data = np.array(heatmap_img)