Update app.py

app.py

@@ -15,7 +15,7 @@ VALID_USERS = {
 scope = ["https://spreadsheets.google.com/feeds", "https://www.googleapis.com/auth/drive"]
 creds = ServiceAccountCredentials.from_json_keyfile_name("credentials.json", scope)
 client = gspread.authorize(creds)
-sheet_file = client.open("userAccess")
+sheet_file = client.open("userAccess")  # Ensure this is the exact sheet name in Google Sheets
 
 # ------------------ HELPERS ------------------
 def load_tab(sheet_name):
@@ -42,13 +42,19 @@ def generate_summary(date_str):
     day_df = df[df['DateStr'] == date_str]
 
     total_visits = day_df.groupby("Rep").size().reset_index(name="Total Visits")
-    current = day_df[day_df["Current/Prospect Custor"] == "Current"]
-    prospect = day_df[day_df["Current/Prospect Custor"] == "Prospect"]
+
+    curr_col = "Current/Prospect Custor"
+    if curr_col not in df.columns:
+        df[curr_col] = ""
+
+    current = day_df[day_df[curr_col] == "Current"]
+    prospect = day_df[day_df[curr_col] == "Prospect"]
     breakdown = pd.DataFrame({
         "Rep": all_reps,
         "Current": [len(current[current["Rep"] == rep]) for rep in all_reps],
         "Prospect": [len(prospect[prospect["Rep"] == rep]) for rep in all_reps]
     })
+
     active_list = total_visits['Rep'].tolist()
     inactive_list = [rep for rep in all_reps if rep not in active_list]
     inactive_df = pd.DataFrame({'Inactive Reps': inactive_list})
@@ -57,6 +63,10 @@ def generate_summary(date_str):
 def get_order_summary(date_str):
     df = load_field_sales()
     day_df = df[df['DateStr'] == date_str]
+
+    if "Order Received" not in df.columns:
+        df["Order Received"] = ""
+
     rep_group = day_df.groupby("Rep").agg({
         "Order Received": lambda x: (x == "Yes").sum(),
         "Order Value": "sum"
@@ -68,8 +78,12 @@ def get_order_summary(date_str):
 
 def get_escalations():
     df = load_field_sales()
-    flagged = df[df["Customer Type"].str.contains("Second", na=False)]
-    return flagged if not flagged.empty else pd.DataFrame([["No second-hand dealerships flagged."]], columns=["Status"])
+    col = "Customer Type & Status"
+    if col in df.columns:
+        flagged = df[df[col].str.contains("Second", na=False)]
+        return flagged if not flagged.empty else pd.DataFrame([["No second-hand dealerships flagged."]], columns=["Status"])
+    else:
+        return pd.DataFrame([["⚠️ Column 'Customer Type & Status' not found."]], columns=["Status"])
 
 # ------------------ TELESALeS ------------------
 def get_telesales_summary():
@@ -110,7 +124,6 @@ with gr.Blocks() as app:
         with gr.Column(visible=False) as main_ui:
             gr.Markdown("## 🗂️ CarMat Dashboard")
 
-            # preload dates for field sales
             df_initial = load_field_sales()
             unique_dates = sorted(df_initial["DateStr"].unique(), reverse=True)