import streamlit as st
import os
from PIL import Image, ImageDraw
import random

# Function to arrange images dynamically
def arrange_images(image_files, output_path="dungeon_layout.png"):
    if not image_files:
        return None

    # Initialize layout
    positions = []  # Keeps track of image positions (x1, y1, x2, y2)
    canvas_size = (3000, 3000)  # Large canvas for positioning
    canvas = Image.new("RGBA", canvas_size, "white")
    draw = ImageDraw.Draw(canvas)

    def get_center(pos):
        """Calculate center of a bounding box (x1, y1, x2, y2)."""
        return ((pos[0] + pos[2]) // 2, (pos[1] + pos[3]) // 2)

    def does_overlap(new_box, existing_boxes):
        """Check if a new bounding box overlaps any existing boxes."""
        for box in existing_boxes:
            if (
                new_box[0] < box[2]
                and new_box[2] > box[0]
                and new_box[1] < box[3]
                and new_box[3] > box[1]
            ):
                return True
        return False

    # Place the first image at the center of the canvas
    first_img_path = os.path.join(map_dir, image_files[0])
    with Image.open(first_img_path) as img:
        width, height = img.size
        x1 = (canvas_size[0] - width) // 2
        y1 = (canvas_size[1] - height) // 2
        x2, y2 = x1 + width, y1 + height
        positions.append((x1, y1, x2, y2))
        canvas.paste(img, (x1, y1))

    # Place remaining images
    for img_file in image_files[1:]:
        placed = False
        img_path = os.path.join(map_dir, img_file)

        with Image.open(img_path) as img:
            width, height = img.size

            while not placed:
                # Choose a random existing image to connect to
                target_box = random.choice(positions)
                target_center = get_center(target_box)

                # Randomly choose a side of the target image
                side = random.choice(["top", "bottom", "left", "right"])

                # Calculate the new image's position based on the chosen side
                if side == "top":
                    x1 = target_center[0] - width // 2
                    y1 = target_box[1] - height
                elif side == "bottom":
                    x1 = target_center[0] - width // 2
                    y1 = target_box[3]
                elif side == "left":
                    x1 = target_box[0] - width
                    y1 = target_center[1] - height // 2
                elif side == "right":
                    x1 = target_box[2]
                    y1 = target_center[1] - height // 2

                x2, y2 = x1 + width, y1 + height

                # Check for overlap
                if not does_overlap((x1, y1, x2, y2), positions):
                    # Place the image
                    positions.append((x1, y1, x2, y2))
                    canvas.paste(img, (x1, y1))
                    placed = True

                    # Draw a connecting line
                    new_center = get_center((x1, y1, x2, y2))
                    draw.line([target_center, new_center], fill="black", width=5)

    canvas.save(output_path)
    return output_path

# Streamlit App
st.title("Dynamic Dungeon Map Generator")
st.write("Automatically generates dungeon maps by arranging PNG images dynamically.")

# Directory for images
map_dir = "."  # Top-level directory

# Scan the directory for .png files
image_files = [f for f in os.listdir(map_dir) if f.endswith(".png")]

if image_files:
    # Add "Create Map" button
    if st.button("🗺️ Create Map"):
        layout_image = arrange_images(image_files)
        if layout_image:
            st.image(layout_image, caption="Generated Dungeon Map Layout")
        else:
            st.write("Failed to generate a map. Please check the images in the directory.")
else:
    st.write("No PNG files found in the top-level directory.")

# Sidebar for file upload
st.sidebar.title("Options")
uploaded_file = st.sidebar.file_uploader("Upload a PNG file", type="png")

if uploaded_file:
    # Save uploaded file to the directory
    with open(os.path.join(map_dir, uploaded_file.name), "wb") as f:
        f.write(uploaded_file.getbuffer())
    st.sidebar.success("File uploaded successfully! Refresh the app to include it in the dungeon map.")