Update app.py

app.py

@@ -3,6 +3,7 @@ import gspread
 import gradio as gr
 from oauth2client.service_account import ServiceAccountCredentials
 from datetime import datetime
+from math import radians, cos, sin, asin, sqrt
 
 # ------------------ AUTH ------------------
 VALID_USERS = {
@@ -25,13 +26,22 @@ def load_tab(sheet_name):
     except:
         return pd.DataFrame([["⚠️ Could not load sheet."]], columns=["Error"])
 
+def haversine(coord1, coord2):
+    lon1, lat1 = map(radians, map(float, coord1.split(',')[::-1]))
+    lon2, lat2 = map(radians, map(float, coord2.split(',')[::-1]))
+    dlon = lon2 - lon1
+    dlat = lat2 - lat1
+    a = sin(dlat/2)**2 + cos(lat1) * cos(lat2) * sin(dlon/2)**2
+    c = 2 * asin(sqrt(a))
+    return 6371 * c
+
 # ------------------ FIELD SALES ------------------
 def load_field_sales():
     df = load_tab("Field Sales")
-    if df.empty:
-        return pd.DataFrame(columns=["Date", "Rep", "Order Value", "Order Received", "Current/Prospect Customer", "Customer Type & Status", "DateStr"])
+    if df.empty or "Date" not in df.columns:
+        return pd.DataFrame()
 
-    df['Date'] = pd.to_datetime(df.get("Date", datetime.today()), errors='coerce')
+    df['Date'] = pd.to_datetime(df['Date'], errors='coerce')
     df = df.dropna(subset=["Date"])
     df['DateStr'] = df['Date'].dt.date.astype(str)
 
@@ -40,101 +50,86 @@ def load_field_sales():
     else:
         df["Order Value"] = pd.to_numeric(df["Order Value"], errors="coerce").fillna(0)
 
+    # Distance calc
+    distances = [0]
+    for i in range(1, len(df)):
+        try:
+            prev = df.loc[i-1, 'Location']
+            curr = df.loc[i, 'Location']
+            if pd.notna(prev) and pd.notna(curr):
+                distances.append(round(haversine(prev, curr), 2))
+            else:
+                distances.append(0)
+        except:
+            distances.append(0)
+    df["Distance Travelled (km)"] = distances
+
+    return df
+
+# ------------------ TELESALES ------------------
+def load_telesales():
+    df = load_tab("TeleSales")
+    if df.empty or "Rep Email" not in df.columns:
+        return pd.DataFrame()
+
+    df["Date"] = pd.to_datetime(df.get("Date", datetime.today()), errors='coerce')
+    df["DateStr"] = df["Date"].dt.date.astype(str)
+    return df
+
+# ------------------ OEM ------------------
+def load_oem():
+    df = load_tab("OEM Visit")
+    if df.empty or "Rep" not in df.columns:
+        return pd.DataFrame()
+    df["Date"] = pd.to_datetime(df.get("Date", datetime.today()), errors='coerce')
+    df["DateStr"] = df["Date"].dt.date.astype(str)
     return df
 
+# ------------------ SUMMARY ------------------
 def generate_summary(date_str):
     df = load_field_sales()
     if df.empty:
-        return pd.DataFrame([["No data"]], columns=["Message"]), pd.DataFrame(), pd.DataFrame()
+        return pd.DataFrame([["No Field Sales data"]], columns=["Message"])*5
 
+    df_day = df[df['DateStr'] == date_str.strip()]
     all_reps = sorted(df['Rep'].dropna().unique())
-    date_str = date_str.strip()
-    day_df = df[df['DateStr'] == date_str]
-    total_visits = day_df.groupby("Rep").size().reset_index(name="Total Visits")
-
     col = "Current/Prospect Customer"
-    if col not in df.columns:
-        df[col] = ""
 
-    current = day_df[day_df[col] == "Current"]
-    prospect = day_df[day_df[col] == "Prospect"]
+    # --- Visits Breakdown
+    total_visits = df_day.groupby("Rep").size().reset_index(name="Total Visits")
+    current = df_day[df_day[col] == "Current"]
+    prospect = df_day[df_day[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]
     })
+    inactive = pd.DataFrame({'Inactive Reps': [rep for rep in all_reps if rep not in total_visits["Rep"].tolist()]})
 
-    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})
-    return total_visits, breakdown, inactive_df
-
-def get_order_summary(date_str):
-    df = load_field_sales()
-    if df.empty:
-        return pd.DataFrame([["No data"]], columns=["Message"])
-
-    date_str = date_str.strip()
-    day_df = df[df['DateStr'] == date_str]
-    if "Order Received" not in df.columns:
-        df["Order Received"] = ""
-
-    rep_group = day_df.groupby("Rep").agg({
+    # --- Field Summary per Rep
+    rep_summary = df_day.groupby("Rep").agg({
+        "Order Value": "sum",
         "Order Received": lambda x: (x == "Yes").sum(),
-        "Order Value": "sum"
-    }).reset_index().rename(columns={
+        "Current/Prospect Customer": lambda x: (x == "Current").sum(),
+        "Distance Travelled (km)": "sum"
+    }).rename(columns={
+        "Order Value": "Total Order Value",
         "Order Received": "Orders Received",
-        "Order Value": "Total Order Value"
-    })
-    return rep_group.sort_values(by="Total Order Value", ascending=False)
-
-def get_escalations():
-    df = load_field_sales()
-    if df.empty:
-        return pd.DataFrame([["No data in Field Sales"]], columns=["Message"])
-
-    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=["Message"])
-    else:
-        return pd.DataFrame([["⚠️ Column 'Customer Type & Status' not found."]], columns=["Message"])
-
-# ------------------ TELESALeS ------------------
-def get_telesales_summary():
-    df = load_tab("TeleSales")
-    if df.empty or "Rep Email" not in df.columns:
-        return pd.DataFrame([["No data available"]], columns=["Message"])
-    
-    df["Date"] = pd.to_datetime(df.get("Date", datetime.today()), errors='coerce')
-    df["DateStr"] = df["Date"].dt.date.astype(str)
-    grouped = df.groupby(["Rep Email", "DateStr"]).size().reset_index(name="Calls Made")
-    return grouped.rename(columns={"Rep Email": "Rep"})
-
-# ------------------ OEM VISITS ------------------
-def get_oem_summary():
-    df = load_tab("OEM Visit")
-    if df.empty or "Rep" not in df.columns:
-        return pd.DataFrame([["No data available"]], columns=["Message"])
+        "Current/Prospect Customer": "Current Customers",
+        "Distance Travelled (km)": "Total Distance (km)"
+    }).reset_index()
 
-    df["Date"] = pd.to_datetime(df.get("Date", datetime.today()), errors='coerce')
-    df["DateStr"] = df["Date"].dt.date.astype(str)
-    return df.groupby(["Rep", "DateStr"]).size().reset_index(name="OEM Visits")
+    # --- TeleSales Summary
+    df_ts = load_telesales()
+    df_ts_day = df_ts[df_ts['DateStr'] == date_str.strip()]
+    ts_summary = df_ts_day.groupby("Rep Email").size().reset_index(name="Total Calls Made") if not df_ts_day.empty else pd.DataFrame([["No Telesales"]], columns=["Info"])
 
-# ------------------ CUSTOMER REQUESTS ------------------
-def get_requests():
-    df = load_tab("Customer Requests")
-    return df if not df.empty else pd.DataFrame([["No requests yet."]], columns=["Message"])
+    # --- OEM Summary
+    df_oem = load_oem()
+    df_oem_day = df_oem[df_oem['DateStr'] == date_str.strip()]
+    oem_summary = df_oem_day.groupby("Rep").size().reset_index(name="Total OEM Visits") if not df_oem_day.empty else pd.DataFrame([["No OEM Visits"]], columns=["Info"])
 
-# ------------------ CUSTOMER LISTINGS ------------------
-def get_listings():
-    df = load_tab("CustomerListings")
-    return df if not df.empty else pd.DataFrame([["No listings found."]], columns=["Message"])
-
-# ------------------ USERS ------------------
-def get_users():
-    df = load_tab("Users")
-    return df if not df.empty else pd.DataFrame([["No users configured."]], columns=["Message"])
+    return total_visits, breakdown, inactive, rep_summary, ts_summary, oem_summary
 
 # ------------------ GRADIO APP ------------------
 with gr.Blocks() as app:
@@ -158,55 +153,12 @@ with gr.Blocks() as app:
                 visits = gr.Dataframe(label="✅ Total Visits")
                 breakdown = gr.Dataframe(label="🏷️ Current vs. Prospect")
                 inactive = gr.Dataframe(label="⚠️ Inactive Reps")
-                date_summary.change(fn=generate_summary, inputs=date_summary, outputs=[visits, breakdown, inactive])
-
-            # --- Orders Tab ---
-            with gr.Tab("📦 Orders"):
-                order_date = gr.Dropdown(label="Select Date", choices=unique_dates)
-                order_table = gr.Dataframe(label="💰 Orders Summary")
-                order_date.change(fn=get_order_summary, inputs=order_date, outputs=order_table)
-
-            # --- Escalations ---
-            with gr.Tab("🚨 Escalations"):
-                esc_table = gr.Dataframe(value=get_escalations, label="Second-hand Dealerships")
-                esc_btn = gr.Button("🔄 Refresh")
-                esc_btn.click(fn=get_escalations, outputs=esc_table)
-
-            # --- TeleSales ---
-            with gr.Tab("📞 TeleSales"):
-                ts_table = gr.Dataframe(value=get_telesales_summary, label="TeleSales Summary")
-                ts_refresh = gr.Button("🔄 Refresh TeleSales")
-                ts_refresh.click(fn=get_telesales_summary, outputs=ts_table)
-
-            # --- OEM Visits ---
-            with gr.Tab("🏭 OEM Visits"):
-                oem_table = gr.Dataframe(value=get_oem_summary, label="OEM Visit Summary")
-                oem_refresh = gr.Button("🔄 Refresh OEM")
-                oem_refresh.click(fn=get_oem_summary, outputs=oem_table)
-
-            # --- Requests ---
-            with gr.Tab("📬 Customer Requests"):
-                req_table = gr.Dataframe(value=get_requests, label="Customer Requests", interactive=False)
-                req_refresh = gr.Button("🔄 Refresh Requests")
-                req_refresh.click(fn=get_requests, outputs=req_table)
-
-            # --- Dealerships ---
-            with gr.Tab("📋 Dealership Directory"):
-                listings_table = gr.Dataframe(value=get_listings, label="Customer Listings")
-                listings_refresh = gr.Button("🔄 Refresh Listings")
-                listings_refresh.click(fn=get_listings, outputs=listings_table)
-
-            # --- Users ---
-            with gr.Tab("👤 Users"):
-                users_table = gr.Dataframe(value=get_users, label="Users")
-                users_refresh = gr.Button("🔄 Refresh Users")
-                users_refresh.click(fn=get_users, outputs=users_table)
-
-            # --- Field Sales Raw Data ---
-            with gr.Tab("🧾 Field Sales Raw Data"):
-                field_df = gr.Dataframe(value=load_field_sales, label="Full Field Sales Data", interactive=False)
-                field_btn = gr.Button("🔄 Refresh Field Sales")
-                field_btn.click(fn=load_field_sales, outputs=field_df)
+                field_summary = gr.Dataframe(label="🚗 Field Sales Summary")
+                ts_summary = gr.Dataframe(label="📞 Telesales Summary")
+                oem_summary = gr.Dataframe(label="🏭 OEM Visit Summary")
+
+                date_summary.change(fn=generate_summary, inputs=date_summary,
+                                    outputs=[visits, breakdown, inactive, field_summary, ts_summary, oem_summary])
 
     def do_login(user, pw):
         if VALID_USERS.get(user) == pw: