Update app.py

app.py

@@ -5,6 +5,9 @@ import pandas as pd
 data_file = "Estimate Refer Database.xlsx"
 data = pd.read_excel(data_file, sheet_name=0)
 
+# Ensure column names are consistent
+data.columns = data.columns.str.strip()
+
 # Define the cities and project types
 cities = [
     "Vancouver", "Calgary", "Edmonton", "Winnipeg", "GTA", "Ottawa/Gatineau", "Montreal", "Halifax", "St. John’s"
@@ -39,6 +42,13 @@ def calculate_estimate(city, project_type, subtype, square_footage):
     if not city or not project_type or not subtype or not square_footage:
         return "Please complete all selections to get an estimate."
 
+    # Debugging: Print the columns to verify names
+    st.write("Columns in the dataset:", data.columns)
+
+    # Check if required columns exist
+    if "City" not in data.columns or "Subtype" not in data.columns:
+        return "Required columns 'City' or 'Subtype' are missing in the dataset."
+
     # Filter the data for the selected city and subtype
     filtered_data = data[(data["City"] == city) & (data["Subtype"] == subtype)]
 
@@ -75,6 +85,3 @@ estimate_button = st.button("Get Estimate")
 if estimate_button:
     result = calculate_estimate(city, project_type, subtype, square_footage)
     st.write(result)
-
-
-