Add mixpanel tracking

app.py

@@ -1,18 +1,26 @@
 import albumentations as A
 import base64
 import cv2
+import logging
 import gradio as gr
 import inspect
 import io
 import numpy as np
+import os
 
 from copy import deepcopy
 from functools import wraps
 from PIL import Image, ImageDraw
 from typing import get_type_hints
+from mixpanel import Mixpanel
 
 from utils import is_not_supported_transform
 
+logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(name)s - %(levelname)s - %(message)s")
+logger = logging.getLogger(__name__)
+
+MIXPANEL_TOKEN = os.getenv("MIXPANEL_TOKEN")
+mp = Mixpanel(MIXPANEL_TOKEN)
 
 HEADER = f"""
 <div align="center">
@@ -82,6 +90,13 @@ for mask in BASE64_DEFAULT_MASKS:
     )
 
 
+def track_event(event_name, user_id="unknown", properties=None):
+    if properties is None:
+        properties = {}
+    mp.track(user_id, event_name, properties)
+    logger.info(f"Event tracked: {event_name} - {properties}")
+
+
 def run_with_retry(compose):
     @wraps(compose)
     def wrapper(*args, **kwargs):
@@ -193,6 +208,7 @@ def get_formatted_signature(function_or_class, indentation=4):
 
 def get_formatted_transform(transform_number):
     transform_name = transforms_keys[transform_number]
+    track_event("transform_selected", properties={"transform_name": transform_name})
     transform = transforms_map[transform_name]
     return f"A.{transform.__name__}{get_formatted_signature(transform)}"
 
@@ -206,6 +222,8 @@ def get_formatted_transform_docs(transform_number):
 def update_augmented_images(image, code):
 
     augmentation = eval(code)
+    track_event("transform_applied", properties={"transform_name": augmentation.__class__.__name__, "code": code})
+
     compose = A.Compose(
         [augmentation],
         bbox_params=A.BboxParams(format="pascal_voc", label_fields=["category_id"]),

requirements.txt

@@ -1,2 +1,3 @@
 albumentations
-Pillow
+Pillow
+mixpanel