Update app.py

app.py

@@ -1,4 +1,3 @@
-# app.py
 import gradio as gr
 import pandas as pd
 import gspread
@@ -32,74 +31,84 @@ def compose_date(year, month, day):
     except:
         return None
 
-# -------------------- REPORT FUNCTIONS --------------------
-def current_week_calls():
-    df = load_sheet("Calls")
-    if "Call Date" not in df.columns:
-        return pd.DataFrame([{"Error": "Missing 'Call Date' column"}])
-    df["Call Date"] = pd.to_datetime(df["Call Date"], errors='coerce').dt.date
+def filter_week(df, column, rep=None):
+    df[column] = pd.to_datetime(df[column], errors='coerce').dt.date
     start, end = get_current_week_range()
-    return df[(df["Call Date"] >= start) & (df["Call Date"] <= end)]
+    filtered = df[(df[column] >= start) & (df[column] <= end)]
+    if rep:
+        filtered = filtered[filtered['Rep'].fillna('').str.lower() == rep.lower()]
+    return filtered
 
-def custom_date_calls(y, m, d):
+def filter_by_date(df, column, y, m, d, rep=None):
     date_str = compose_date(y, m, d)
     if not date_str:
-        return pd.DataFrame([{"Error": "Invalid date input"}])
+        return pd.DataFrame([{"Error": "Invalid date"}])
+    df[column] = pd.to_datetime(df[column], errors='coerce').dt.date.astype(str)
+    result = df[df[column] == date_str]
+    if rep:
+        result = result[result['Rep'].fillna('').str.lower() == rep.lower()]
+    return result
+
+# -------------------- FUNCTIONS --------------------
+def current_week_calls(rep=None):
     df = load_sheet("Calls")
-    df["Call Date"] = pd.to_datetime(df["Call Date"], errors='coerce').dt.date.astype(str)
-    return df[df["Call Date"] == date_str]
+    if "Call Date" not in df.columns:
+        return pd.DataFrame([{"Error": "Missing 'Call Date' column"}])
+    return filter_week(df, "Call Date", rep)
 
-def current_week_appointments():
+def current_week_appointments(rep=None):
     df = load_sheet("Appointments")
     if "Appointment Date" not in df.columns:
         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()
-    return df[(df["Appointment Date"] >= start) & (df["Appointment Date"] <= end)]
+    return filter_week(df, "Appointment Date", rep)
 
-def custom_date_appointments(y, m, d):
-    date_str = compose_date(y, m, d)
-    if not date_str:
-        return pd.DataFrame([{"Error": "Invalid date input"}])
+def custom_date_calls(y, m, d, rep=None):
+    df = load_sheet("Calls")
+    return filter_by_date(df, "Call Date", y, m, d, rep)
+
+def custom_date_appointments(y, m, d, rep=None):
     df = load_sheet("Appointments")
-    df["Appointment Date"] = pd.to_datetime(df["Appointment Date"], errors='coerce').dt.date.astype(str)
-    return df[df["Appointment Date"] == date_str]
+    return filter_by_date(df, "Appointment Date", y, m, d, rep)
 
 def appointed_leads():
     df = load_sheet("AllocatedLeads")
-    if "Rep" not in df.columns or "Company Name" not in df.columns:
-        return pd.DataFrame([{"Error": "Missing 'Rep' or 'Company Name' column"}])
-    return df.groupby("Rep")["Company Name"].apply(list).reset_index()
+    if "Assigned Rep" not in df.columns or "Company Name" not in df.columns:
+        return pd.DataFrame([{"Error": "Missing 'Assigned Rep' or 'Company Name' column"}])
+    return df.groupby("Assigned Rep")["Company Name"].apply(list).reset_index()
 
 # -------------------- UI --------------------
 with gr.Blocks(title="Graffiti Admin Dashboard") as app:
-    gr.Markdown("## 📊 Graffiti Admin Dashboard")
+    gr.Markdown("## 📅 Graffiti Admin Dashboard")
 
-    with gr.Tab("📞 Calls Report"):
+    with gr.Tab("Calls Report"):
+        calls_rep = gr.Textbox(label="Optional Rep Filter")
         calls_btn = gr.Button("Load Current Week Calls")
         calls_table = gr.Dataframe()
-        calls_btn.click(fn=current_week_calls, outputs=calls_table)
+        calls_btn.click(fn=current_week_calls, inputs=calls_rep, outputs=calls_table)
 
-        gr.Markdown("### 🔎 Search Calls by Specific Date")
-        cy, cm, cd = gr.Textbox(label="Year"), gr.Textbox(label="Month"), gr.Textbox(label="Day")
-        custom_calls_btn = gr.Button("Search Calls by Date")
-        custom_calls_table = gr.Dataframe()
-        custom_calls_btn.click(fn=custom_date_calls, inputs=[cy, cm, cd], outputs=custom_calls_table)
+        gr.Markdown("### 🔍 Search Calls by Specific Date")
+        y, m, d = gr.Textbox(label="Year"), gr.Textbox(label="Month"), gr.Textbox(label="Day")
+        call_rep = gr.Textbox(label="Optional Rep Filter")
+        date_calls_btn = gr.Button("Search Calls by Date")
+        date_calls_table = gr.Dataframe()
+        date_calls_btn.click(fn=custom_date_calls, inputs=[y, m, d, call_rep], outputs=date_calls_table)
 
-    with gr.Tab("📅 Appointments Report"):
+    with gr.Tab("Appointments Report"):
+        appt_rep = gr.Textbox(label="Optional Rep Filter")
         appt_btn = gr.Button("Load Current Week Appointments")
         appt_table = gr.Dataframe()
-        appt_btn.click(fn=current_week_appointments, outputs=appt_table)
+        appt_btn.click(fn=current_week_appointments, inputs=appt_rep, outputs=appt_table)
 
-        gr.Markdown("### 🔎 Search Appointments by Specific Date")
+        gr.Markdown("### 🔍 Search Appointments by Specific Date")
         ay, am, ad = gr.Textbox(label="Year"), gr.Textbox(label="Month"), gr.Textbox(label="Day")
-        custom_appt_btn = gr.Button("Search Appointments by Date")
-        custom_appt_table = gr.Dataframe()
-        custom_appt_btn.click(fn=custom_date_appointments, inputs=[ay, am, ad], outputs=custom_appt_table)
+        appt_rep2 = gr.Textbox(label="Optional Rep Filter")
+        date_appt_btn = gr.Button("Search Appointments by Date")
+        date_appt_table = gr.Dataframe()
+        date_appt_btn.click(fn=custom_date_appointments, inputs=[ay, am, ad, appt_rep2], outputs=date_appt_table)
 
-    with gr.Tab("👥 Appointed Leads"):
+    with gr.Tab("Appointed Leads"):
         leads_btn = gr.Button("View Appointed Leads")
         leads_table = gr.Dataframe()
         leads_btn.click(fn=appointed_leads, outputs=leads_table)
 
-app.launch()
+app.launch(share=True)