Update app.py

app.py

@@ -94,7 +94,7 @@ def authenticate(pwd):
     return True
 
 # =====================================================
-# DATA FETCH — ALL COMPANY-LEVEL FIELDS
+# DATA FETCH (ALL FIELDS, COMPANY LEVEL)
 # =====================================================
 def fetch_company_data(start_date, end_date):
     page, rows = 1, []
@@ -111,50 +111,66 @@ def fetch_company_data(start_date, end_date):
             }
         )
         r.raise_for_status()
-        payload = r.json()
-
-        if not payload.get("data"):
+        js = r.json()
+        if not js.get("data"):
             break
 
-        location_map = {
+        loc_map = {
             i["id"]: i["attributes"]
-            for i in payload.get("included", [])
+            for i in js.get("included", [])
             if i["type"] == "locations"
         }
 
-        for lead in payload["data"]:
+        for lead in js["data"]:
             a = lead["attributes"]
             loc_id = lead.get("relationships", {}).get("location", {}).get("data", {}).get("id")
-            loc = location_map.get(loc_id, {})
+            loc = loc_map.get(loc_id, {})
 
             rows.append({
+                "lead_id": lead.get("id"),
                 "company_name": a.get("name"),
                 "website_url": a.get("website_url"),
+                "phone": a.get("phone"),
                 "business_id": a.get("business_id"),
-                "logo_url": a.get("logo_url"),
+
                 "primary_industry": a.get("industry"),
-                "all_industries": ", ".join([i["name"] for i in a.get("industries", [])]),
+                "all_industries": ", ".join([i.get("name") for i in a.get("industries", [])])
+                    if a.get("industries") else None,
+
                 "first_visit_date": a.get("first_visit_date"),
                 "last_visit_date": a.get("last_visit_date"),
                 "total_visits": a.get("visits"),
                 "lead_quality_score": a.get("quality"),
+
                 "employee_count": a.get("employee_count"),
+                "employees_min": a.get("employees_range", {}).get("min") if a.get("employees_range") else None,
+                "employees_max": a.get("employees_range", {}).get("max") if a.get("employees_range") else None,
                 "revenue": a.get("revenue"),
+
                 "assignee": a.get("assignee"),
+                "emailed_to": a.get("emailed_to"),
                 "crm_lead_id": a.get("crm_lead_id"),
                 "crm_organization_id": a.get("crm_organization_id"),
                 "tags": ", ".join(a.get("tags", [])) if a.get("tags") else None,
+
                 "linkedin_url": a.get("linkedin_url"),
                 "twitter_handle": a.get("twitter_handle"),
                 "facebook_url": a.get("facebook_url"),
+                "logo_url": a.get("logo_url"),
+
                 "country": loc.get("country"),
+                "country_code": loc.get("country_code"),
                 "region": loc.get("region"),
                 "city": loc.get("city"),
+
+                "leadfeeder_url": a.get("view_in_leadfeeder"),
             })
 
         page += 1
 
-    return pd.DataFrame(rows)
+    df = pd.DataFrame(rows)
+    df["last_visit_date"] = pd.to_datetime(df["last_visit_date"], errors="coerce")
+    return df
 
 # =====================================================
 # KPI CARDS