Update app.py

app.py

@@ -1,59 +1,57 @@
 import streamlit as st
-import random
-from PIL import Image, ImageDraw
+import matplotlib.pyplot as plt
+import matplotlib.patches as patches
 
-def generate_floor_plan(plot_length, plot_width, num_bedrooms, num_bathrooms, num_drawing_rooms, num_dining_rooms, num_kitchens, num_porch, num_store, num_entrance):
-    # Create a blank floor plan image
-    image_width, image_height = plot_width * 10, plot_length * 10
-    img = Image.new('RGB', (image_width, image_height), 'white')
-    draw = ImageDraw.Draw(img)
+def generate_floor_plan(plot_length, plot_width, rooms):
+    fig, ax = plt.subplots(figsize=(10, 8))
 
-    # Placeholder: Divide the plot into simple rectangles for each room
-    rooms = {
-        'Bedroom': num_bedrooms,
-        'Bathroom': num_bathrooms,
-        'Drawing Room': num_drawing_rooms,
-        'Dining Room': num_dining_rooms,
-        'Kitchen': num_kitchens,
-        'Porch': num_porch,
-        'Store': num_store,
-        'Entrance': num_entrance
-    }
+    # Plot dimensions
+    ax.set_xlim(0, plot_length)
+    ax.set_ylim(0, plot_width)
+    ax.set_aspect("equal")
 
+    # Draw boundary
+    ax.add_patch(patches.Rectangle((0, 0), plot_length, plot_width, edgecolor="black", facecolor="none", linewidth=2))
+
+    # Layout rooms as simple rectangles
     x, y = 0, 0
-    room_width, room_height = image_width // 4, image_height // 4
-    for room_type, count in rooms.items():
-        for _ in range(count):
-            draw.rectangle([x, y, x + room_width, y + room_height], outline="black", fill=random.choice(["#f0e68c", "#add8e6", "#ffcccb", "#98fb98"]))
-            draw.text((x + 10, y + 10), room_type, fill="black")
-            x += room_width
-            if x >= image_width:
+    room_width, room_height = plot_length / 4, plot_width / len(rooms)
+    for room_name, room_count in rooms.items():
+        for _ in range(room_count):
+            if x + room_width > plot_length:  # Move to the next row
                 x = 0
                 y += room_height
+            ax.add_patch(patches.Rectangle((x, y), room_width, room_height, edgecolor="blue", facecolor="lightblue", linewidth=1))
+            ax.text(x + room_width / 2, y + room_height / 2, room_name, ha="center", va="center", fontsize=10)
+            x += room_width
 
-    return img
+    st.pyplot(fig)
 
+# Streamlit UI
 st.title("Floor Plan Generator")
-
-st.sidebar.header("Input Plot Details")
-plot_length = st.sidebar.number_input("Length of the plot (in feet):", min_value=10, max_value=100, value=50)
-plot_width = st.sidebar.number_input("Width of the plot (in feet):", min_value=10, max_value=100, value=30)
-
-st.sidebar.header("Input Room Details")
-num_bedrooms = st.sidebar.number_input("Number of Bedrooms:", min_value=0, max_value=10, value=2)
-num_bathrooms = st.sidebar.number_input("Number of Bathrooms:", min_value=0, max_value=10, value=2)
-num_drawing_rooms = st.sidebar.number_input("Number of Drawing Rooms:", min_value=0, max_value=5, value=1)
-num_dining_rooms = st.sidebar.number_input("Number of Dining Rooms:", min_value=0, max_value=5, value=1)
-num_kitchens = st.sidebar.number_input("Number of Kitchens:", min_value=0, max_value=3, value=1)
-num_porch = st.sidebar.number_input("Number of Porches:", min_value=0, max_value=2, value=1)
-num_store = st.sidebar.number_input("Number of Stores:", min_value=0, max_value=3, value=1)
-num_entrance = st.sidebar.number_input("Number of Entrance Areas:", min_value=0, max_value=2, value=1)
-
-if st.button("Generate Floor Plan"):
-    st.write("### Generated Floor Plan")
-    floor_plan_image = generate_floor_plan(
-        plot_length, plot_width, num_bedrooms, num_bathrooms,
-        num_drawing_rooms, num_dining_rooms, num_kitchens,
-        num_porch, num_store, num_entrance
-    )
-    st.image(floor_plan_image, caption="Floor Plan", use_column_width=True)
+st.sidebar.header("Plot Dimensions")
+plot_length = st.sidebar.number_input("Plot Length (in feet)", min_value=10, max_value=200, value=50, step=1)
+plot_width = st.sidebar.number_input("Plot Width (in feet)", min_value=10, max_value=200, value=40, step=1)
+
+st.sidebar.header("Rooms Configuration")
+bedrooms = st.sidebar.number_input("Number of Bedrooms", min_value=0, max_value=10, value=2, step=1)
+bathrooms = st.sidebar.number_input("Number of Bathrooms", min_value=0, max_value=10, value=1, step=1)
+drawing_rooms = st.sidebar.number_input("Number of Drawing Rooms", min_value=0, max_value=5, value=1, step=1)
+dining_rooms = st.sidebar.number_input("Number of Dining Rooms", min_value=0, max_value=5, value=1, step=1)
+kitchens = st.sidebar.number_input("Number of Kitchens", min_value=0, max_value=3, value=1, step=1)
+porches = st.sidebar.number_input("Number of Porches", min_value=0, max_value=3, value=1, step=1)
+stores = st.sidebar.number_input("Number of Stores", min_value=0, max_value=3, value=1, step=1)
+entrance_areas = st.sidebar.number_input("Number of Entrance Areas", min_value=0, max_value=3, value=1, step=1)
+
+if st.sidebar.button("Generate Floor Plan"):
+    rooms = {
+        "Bedroom": bedrooms,
+        "Bathroom": bathrooms,
+        "Drawing Room": drawing_rooms,
+        "Dining Room": dining_rooms,
+        "Kitchen": kitchens,
+        "Porch": porches,
+        "Store": stores,
+        "Entrance Area": entrance_areas,
+    }
+    generate_floor_plan(plot_length, plot_width, rooms)