Update app.py

app.py

@@ -440,6 +440,25 @@ def init_db():
         )
     """)
 
+    # Backward-compatible analytics migration
+    analytics_columns = {
+        "referrer": "TEXT",
+        "utm_source": "TEXT",
+        "utm_medium": "TEXT",
+        "utm_campaign": "TEXT",
+        "utm_content": "TEXT",
+        "utm_term": "TEXT",
+    }
+
+    cur.execute("PRAGMA table_info(analytics)")
+    existing_columns = {row[1] for row in cur.fetchall()}
+
+    for column, column_type in analytics_columns.items():
+        if column not in existing_columns:
+            cur.execute(
+                f"ALTER TABLE analytics ADD COLUMN {column} {column_type}"
+            )
+    
     conn.commit()
     conn.close()
     print(f"✅ Database initialized at: {DB_PATH}")
@@ -525,26 +544,75 @@ def save_calculation(email, country, currency, breed, birds, weeks, days, total_
 
 
 def track_visit(request: gr.Request):
-    """Track page visit."""
+    """Track page visit, country, referrer and UTM campaign data."""
     try:
-        client_ip = request.client.host if request and hasattr(request, 'client') else "unknown"
-        
+        client_ip = (
+            request.client.host
+            if request and hasattr(request, "client")
+            else "unknown"
+        )
+
+        headers = request.headers if request and hasattr(request, "headers") else {}
+
+        # Referrer
+        referrer = (
+            headers.get("referer")
+            or headers.get("referrer")
+            or ""
+        )
+
+        # UTM parameters
+        query_params = {}
+
+        if request and hasattr(request, "query_params"):
+            try:
+                query_params = dict(request.query_params)
+            except Exception:
+                query_params = {}
+
+        utm_source = query_params.get("utm_source", "")
+        utm_medium = query_params.get("utm_medium", "")
+        utm_campaign = query_params.get("utm_campaign", "")
+        utm_content = query_params.get("utm_content", "")
+        utm_term = query_params.get("utm_term", "")
+
+        country = detect_country_from_env() or "NG"
+
         conn = sqlite3.connect(DB_PATH)
         cur = conn.cursor()
-        
+
         cur.execute("""
-            INSERT INTO analytics (visit_date, ip, country, action, created_at)
-            VALUES (?, ?, ?, ?, ?)
+            INSERT INTO analytics (
+                visit_date,
+                ip,
+                country,
+                action,
+                created_at,
+                referrer,
+                utm_source,
+                utm_medium,
+                utm_campaign,
+                utm_content,
+                utm_term
+            )
+            VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
         """, (
             datetime.now().strftime("%Y-%m-%d"),
             client_ip,
-            detect_country_from_env() or "NG",
+            country,
             "page_visit",
-            datetime.now().isoformat()
+            datetime.now(timezone.utc).isoformat(),
+            referrer,
+            utm_source,
+            utm_medium,
+            utm_campaign,
+            utm_content,
+            utm_term,
         ))
-        
+
         conn.commit()
         conn.close()
+
     except Exception as e:
         print(f"Analytics error: {e}")