Update app.py

app.py

@@ -2,7 +2,6 @@ import streamlit as st
 import matplotlib.pyplot as plt
 import matplotlib.patches as patches
 import io
-import random
 
 # Function to convert the figure to bytes for download
 def fig_to_bytes(fig):
@@ -11,27 +10,25 @@ def fig_to_bytes(fig):
     img_bytes.seek(0)
     return img_bytes.read()
 
-# Function to draw the house floor plan with detailed rooms, windows, doors, and walls
+# Function to draw the house floor plan with precise room dimensions and fixed placement
 def draw_house(length, width, unit):
-    fig, ax = plt.subplots(figsize=(10, 10))
-    
+    fig, ax = plt.subplots(figsize=(12, 12))
+
     # 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
 
-    # Define Room Dimensions and Placement (For this example we place 5 rooms)
-    
-    # Bedroom (Fixed position for simplicity, real app should randomize it)
+    # Room 1 - Bedroom
     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
+
+    # Room 2 - Living Room
     living_room_length = 25
     living_room_width = 18
     living_room_x = 5
@@ -39,7 +36,7 @@ def draw_house(length, width, unit):
     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
+    # Room 3 - Kitchen
     kitchen_length = 15
     kitchen_width = 10
     kitchen_x = 30
@@ -47,7 +44,7 @@ def draw_house(length, width, unit):
     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
+    # Room 4 - Bathroom
     bathroom_length = 8
     bathroom_width = 6
     bathroom_x = 30
@@ -55,7 +52,7 @@ def draw_house(length, width, unit):
     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
+    # Room 5 - Garage
     garage_length = 20
     garage_width = 15
     garage_x = 5
@@ -63,17 +60,17 @@ def draw_house(length, width, unit):
     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
+    # Adding doors (as small rectangles on the walls)
     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)
+    # Adding windows (small rectangles on the walls)
     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)
+    # Adding ventilation (small lines representing windows for ventilation)
     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