Update app.py

app.py

@@ -3,9 +3,11 @@ import numpy as np
 from typing import List, Tuple
 from transformers import pipeline
 import matplotlib.pyplot as plt
-from matplotlib.backends.backend_agg import RendererAgg
+from matplotlib.backends.backend_agg import FigureCanvasAgg
 from streamlit_drawable_canvas import st_canvas
 import time
+from PIL import Image
+import io
 
 # Constants
 WIDTH, HEIGHT = 600, 600
@@ -63,13 +65,21 @@ fig, ax = plt.subplots(figsize=(6, 6))
 ax.imshow(sensation_map, cmap='coolwarm', extent=[0, WIDTH, HEIGHT, 0])
 ax.axis('off')
 
+# Convert matplotlib figure to Image
+canvas = FigureCanvasAgg(fig)
+canvas.draw()
+buf = io.BytesIO()
+plt.savefig(buf, format='png')
+buf.seek(0)
+img = Image.open(buf)
+
 # Use streamlit-drawable-canvas for interaction
 canvas_result = st_canvas(
     fill_color="rgba(255, 165, 0, 0.3)",
     stroke_width=3,
     stroke_color="#e00",
     background_color="#eee",
-    background_image=fig,
+    background_image=img,
     update_streamlit=True,
     height=HEIGHT,
     width=WIDTH,