Update app.py

app.py

@@ -18,116 +18,118 @@ creds = ServiceAccountCredentials.from_json_keyfile_name("deep-mile-461309-t8-0e
 client = gspread.authorize(creds)
 sheet_url = "https://docs.google.com/spreadsheets/d/1if4KoVQvw5ZbhknfdZbzMkcTiPfsD6bz9V3a1th-bwQ"
 
-# ------------------ SHEET REFRESH ------------------
+# ------------------ SHEET ACCESS UTILS ------------------
 def load_sheet(sheet_name):
     sheet = client.open_by_url(sheet_url).worksheet(sheet_name)
-    df = pd.DataFrame(sheet.get_all_records())
-    return df
-
-# ------------------ REPORT HELPERS ------------------
-def week_range(date):
-    start = date - timedelta(days=date.weekday())
+    data = sheet.get_all_records()
+    return pd.DataFrame(data)
+
+# ------------------ REPORT UTILS ------------------
+def parse_date(date_str):
+    try:
+        return pd.to_datetime(date_str, errors='coerce').date()
+    except:
+        return None
+
+def get_week_range(start_date):
+    start = start_date - timedelta(days=start_date.weekday())
     end = start + timedelta(days=6)
     return start, end
 
-def month_range(date):
-    start = date.replace(day=1)
-    if start.month == 12:
-        end = start.replace(year=start.year+1, month=1, day=1) - timedelta(days=1)
-    else:
-        end = start.replace(month=start.month+1, day=1) - timedelta(days=1)
-    return start, end
+def get_month_range(date_obj):
+    start = date_obj.replace(day=1)
+    next_month = start + pd.DateOffset(months=1)
+    end = (next_month - timedelta(days=1)).date()
+    return start.date(), end
 
-# ------------------ REPORT FILTERS ------------------
-def filter_calls(date, period="Daily"):
+# ------------------ CALLS ------------------
+def filter_calls_by_date_range(start, end):
     df = load_sheet("Calls")
-    df['Date'] = pd.to_datetime(df['Date'], errors='coerce')
-    if period == "Weekly":
-        start, end = week_range(date)
-        df = df[(df['Date'] >= start) & (df['Date'] <= end)]
-    elif period == "Monthly":
-        start, end = month_range(date)
-        df = df[(df['Date'] >= start) & (df['Date'] <= end)]
-    else:
-        df = df[df['Date'] == date]
-    return df
-
-def filter_appointments(date, period="Daily"):
+    df['Date'] = pd.to_datetime(df['Date'], errors='coerce').dt.date
+    return df[(df['Date'] >= start) & (df['Date'] <= end)]
+
+# ------------------ APPOINTMENTS ------------------
+def filter_appointments_by_date_range(start, end):
     df = load_sheet("Appointments")
-    df['Date'] = pd.to_datetime(df['Date'], errors='coerce')
-    if period == "Weekly":
-        start, end = week_range(date)
-        df = df[(df['Date'] >= start) & (df['Date'] <= end)]
-    elif period == "Monthly":
-        start, end = month_range(date)
-        df = df[(df['Date'] >= start) & (df['Date'] <= end)]
-    else:
-        df = df[df['Date'] == date]
-    return df
-
-# ------------------ USERS & METRICS ------------------
-def get_rep_metrics(date, period="Weekly"):
-    calls = filter_calls(date, period)
-    appts = filter_appointments(date, period)
-    users = load_sheet("Users")
-
-    call_counts = calls['Rep'].value_counts().to_dict()
-    appt_counts = appts['Rep'].value_counts().to_dict()
-
-    results = []
-    for _, row in users.iterrows():
-        rep = row['Rep']
-        calls_done = call_counts.get(rep, 0)
-        appts_done = appt_counts.get(rep, 0)
-        results.append({
-            "Rep": rep,
-            "Weekly Target": row.get('Weekly Target', 0),
-            "Calls Made": calls_done,
-            "Appointments Booked": appts_done,
-            "Gap to Target": max(0, row.get('Weekly Target', 0) - calls_done)
-        })
-    return pd.DataFrame(results)
+    df['Date'] = pd.to_datetime(df['Date'], errors='coerce').dt.date
+    return df[(df['Date'] >= start) & (df['Date'] <= end)]
+
+# ------------------ APPOINTED LEADS ------------------
+def appointed_leads_table():
+    df = load_sheet("Appointed Leads")
+    return df.groupby('Rep')['Customer Name'].apply(list).reset_index()
+
+# ------------------ INTERACTIVE QUERY ------------------
+def search_table(sheet_name, field, keyword):
+    df = load_sheet(sheet_name)
+    if field not in df.columns:
+        return pd.DataFrame(), "Field not found."
+    results = df[df[field].astype(str).str.contains(keyword, case=False, na=False)]
+    return results, f"Found {len(results)} results."
 
 # ------------------ UI ------------------
 with gr.Blocks() as app:
     with gr.Row():
         with gr.Column(visible=True) as login_ui:
-            gr.Markdown("## 🔐 Login Required")
+            gr.Markdown("## \ud83d\udd10 Login Required")
             email = gr.Textbox(label="Email")
             password = gr.Textbox(label="Password", type="password")
             login_btn = gr.Button("Login")
             login_msg = gr.Markdown("")
 
         with gr.Column(visible=False) as main_ui:
-            gr.Markdown("## 📊 Graffiti Admin Dashboard")
-
-            with gr.Tab("📞 Calls Report"):
-                calls_period = gr.Radio(choices=["Daily", "Weekly", "Monthly"], value="Daily", label="Report Period")
-                calls_date = gr.Date(label="Select Date")
-                calls_output = gr.Dataframe()
-                calls_date.change(fn=filter_calls, inputs=[calls_date, calls_period], outputs=calls_output)
-                calls_period.change(fn=filter_calls, inputs=[calls_date, calls_period], outputs=calls_output)
-
-            with gr.Tab("📅 Appointments Report"):
-                appt_period = gr.Radio(choices=["Daily", "Weekly", "Monthly"], value="Daily", label="Report Period")
-                appt_date = gr.Date(label="Select Date")
-                appt_output = gr.Dataframe()
-                appt_date.change(fn=filter_appointments, inputs=[appt_date, appt_period], outputs=appt_output)
-                appt_period.change(fn=filter_appointments, inputs=[appt_date, appt_period], outputs=appt_output)
-
-            with gr.Tab("🎯 Rep Metrics vs Targets"):
-                metric_date = gr.Date(label="Select Any Date in the Week/Month")
-                metric_period = gr.Radio(choices=["Weekly", "Monthly"], value="Weekly", label="Metric Period")
-                metric_table = gr.Dataframe()
-                metric_date.change(fn=get_rep_metrics, inputs=[metric_date, metric_period], outputs=metric_table)
-                metric_period.change(fn=get_rep_metrics, inputs=[metric_date, metric_period], outputs=metric_table)
+            gr.Markdown("## \ud83d\udcc2 Graffiti Admin Dashboard")
+
+            with gr.Tab("\ud83d\udcc5 Calls Report"):
+                report_type_call = gr.Radio(["Weekly", "Monthly"], label="Report Type")
+                date_input_call = gr.Textbox(label="Enter Start Date (YYYY-MM-DD)")
+                calls_table = gr.Dataframe()
+                
+                def generate_call_report(rt, date_str):
+                    date = parse_date(date_str)
+                    if not date:
+                        return pd.DataFrame()
+                    start, end = get_week_range(date) if rt == "Weekly" else get_month_range(date)
+                    return filter_calls_by_date_range(start, end)
+                
+                report_type_call.change(generate_call_report, inputs=[report_type_call, date_input_call], outputs=calls_table)
+                date_input_call.change(generate_call_report, inputs=[report_type_call, date_input_call], outputs=calls_table)
+
+            with gr.Tab("\ud83d\udcc5 Appointments Report"):
+                report_type_appt = gr.Radio(["Weekly", "Monthly"], label="Report Type")
+                date_input_appt = gr.Textbox(label="Enter Start Date (YYYY-MM-DD)")
+                appt_table = gr.Dataframe()
+
+                def generate_appt_report(rt, date_str):
+                    date = parse_date(date_str)
+                    if not date:
+                        return pd.DataFrame()
+                    start, end = get_week_range(date) if rt == "Weekly" else get_month_range(date)
+                    return filter_appointments_by_date_range(start, end)
+
+                report_type_appt.change(generate_appt_report, inputs=[report_type_appt, date_input_appt], outputs=appt_table)
+                date_input_appt.change(generate_appt_report, inputs=[report_type_appt, date_input_appt], outputs=appt_table)
+
+            with gr.Tab("\ud83e\uddcd Appointed Leads"):
+                leads_btn = gr.Button("View Appointed Leads")
+                leads_output = gr.Dataframe()
+                leads_btn.click(fn=appointed_leads_table, outputs=leads_output)
+
+            with gr.Tab("\ud83d\udd0d Query Live Sheets"):
+                sheet_choice = gr.Dropdown(choices=["LiveQuotes", "LiveCustomer", "LiveJobBags"], label="Select Sheet")
+                field_input = gr.Textbox(label="Field (column name)")
+                keyword_input = gr.Textbox(label="Keyword to search")
+                query_btn = gr.Button("Search")
+                query_table = gr.Dataframe()
+                query_info = gr.Markdown()
+                query_btn.click(fn=search_table, inputs=[sheet_choice, field_input, keyword_input], outputs=[query_table, query_info])
 
     def do_login(user, pw):
         if VALID_USERS.get(user) == pw:
             return gr.update(visible=False), gr.update(visible=True), ""
         else:
-            return gr.update(visible=True), gr.update(visible=False), "❌ Invalid email or password."
+            return gr.update(visible=True), gr.update(visible=False), "\u274c Invalid email or password."
 
     login_btn.click(fn=do_login, inputs=[email, password], outputs=[login_ui, main_ui, login_msg])
 
-app.launch()
+app.launch()