Update app.py

app.py

@@ -1,115 +1,110 @@
 import streamlit as st
 import matplotlib.pyplot as plt
+import matplotlib.patches as patches
 import io
 import random
 
-# Convert the plot to bytes for download
+# Function to convert the figure to bytes for download
 def fig_to_bytes(fig):
     img_bytes = io.BytesIO()
     fig.savefig(img_bytes, format="png")
     img_bytes.seek(0)
     return img_bytes.read()
 
-# Function to draw the house floor plan with detailed layout and random room placement
+# Function to draw the house floor plan with detailed rooms, windows, doors, and walls
 def draw_house(length, width, unit):
-    fig, ax = plt.subplots(figsize=(8, 8))
-
-    # Plotting the plot boundary
+    fig, ax = plt.subplots(figsize=(10, 10))
+    
+    # Plot the boundary of the house (Plot limits)
     ax.plot([0, length], [0, 0], color="black", lw=2)  # Bottom side
     ax.plot([0, length], [width, width], color="black", lw=2)  # Top side
     ax.plot([0, 0], [0, width], color="black", lw=2)  # Left side
     ax.plot([length, length], [0, width], color="black", lw=2)  # Right side
 
-    # Randomly place rooms within the plot
-    # Bedroom (Randomly positioned)
-    x1 = random.randint(0, length//2)
-    y1 = random.randint(width//2, width)
-    room_length1 = random.randint(10, 20)
-    room_width1 = random.randint(6, 12)
-    ax.plot([x1, x1 + room_length1], [y1, y1], color="blue", lw=2)
-    ax.plot([x1 + room_length1, x1 + room_length1], [y1, y1 - room_width1], color="blue", lw=2)
-    ax.plot([x1 + room_length1, x1], [y1 - room_width1, y1 - room_width1], color="blue", lw=2)
-    ax.plot([x1, x1], [y1, y1 - room_width1], color="blue", lw=2)
-    ax.text(x1 + room_length1//2, y1 - room_width1//2, "Bedroom 1", fontsize=10, ha="center", color="blue")
-
-    # Drawing Room (Randomly positioned)
-    x2 = random.randint(length//2, length)
-    y2 = random.randint(width//2, width)
-    room_length2 = random.randint(10, 20)
-    room_width2 = random.randint(6, 12)
-    ax.plot([x2, x2 + room_length2], [y2, y2], color="green", lw=2)
-    ax.plot([x2 + room_length2, x2 + room_length2], [y2, y2 - room_width2], color="green", lw=2)
-    ax.plot([x2 + room_length2, x2], [y2 - room_width2, y2 - room_width2], color="green", lw=2)
-    ax.plot([x2, x2], [y2, y2 - room_width2], color="green", lw=2)
-    ax.text(x2 + room_length2//2, y2 - room_width2//2, "Drawing Room", fontsize=10, ha="center", color="green")
-
-    # Kitchen (Randomly positioned)
-    x3 = random.randint(0, length//2)
-    y3 = random.randint(0, width//2)
-    room_length3 = random.randint(8, 15)
-    room_width3 = random.randint(5, 10)
-    ax.plot([x3, x3 + room_length3], [y3, y3], color="red", lw=2)
-    ax.plot([x3 + room_length3, x3 + room_length3], [y3, y3 - room_width3], color="red", lw=2)
-    ax.plot([x3 + room_length3, x3], [y3 - room_width3, y3 - room_width3], color="red", lw=2)
-    ax.plot([x3, x3], [y3, y3 - room_width3], color="red", lw=2)
-    ax.text(x3 + room_length3//2, y3 - room_width3//2, "Kitchen", fontsize=10, ha="center", color="red")
-
-    # Lounge (Randomly positioned)
-    x4 = random.randint(length//2, length)
-    y4 = random.randint(0, width//2)
-    room_length4 = random.randint(10, 18)
-    room_width4 = random.randint(6, 12)
-    ax.plot([x4, x4 + room_length4], [y4, y4], color="purple", lw=2)
-    ax.plot([x4 + room_length4, x4 + room_length4], [y4, y4 - room_width4], color="purple", lw=2)
-    ax.plot([x4 + room_length4, x4], [y4 - room_width4, y4 - room_width4], color="purple", lw=2)
-    ax.plot([x4, x4], [y4, y4 - room_width4], color="purple", lw=2)
-    ax.text(x4 + room_length4//2, y4 - room_width4//2, "Lounge", fontsize=10, ha="center", color="purple")
-
-    # Car Parking (Randomly positioned)
-    x5 = random.randint(length//4, length//2)
-    y5 = random.randint(0, width//4)
-    room_length5 = random.randint(12, 20)
-    room_width5 = random.randint(8, 15)
-    ax.plot([x5, x5 + room_length5], [y5, y5], color="orange", lw=2)
-    ax.plot([x5 + room_length5, x5 + room_length5], [y5, y5 - room_width5], color="orange", lw=2)
-    ax.plot([x5 + room_length5, x5], [y5 - room_width5, y5 - room_width5], color="orange", lw=2)
-    ax.plot([x5, x5], [y5, y5 - room_width5], color="orange", lw=2)
-    ax.text(x5 + room_length5//2, y5 - room_width5//2, "Car Parking", fontsize=10, ha="center", color="orange")
-
-    # Servant Room (Randomly positioned)
-    x6 = random.randint(0, length//4)
-    y6 = random.randint(0, width//4)
-    room_length6 = random.randint(6, 10)
-    room_width6 = random.randint(4, 8)
-    ax.plot([x6, x6 + room_length6], [y6, y6], color="brown", lw=2)
-    ax.plot([x6 + room_length6, x6 + room_length6], [y6, y6 - room_width6], color="brown", lw=2)
-    ax.plot([x6 + room_length6, x6], [y6 - room_width6, y6 - room_width6], color="brown", lw=2)
-    ax.plot([x6, x6], [y6, y6 - room_width6], color="brown", lw=2)
-    ax.text(x6 + room_length6//2, y6 - room_width6//2, "Servant Room", fontsize=10, ha="center", color="brown")
-
-    # Ventilation (Randomly placed)
-    x7 = random.randint(length//2, length)
-    y7 = random.randint(width//2, width)
-    ax.plot([x7, x7 + 4], [y7, y7], color="grey", lw=2)
-    ax.text(x7 + 2, y7, "Ventilation", fontsize=8, ha="center", color="grey")
-
-    # Adding Labels and Title
-    ax.set_title(f"Random House Layout: Length = {length} {unit}, Width = {width} {unit}")
+    # Define Room Dimensions and Placement (For this example we place 5 rooms)
+    
+    # Bedroom (Fixed position for simplicity, real app should randomize it)
+    bedroom_length = 20
+    bedroom_width = 15
+    bedroom_x = 5
+    bedroom_y = 15
+    ax.add_patch(patches.Rectangle((bedroom_x, bedroom_y), bedroom_length, bedroom_width, linewidth=2, edgecolor="blue", facecolor="lightblue"))
+    ax.text(bedroom_x + bedroom_length/2, bedroom_y + bedroom_width/2, "Bedroom", fontsize=12, ha="center", va="center", color="black")
+    
+    # Living Room
+    living_room_length = 25
+    living_room_width = 18
+    living_room_x = 5
+    living_room_y = 0
+    ax.add_patch(patches.Rectangle((living_room_x, living_room_y), living_room_length, living_room_width, linewidth=2, edgecolor="green", facecolor="lightgreen"))
+    ax.text(living_room_x + living_room_length/2, living_room_y + living_room_width/2, "Living Room", fontsize=12, ha="center", va="center", color="black")
+
+    # Kitchen
+    kitchen_length = 15
+    kitchen_width = 10
+    kitchen_x = 30
+    kitchen_y = 0
+    ax.add_patch(patches.Rectangle((kitchen_x, kitchen_y), kitchen_length, kitchen_width, linewidth=2, edgecolor="red", facecolor="lightcoral"))
+    ax.text(kitchen_x + kitchen_length/2, kitchen_y + kitchen_width/2, "Kitchen", fontsize=12, ha="center", va="center", color="black")
+
+    # Bathroom
+    bathroom_length = 8
+    bathroom_width = 6
+    bathroom_x = 30
+    bathroom_y = 12
+    ax.add_patch(patches.Rectangle((bathroom_x, bathroom_y), bathroom_length, bathroom_width, linewidth=2, edgecolor="purple", facecolor="violet"))
+    ax.text(bathroom_x + bathroom_length/2, bathroom_y + bathroom_width/2, "Bathroom", fontsize=12, ha="center", va="center", color="black")
+
+    # Garage
+    garage_length = 20
+    garage_width = 15
+    garage_x = 5
+    garage_y = 35
+    ax.add_patch(patches.Rectangle((garage_x, garage_y), garage_length, garage_width, linewidth=2, edgecolor="orange", facecolor="orange"))
+    ax.text(garage_x + garage_length/2, garage_y + garage_width/2, "Garage", fontsize=12, ha="center", va="center", color="black")
+
+    # Drawing some doors as small rectangles
+    ax.add_patch(patches.Rectangle((bedroom_x + 10, bedroom_y), 3, 1, linewidth=2, edgecolor="black", facecolor="brown"))  # Bedroom Door
+    ax.add_patch(patches.Rectangle((living_room_x + 10, living_room_y), 3, 1, linewidth=2, edgecolor="black", facecolor="brown"))  # Living Room Door
+    ax.add_patch(patches.Rectangle((kitchen_x + 5, kitchen_y), 3, 1, linewidth=2, edgecolor="black", facecolor="brown"))  # Kitchen Door
+    ax.add_patch(patches.Rectangle((garage_x + 10, garage_y), 3, 1, linewidth=2, edgecolor="black", facecolor="brown"))  # Garage Door
+
+    # Adding window details (small rectangles representing windows)
+    ax.add_patch(patches.Rectangle((bedroom_x + 5, bedroom_y + 15), 5, 1, linewidth=2, edgecolor="blue", facecolor="lightblue"))  # Bedroom window
+    ax.add_patch(patches.Rectangle((living_room_x + 5, living_room_y + 18), 5, 1, linewidth=2, edgecolor="green", facecolor="lightgreen"))  # Living room window
+
+    # Adding Ventilation (Small lines for windows)
+    ax.plot([living_room_x + 20, living_room_x + 22], [living_room_y + 5, living_room_y + 5], color="black", lw=2)  # Ventilation Line
+
+    # Add title and labels
+    ax.set_title(f"House Floor Plan: Length = {length} {unit}, Width = {width} {unit}", fontsize=14)
     ax.set_xlabel(f"Length ({unit})")
     ax.set_ylabel(f"Width ({unit})")
 
-    # Set the limits
+    # Set limits and remove axes for clarity
     ax.set_xlim(0, length + 10)
     ax.set_ylim(0, width + 10)
-
-    # Hide the axes for better visual
     ax.axis('off')
 
     return fig
 
 # Streamlit interface
-st.title("Random House Layout Drawing App")
+st.title("AutoCAD-style House Layout Drawing")
 
 # User input for plot dimensions and units
 length = st.number_input("Enter the Length of the Plot", min_value=1, max_value=1000, value=50)
-width = st.number_input("Enter the Width of
+width = st.number_input("Enter the Width of the Plot", min_value=1, max_value=1000, value=30)
+
+# Dropdown menu for unit selection
+unit = st.selectbox("Select Unit", ["ft", "m", "cm"])
+
+# Draw the house layout when inputs are provided
+if length and width:
+    fig = draw_house(length, width, unit)
+    
+    # Display the plot
+    st.pyplot(fig)
+
+    # Option to download the floor plan as a PNG image
+    img_data = fig_to_bytes(fig)
+    st.download_button("Download Floor Plan", img_data, file_name="house_layout.png", mime="image/png")