Update app.py

app.py

@@ -17,15 +17,19 @@ client = gspread.authorize(creds)
 sheet_url = "https://docs.google.com/spreadsheets/d/1if4KoVQvw5ZbhknfdZbzMkcTiPfsD6bz9V3a1th-bwQ"
 
 # -------------------- UTILS --------------------
-def normalize_columns(df: pd.DataFrame) -> pd.DataFrame:
-    df.columns = df.columns.str.strip().str.title()
-    return df
+def normalize_header(raw_header):
+    # strip and titleize
+    return [h.strip().title() for h in raw_header]
 
 def load_sheet(sheet_name: str) -> pd.DataFrame:
-    records = client.open_by_url(sheet_url) \
-                    .worksheet(sheet_name) \
-                    .get_all_records()
-    return normalize_columns(pd.DataFrame(records))
+    ws = client.open_by_url(sheet_url).worksheet(sheet_name)
+    all_vals = ws.get_all_values()
+    if not all_vals or len(all_vals) < 2:
+        return pd.DataFrame()
+    header = normalize_header(all_vals[0])
+    rows = all_vals[1:]
+    df = pd.DataFrame(rows, columns=header)
+    return df
 
 def get_current_week_range():
     today = datetime.now()
@@ -36,7 +40,7 @@ def get_current_week_range():
 # -------------------- CALLS --------------------
 def get_calls(rep=None):
     df = load_sheet("Calls")
-    if "Call Date" not in df.columns:
+    if "Call Date" not in df:
         return pd.DataFrame([{"Error": "Missing 'Call Date' column"}])
     df["Call Date"] = pd.to_datetime(df["Call Date"], errors="coerce").dt.date
     start, end = get_current_week_range()
@@ -47,7 +51,7 @@ def get_calls(rep=None):
 
 def search_calls_by_date(y, m, d, rep):
     df = load_sheet("Calls")
-    if "Call Date" not in df.columns:
+    if "Call Date" not in df:
         return pd.DataFrame([{"Error": "Missing 'Call Date' column"}])
     try:
         target = datetime(int(y), int(m), int(d)).date()
@@ -62,7 +66,7 @@ def search_calls_by_date(y, m, d, rep):
 # -------------------- APPOINTMENTS --------------------
 def appointments_detail(rep=None):
     df = load_sheet("Appointments")
-    if "Appointment Date" not in df.columns:
+    if "Appointment Date" not in df:
         return pd.DataFrame([{"Error": "Missing 'Appointment Date' column"}])
     df["Appointment Date"] = pd.to_datetime(df["Appointment Date"], errors="coerce").dt.date
     start, end = get_current_week_range()
@@ -81,7 +85,7 @@ def appointments_summary(rep=None):
 
 def search_appointments_by_date(y, m, d, rep):
     df = load_sheet("Appointments")
-    if "Appointment Date" not in df.columns:
+    if "Appointment Date" not in df:
         return pd.DataFrame([{"Error": "Missing 'Appointment Date' column"}])
     try:
         target = datetime(int(y), int(m), int(d)).date()
@@ -96,8 +100,7 @@ def search_appointments_by_date(y, m, d, rep):
 # -------------------- LEADS --------------------
 def get_leads_detail():
     df = load_sheet("AllocatedLeads")
-    df = df.rename(columns={"Assigned Rep": "Assigned Rep"})
-    if "Assigned Rep" not in df.columns or "Company Name" not in df.columns:
+    if "Assigned Rep" not in df or "Company Name" not in df:
         return pd.DataFrame([{"Error": "Missing 'Assigned Rep' or 'Company Name' column"}])
     return df
 
@@ -189,9 +192,9 @@ with gr.Blocks(title="Graffiti Admin Dashboard") as app:
 
     # Appointed Leads
     with gr.Tab("Appointed Leads"):
-        leads_btn    = gr.Button("View Appointed Leads")
-        leads_sum    = gr.Dataframe(label="📊 Leads Count by Rep")
-        leads_det    = gr.Dataframe(label="🔎 Detailed Leads")
+        leads_btn = gr.Button("View Appointed Leads")
+        leads_sum = gr.Dataframe(label="📊 Leads Count by Rep")
+        leads_det = gr.Dataframe(label="🔎 Detailed Leads")
         leads_btn.click(
             fn=lambda: (get_leads_summary(), get_leads_detail()),
             outputs=[leads_sum, leads_det]