Update app.py

app.py

@@ -137,11 +137,20 @@ if st.sidebar.button("🗺️ Regenerate Map"):
 # Zoom controls
 st.sidebar.title("Zoom")
 zoom_level = st.sidebar.slider("Zoom Level", min_value=0.5, max_value=2.0, value=1.0, step=0.1)
-st.image(
-    st.session_state["layout_image"],
-    caption="Zoomed Dungeon Map Layout",
-    use_container_width=False,
-    output_format="PNG",
-    clamp=True,
-    width=int(canvas_size * zoom_level),
-)
+
+# Resize the canvas for zoom functionality
+if st.session_state["canvas"] is not None:
+    zoomed_canvas = st.session_state["canvas"].resize(
+        (int(canvas_size * zoom_level), int(canvas_size * zoom_level)),
+        resample=Image.ANTIALIAS
+    )
+    buffer = BytesIO()
+    zoomed_canvas.save(buffer, format="PNG")
+    buffer.seek(0)
+
+    st.image(
+        buffer,
+        caption=f"Zoomed Dungeon Map Layout (Zoom: {zoom_level}x)",
+        use_container_width=False,
+        output_format="PNG",
+    )