Update app.py

app.py

@@ -1,9 +1,7 @@
 import streamlit as st
 import os
+from PIL import Image, ImageDraw
 import random
-from PIL import Image
-import networkx as nx
-import matplotlib.pyplot as plt
 
 # Function to calculate aspect ratio
 def calculate_aspect_ratio(image_path):
@@ -13,105 +11,75 @@ def calculate_aspect_ratio(image_path):
         divisor = gcd(width, height)
         return width // divisor, height // divisor
 
-# Function to classify room types based on aspect ratio
-def classify_room(aspect_ratio):
-    if aspect_ratio == (16, 9) or aspect_ratio == (19, 6):
-        return "Hallway"
-    elif aspect_ratio[0] < 8 and aspect_ratio[1] < 8:
-        return "Door"
-    else:
-        return "Room"
-
-# Function to generate a layout
-def generate_dungeon_map(image_files):
-    dungeon = {"Hallways": [], "Doors": [], "Rooms": []}
-
+# Function to randomly position images while ensuring no overlap
+def arrange_images(image_files, output_path="dungeon_layout.png"):
+    if not image_files:
+        return None
+    
+    # Initialize variables for layout
+    positions = []  # Keeps track of image positions
+    canvas_size = (3000, 3000)  # Adjust based on your needs
+    canvas = Image.new("RGBA", canvas_size, "white")
+    draw = ImageDraw.Draw(canvas)
+    
     for img_file in image_files:
         img_path = os.path.join(map_dir, img_file)
-        aspect_ratio = calculate_aspect_ratio(img_path)
-        room_type = classify_room(aspect_ratio)
-
-        if room_type == "Hallway":
-            dungeon["Hallways"].append(img_file)
-        elif room_type == "Door":
-            dungeon["Doors"].append(img_file)
-        else:
-            dungeon["Rooms"].append(img_file)
-
-    # Generate connections between elements
-    layout = nx.Graph()
-    for i, hallway in enumerate(dungeon["Hallways"]):
-        layout.add_node(f"Hallway {i+1}", type="Hallway")
-        if dungeon["Rooms"]:
-            room = dungeon["Rooms"].pop(0)
-            layout.add_node(room, type="Room")
-            layout.add_edge(f"Hallway {i+1}", room)
-
-        if dungeon["Doors"]:
-            door = dungeon["Doors"].pop(0)
-            layout.add_node(door, type="Door")
-            layout.add_edge(f"Hallway {i+1}", door)
-
-    return layout, dungeon
-
-# Function to visualize the dungeon layout
-def visualize_dungeon(layout, output_path="dungeon_tree.png"):
-    plt.figure(figsize=(12, 8))
-    pos = nx.spring_layout(layout)  # Positions for nodes
-    types = nx.get_node_attributes(layout, 'type')
-
-    # Draw nodes with different colors
-    color_map = {
-        "Hallway": "blue",
-        "Door": "green",
-        "Room": "orange"
-    }
-    colors = [color_map[types[node]] for node in layout.nodes]
-
-    nx.draw(
-        layout, pos, with_labels=True, node_color=colors, node_size=2000,
-        font_size=10, font_color="white", font_weight="bold", edge_color="gray"
-    )
-    plt.savefig(output_path)
-    plt.close()
+        with Image.open(img_path) as img:
+            width, height = img.size
+            
+            # Randomly decide horizontal or vertical placement
+            if random.choice(["horizontal", "vertical"]) == "horizontal":
+                x_center = random.randint(width // 2, canvas_size[0] - width // 2)
+                y_center = canvas_size[1] // 2
+            else:
+                x_center = canvas_size[0] // 2
+                y_center = random.randint(height // 2, canvas_size[1] - height // 2)
+            
+            # Calculate position
+            x1 = x_center - width // 2
+            y1 = y_center - height // 2
+            x2 = x1 + width
+            y2 = y1 + height
+            
+            # Ensure no overlap with existing images
+            overlap = any(
+                x1 < pos[2] and x2 > pos[0] and y1 < pos[3] and y2 > pos[1]
+                for pos in positions
+            )
+            
+            if not overlap:
+                positions.append((x1, y1, x2, y2))
+                canvas.paste(img, (x1, y1))
+                
+                # Draw connection line
+                if len(positions) > 1:
+                    prev_x, prev_y = (positions[-2][0] + positions[-2][2]) // 2, (positions[-2][1] + positions[-2][3]) // 2
+                    curr_x, curr_y = (x1 + x2) // 2, (y1 + y2) // 2
+                    draw.line([(prev_x, prev_y), (curr_x, curr_y)], 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 analyzing PNG images in a directory.")
+st.write("Automatically generates dungeon maps by analyzing PNG images in a top-level directory.")
 
 # Directory for images
-map_dir = "maps"  # Replace with the actual directory path
+map_dir = "."  # Top-level directory
 
 # Scan the directory for .png files
-if os.path.exists(map_dir):
-    image_files = [f for f in os.listdir(map_dir) if f.endswith(".png")]
-
-    if image_files:
-        st.subheader("Generate Dungeon Map")
-
-        # Add emoji buttons
-        if st.button("🔄 Regenerate Dungeon"):
-            layout, dungeon = generate_dungeon_map(image_files)
-            visualize_dungeon(layout)
-
-            st.image("dungeon_tree.png", caption="Dungeon Layout Tree")
-
-            # Display classified rooms
-            st.subheader("Classified Rooms")
-            st.write("### Hallways")
-            st.write(", ".join(dungeon["Hallways"]) if dungeon["Hallways"] else "No hallways found.")
-
-            st.write("### Doors")
-            st.write(", ".join(dungeon["Doors"]) if dungeon["Doors"] else "No doors found.")
-
-            st.write("### Rooms")
-            st.write(", ".join(dungeon["Rooms"]) if dungeon["Rooms"] else "No rooms found.")
+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("Click the button above to generate a dungeon map.")
-    else:
-        st.write("No PNG files found in the specified directory.")
+            st.write("Failed to generate a map. Please check the images in the directory.")
 else:
-    st.write("The specified directory does not exist. Please check the path.")
+    st.write("No PNG files found in the top-level directory.")
 
 # Sidebar for file upload
 st.sidebar.title("Options")