Update app.py

app.py

@@ -16,7 +16,6 @@ creds = ServiceAccountCredentials.from_json_keyfile_name(
 )
 client = gspread.authorize(creds)
 
-# YOUR SPREADSHEET URL
 SHEET_URL = "https://docs.google.com/spreadsheets/d/1if4KoVQvw5ZbhknfdZbzMkcTiPfsD6bz9V3a1th-bwQ"
 
 # -------------------- UTILS --------------------
@@ -29,7 +28,6 @@ def load_sheet_df(name):
     if not data:
         return pd.DataFrame()
     raw_header, *rows = data
-    # make header unique
     counts = Counter()
     header = []
     for col in raw_header:
@@ -167,9 +165,17 @@ def quote_year_choices():
 
 def quote_month_choices(year=None):
     df = get_quotes_df()
-    if year and "Year" in df.columns and "Month" in df.columns:
-        months = df[df["Year"].astype(str) == str(year)]["Month"].dropna().unique()
-        months = [str(int(m)) for m in sorted(pd.to_numeric(months, errors="coerce").dropna().unique())]
+    # Always return a list (possibly empty)
+    if (
+        year 
+        and "Year" in df.columns 
+        and "Month" in df.columns
+        and not df.empty
+    ):
+        subset = df[df["Year"].astype(str) == str(year)]
+        months = pd.to_numeric(subset["Month"], errors="coerce").dropna().astype(int).unique()
+        # Keep months only 1-12
+        months = sorted(set(str(m) for m in months if 1 <= m <= 12))
         return months
     return []
 
@@ -259,10 +265,8 @@ with gr.Blocks(title="Graffiti Admin Dashboard") as app:
 
         # Dynamic month options for summary
         def update_month_choices_summary(year):
-            if year:
-                return gr.Dropdown.update(choices=[""] + quote_month_choices(year), value="")
-            else:
-                return gr.Dropdown.update(choices=[""], value="")
+            months = quote_month_choices(year) if year else []
+            return gr.Dropdown.update(choices=[""] + months, value="")
         year_qs.change(update_month_choices_summary, year_qs, month_qs)
 
         def quotes_summary_wrapper(year, month):
@@ -278,10 +282,8 @@ with gr.Blocks(title="Graffiti Admin Dashboard") as app:
 
         # Dynamic month options for rep quotes
         def update_month_choices(year):
-            if year:
-                return gr.Dropdown.update(choices=[""] + quote_month_choices(year), value="")
-            else:
-                return gr.Dropdown.update(choices=[""], value="")
+            months = quote_month_choices(year) if year else []
+            return gr.Dropdown.update(choices=[""] + months, value="")
         year_q.change(update_month_choices, year_q, month_q)
 
         def get_rep_quotes_filtered_wrapper(rep, year, month):