lightning-pose-app 2.0.4__py3-none-any.whl → 2.0.4.1__py3-none-any.whl

{lightning_pose_app-2.0.4.dist-info → lightning_pose_app-2.0.4.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lightning-pose-app
-Version: 2.0.4
+Version: 2.0.4.1
 Summary: 
 Requires-Python: >=3.10
 Classifier: Programming Language :: Python :: 3

{lightning_pose_app-2.0.4.dist-info → lightning_pose_app-2.0.4.1.dist-info}/RECORD

@@ -4,14 +4,14 @@ litpose_app/config_default.yaml,sha256=Dx83muIy0ceqG3xXlwOA17zQME_pVSUY0PBEaio34
 litpose_app/config_default_multiview.yaml,sha256=vTEkvrEfcJoOXQJWy9gnTK6U2iBaV9LS1o5OAbTaguo,7883
 litpose_app/datatypes.py,sha256=QIz77uP5dPr0N0DfaLT1KZIjRPzDqWM35i2KyetEUOQ,618
 litpose_app/deps.py,sha256=d_YWr8wE4T4Yv87A92ce5ziPp77eQzk6UbyVpbYxbUs,2628
-litpose_app/main.py,sha256=R3FfiilS_D4GUzTLq47EXG2cTqqs3v5JY6J2RXwwGkQ,7016
+litpose_app/main.py,sha256=qHHA-lUA7B6cR7IWfzzcw7DCKvLVtU-nQyRqJhwQOHg,8527
 litpose_app/ngdist/ng_app/3rdpartylicenses.txt,sha256=xJQdd3vjhodQmCRoPgqCuuLFpmWdo2Fo44IZjz4Jgpk,26440
 litpose_app/ngdist/ng_app/browser/app.component-AUUQIS4L.css.map,sha256=e6lXWzmK3xppKK3tXHUccK1yGZqd1kzyTpDH0F1nC2g,344
 litpose_app/ngdist/ng_app/browser/create-model-dialog.component-XM4MKM6L.css.map,sha256=VoykZn4fCKe9ybsL20I-uR6tCGuxLAO92mFmFcuTphk,385
 litpose_app/ngdist/ng_app/browser/error-dialog.component-HYLQSJEP.css.map,sha256=zJuF-LfB994Y1IrnIz38mariDFb8yucffbWPXgHGbvw,355
 litpose_app/ngdist/ng_app/browser/extract-frames-dialog.component-DKKIPRKS.css.map,sha256=R7j-k-UWnqcy0Lwyrg41j9K1vJ8pdu3g27YzTgvsdLw,397
 litpose_app/ngdist/ng_app/browser/favicon.ico,sha256=QtbXVfx3HI-WqWh_kkBIQyYzJsDmLw_3Y4_UN7pBepE,15406
-litpose_app/ngdist/ng_app/browser/index.html,sha256=PFG_NNJL82qNjYlcHJnSPF7MbJMJw8IT5C_IoaWGvUw,25760
+litpose_app/ngdist/ng_app/browser/index.html,sha256=CMRvb89BaTkkm0EhB8wsdXjBHgjskQMNS7oKzxcbtNU,25799
 litpose_app/ngdist/ng_app/browser/keypoint-container.component-KO24463N.css.map,sha256=pdxBbg2QoKPjtR_DYnJFxIMYZiYz1OQDgrJEItSOAqg,1275
 litpose_app/ngdist/ng_app/browser/keypoint-container.component-SZ5BMRPW.css.map,sha256=jANFxnVymflWlxk76V9WnPEACHDUoPdGTTxNuJErCE8,2022
 litpose_app/ngdist/ng_app/browser/labeler-page.component-QRMYNSCU.css.map,sha256=sVCjR7JatS-3NB9JZMH7l8xR4AuTuJgqLSAm1rvDCt4,212
@@ -45,7 +45,7 @@ litpose_app/paths/__init__.py,sha256=qVtaWuPqpIaeEZX3cX30eOcWbDmzTe2eTyTGlt7rHSo
 litpose_app/paths/path_util.py,sha256=ordQtLrAP7wZxcYhoX9-0jB-3k_qb12K_7HNmry61Oc,1614
 litpose_app/paths/path_util_legacy.py,sha256=5nriPHw6XuUPz1i5oX6gWVTa8PBM1Z9_S7yTumjQwuI,10623
 litpose_app/project.py,sha256=xYzyGxPx948_AtyrJoZBpnLC7KH4t77locFKIvcw-3k,2641
-litpose_app/rootconfig.py,sha256=NpB6XCn1tovIOaWRVayqjG65L1Pn_TR3uniLLPrcq-4,1030
+litpose_app/rootconfig.py,sha256=rBjnDa54hml70Esbv1BPfMd4KECjA8cExfwBpvnfY8I,1209
 litpose_app/routes/config_root.py,sha256=qLEBcnovU4RWshTsNbDmJtSsFPjF7WdLCZ99BJ92WIg,534
 litpose_app/routes/configs.py,sha256=Ua3AFQFadYas8i2W_OE2eN10o2EiLk9B518U9nA0dKI,2398
 litpose_app/routes/extract_frames.py,sha256=Xqm6qm-7v-pB82VQdO9dv9bzEEeqgxWdewLR2cvuPM8,4043
@@ -67,6 +67,7 @@ litpose_app/tasks/extract_frames.py,sha256=5OzKCqwOCAMkjqqjuVw4x96Pqj9UTv8l7ombe
 litpose_app/tasks/transcode_fine.py,sha256=OrGH0OA0F8SsN_Aa00hFYg6wEJfQmyuh18dO82VPsOw,3879
 litpose_app/train_scheduler.py,sha256=4q20pUL89XpWnOhuGTzRCF7oe7FS6c2Ykh6qN_xJZh0,9659
 litpose_app/utils/__init__.py,sha256=YW1PsTomhfvpjdA-A5nSzrQfwdYzfizy82N711GlCYg,77
+litpose_app/utils/check_for_upgrade.py,sha256=xcNOnY9WtpdbSIyJXMJ_xTPmzU0TiMos1m55XXC_d30,1554
 litpose_app/utils/enqueue.py,sha256=RqOhL0ObUBmCmlSHHIugDy1-jVOKTfLuqNbIILawxvo,1904
 litpose_app/utils/file_response.py,sha256=Hbh87Upiy0YGe7ioPFoSh5D697HTEWM99-H_A42XSCI,1938
 litpose_app/utils/fix_empty_first_row.py,sha256=34cSDwG4YoAxXUPjDTCcieI-kW8qK8lH8rbZieSN36Q,565
@@ -74,6 +75,6 @@ litpose_app/utils/mv_label_file.py,sha256=VarVDSF0Rx90WThLZfArS7nmKiGcTmONNWjOGl
 litpose_app/utils/video/__init__.py,sha256=dSc-yVjXmn7ajNR7WgkCUUFdn9RLS2NBM2CjH0ePQXE,464
 litpose_app/utils/video/export_frames.py,sha256=kAIqjMarqTe51mJd3jk9xw4MotEUV1UfxtgTepyw2mc,1738
 litpose_app/utils/video/frame_selection.py,sha256=2x_F6Ez7WHDmvC-FeD6lO31kPvG5oRLrDQIhndoRT-g,4736
-lightning_pose_app-2.0.4.dist-info/METADATA,sha256=Aj2mJ88piV_V6ZuyyoFi1az-mKuIQ_XWh6CbvrrASTk,1397
-lightning_pose_app-2.0.4.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
-lightning_pose_app-2.0.4.dist-info/RECORD,,
+lightning_pose_app-2.0.4.1.dist-info/METADATA,sha256=CNSoUw9Zghtca3kstKJrMgQou_2lCE2otuJr9Usdmrg,1399
+lightning_pose_app-2.0.4.1.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
+lightning_pose_app-2.0.4.1.dist-info/RECORD,,

litpose_app/main.py

@@ -9,16 +9,18 @@ from textwrap import dedent
 
 import anyio
 import uvicorn
-from fastapi import FastAPI, HTTPException, APIRouter, Request
-from fastapi.responses import FileResponse
+from fastapi import FastAPI, HTTPException, APIRouter, Request, Depends
+from fastapi.responses import FileResponse, HTMLResponse
 from starlette import status
 from starlette.responses import Response
 from starlette.staticfiles import StaticFiles
 
 from . import deps
+from .rootconfig import RootConfig
 from .routes.labeler.multiview_autolabel import warm_up_anipose
 from .routes.videos import cleanup_old_uploads
 from .train_scheduler import _train_scheduler_process_target
+from .utils.check_for_upgrade import check_for_upgrade
 from .utils.file_response import file_response
 
 ## Setup logging
@@ -191,11 +193,42 @@ async def favicon():
 
 # Catch-all route. serve index.html.
 @app.get("/{full_path:path}")
-async def index():
-    return FileResponse(
-        Path(__file__).parent / "ngdist" / "ng_app" / "browser" / "index.html"
-    )
+def index(rc: RootConfig = Depends(deps.root_config)):
+    index_path = Path(__file__).parent / "ngdist" / "ng_app" / "browser" / "index.html"
+    content = index_path.read_text(encoding="utf-8")
+    analytics_tag_placeholder = "<!-- ANALYTICS_TAG_PLACEHOLDER -->"
+    assert analytics_tag_placeholder in content
+
+    # Skip analytics tag insertion if DO_NOT_TRACK is set
+    import os
+
+    if os.environ.get("DO_NOT_TRACK") != "1":
+        content = content.replace(analytics_tag_placeholder, rc.UMAMI_ANALYTICS_TAG)
+
+    return HTMLResponse(content=content)
 
 
 def run_app(host: str, port: int):
+    import os
+
+    if os.environ.get("DO_NOT_TRACK") != "1":
+        usage_message = dedent(
+            """
+            ================================================================================
+            📊  Lightning Pose App Usage Tracking
+            
+            To help us improve this tool and secure future grant funding, launching the app
+            in your browser will initiate privacy-focused web event tracking. We use Umami,
+            which ensures no personal data is ever collected or stored.
+            
+            To opt out, please restart the app with the environment variable DO_NOT_TRACK=1
+            For example: DO_NOT_TRACK=1 litpose run_app
+            ================================================================================
+            """
+        )
+        print(usage_message, file=sys.stderr)
+    else:
+        logger.info("App tracking is opted out, skipping analytics tag insertion")
+
+    check_for_upgrade()
     uvicorn.run(app, host=host, port=port, timeout_graceful_shutdown=1)

litpose_app/ngdist/ng_app/browser/index.html

@@ -301,6 +301,7 @@
   -webkit-font-smoothing: antialiased;
 }
 </style>
+    <!-- ANALYTICS_TAG_PLACEHOLDER -->
   <style>
 @layer properties;
 @layer theme, base, components, utilities;

litpose_app/rootconfig.py

@@ -16,6 +16,11 @@ class RootConfig(BaseSettings):
         default_factory=lambda data: data["LP_SYSTEM_DIR"] / "uploads"
     )
 
+    UMAMI_ANALYTICS_TAG: str = (
+        '<script defer src="https://cloud.umami.is/script.js" '
+        'data-website-id="ba9731f2-f746-4c4e-8fc6-d876e1556a6e"></script>'
+    )
+
     @model_validator(mode="after")
     def init_dirs_and_files(self):
         """Creates directories as needed on application startup"""

litpose_app/utils/check_for_upgrade.py

@@ -0,0 +1,49 @@
+import logging
+import os
+import sys
+from textwrap import dedent
+
+logger = logging.getLogger(__name__)
+
+
+def check_for_upgrade():
+    if os.environ.get("LP_IGNORE_UPGRADE") == "1":
+        logger.info(f"LP_IGNORE_UPGRADE=1 present, Skipping upgrade check ")
+        return
+
+    import json
+    import urllib.request
+    from packaging import version
+    import importlib.metadata
+
+    package_name = "lightning-pose-app"
+    try:
+        current_version = importlib.metadata.version(package_name)
+        # Fetch latest version from PyPI
+        with urllib.request.urlopen(
+            f"https://pypi.org/pypi/{package_name}/json", timeout=5
+        ) as response:
+            data = json.load(response)
+            latest_version = data["info"]["version"]
+
+        if version.parse(latest_version) > version.parse(current_version):
+            print(
+                dedent(
+                    f"""
+                {'-' * 80}
+                🚀 A new version of {package_name} is available: {latest_version} (current: {current_version})
+                
+                To upgrade, run:
+                    pip install --upgrade {package_name}
+
+                To skip this check in the future, use the LP_IGNORE_UPGRADE env variable:
+                    LP_IGNORE_UPGRADE=1 litpose run_app
+                {'-' * 80}
+            """
+                )
+            )
+
+            sys.exit(0)
+    except Exception as e:
+        # Don't block startup if PyPI is down or package isn't installed via pip
+        logger.debug(f"Upgrade check failed: {e}")