Update app.py

app.py

@@ -108,6 +108,16 @@ 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"):
@@ -124,29 +134,14 @@ 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)
-
-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",
-    )
+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),
+)