Update app.py

app.py

@@ -108,16 +108,6 @@ if st.session_state["layout_image"] is None:
         st.session_state["layout_image"] = layout_image
         st.session_state["canvas"] = canvas
 
-# Display map with scroll and zoom controls
-if st.session_state["layout_image"] is not None:
-    st.image(
-        st.session_state["layout_image"],
-        caption="Generated Dungeon Map Layout",
-        use_container_width=True,
-        output_format="PNG",
-        clamp=True,
-    )
-
 # Sidebar Controls
 st.sidebar.title("Map Controls")
 if st.sidebar.button("💾 Save Map"):
@@ -134,14 +124,29 @@ if st.sidebar.button("🗺️ Regenerate Map"):
         st.session_state["canvas"] = canvas
         st.rerun()
 
-# Zoom controls
+# Zoom Controls
 st.sidebar.title("Zoom")
+show_zoomed = st.sidebar.checkbox("Show Zoomed Version")
 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),
-)
+
+if show_zoomed and 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.Resampling.LANCZOS,
+    )
+    buffer = BytesIO()
+    zoomed_canvas.save(buffer, format="PNG")
+    buffer.seek(0)
+    st.image(
+        buffer,
+        caption=f"Zoomed Dungeon Map Layout (Zoom Level: {zoom_level}x)",
+        use_container_width=False,
+        output_format="PNG",
+    )
+else:
+    st.image(
+        st.session_state["layout_image"],
+        caption="Generated Dungeon Map Layout",
+        use_container_width=True,
+        output_format="PNG",
+    )