Update app.py

app.py

@@ -3,7 +3,7 @@ import matplotlib.pyplot as plt
 import numpy as np
 
 # Function to draw the plot
-def draw_plot(length, width):
+def draw_plot(length, width, unit):
     fig, ax = plt.subplots(figsize=(8, 8))
 
     # Plotting the plot as a rectangle
@@ -13,9 +13,9 @@ def draw_plot(length, width):
     ax.plot([length, length], [0, width], color="blue")  # Right side
 
     # Adding labels and title
-    ax.set_title(f"Plot: Length = {length} units, Width = {width} units")
-    ax.set_xlabel("Length (units)")
-    ax.set_ylabel("Width (units)")
+    ax.set_title(f"Plot: Length = {length} {unit}, Width = {width} {unit}")
+    ax.set_xlabel(f"Length ({unit})")
+    ax.set_ylabel(f"Width ({unit})")
 
     # Set the limits
     ax.set_xlim(0, length + 10)
@@ -29,13 +29,16 @@ def draw_plot(length, width):
 # Streamlit interface
 st.title("Plot Drawing App")
 
-# User input for plot dimensions
-length = st.number_input("Enter the Length of the Plot (in units)", min_value=1, max_value=1000, value=50)
-width = st.number_input("Enter the Width of the Plot (in units)", min_value=1, max_value=1000, value=30)
+# 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)
+
+# Dropdown menu for unit selection
+unit = st.selectbox("Select Unit", ["ft", "m", "cm"])
 
 # Draw the plot when inputs are provided
 if length and width:
-    fig = draw_plot(length, width)
+    fig = draw_plot(length, width, unit)
     
     # Display the plot
     st.pyplot(fig)