Update app.py

app.py

@@ -1,107 +1,40 @@
 import streamlit as st
 import matplotlib.pyplot as plt
-import matplotlib.patches as patches
-import io
 
-# 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 precise room dimensions and fixed placement
-def draw_house(length, width, unit):
-    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
-
-    # 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")
-
-    # Room 2 - 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")
-
-    # Room 3 - 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")
-
-    # Room 4 - 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")
-
-    # Room 5 - 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")
-
-    # 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 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 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
-    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 limits and remove axes for clarity
-    ax.set_xlim(0, length + 10)
-    ax.set_ylim(0, width + 10)
-    ax.axis('off')
-
-    return fig
-
-# Streamlit interface
-st.title("AutoCAD-style House Layout Drawing")
+def convert_to_meters(value, unit):
+    conversion_factors = {
+        "m": 1,
+        "ft": 0.3048,
+        "cm": 0.01,
+        "in": 0.0254
+    }
+    return value * conversion_factors[unit]
+
+def draw_floor_plan(length, width):
+    fig, ax = plt.subplots(figsize=(8, 8))
+    ax.plot([0, length, length, 0, 0], [0, 0, width, width, 0], marker='o', color='blue')
+    ax.set_xlim(-1, length + 1)
+    ax.set_ylim(-1, width + 1)
+    ax.set_aspect('equal', adjustable='box')
+    ax.set_title("Floor Plan")
+    ax.set_xlabel("Length (m)")
+    ax.set_ylabel("Width (m)")
+    st.pyplot(fig)
 
-# 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 the Plot", min_value=1, max_value=1000, value=30)
+st.title("Floor Plan Generator")
 
-# Dropdown menu for unit selection
-unit = st.selectbox("Select Unit", ["ft", "m", "cm"])
+st.write("Enter the dimensions of the floor plan and select the unit of measurement.")
 
-# Draw the house layout when inputs are provided
-if length and width:
-    fig = draw_house(length, width, unit)
-    
-    # Display the plot
-    st.pyplot(fig)
+# Input fields
+length = st.number_input("Enter the length:", min_value=0.0, format="%.2f")
+width = st.number_input("Enter the width:", min_value=0.0, format="%.2f")
+unit = st.selectbox("Select the unit:", ["m", "ft", "cm", "in"], index=0)
 
-    # 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")
+if st.button("Generate Floor Plan"):
+    if length > 0 and width > 0:
+        length_m = convert_to_meters(length, unit)
+        width_m = convert_to_meters(width, unit)
+        st.success(f"Dimensions in meters: Length = {length_m:.2f} m, Width = {width_m:.2f} m")
+        draw_floor_plan(length_m, width_m)
+    else:
+        st.error("Please enter positive values for length and width.")