endoreg-db 0.4.5__py3-none-any.whl → 0.8.6.3__py3-none-any.whl

endoreg_db/admin.py

@@ -1,3 +1,92 @@
+from django.urls import path
 from django.contrib import admin
+from django.http import JsonResponse
+from endoreg_db.models import (
+    Patient,
+    Examination,
+    # PatientExamination,
+    Finding,
+    FindingClassification,
+    FindingClassificationChoice,
+    FindingIntervention,  #  Import Finding Interventions
+    PatientFindingIntervention,
+)
+# from endoreg_db.forms.patient_finding_intervention_form import (
+#     PatientFindingInterventionForm,
+# )
+from endoreg_db.forms.patient_form import PatientForm
 
-# Register your models here.
+
+@admin.register(Patient)
+class PatientAdmin(admin.ModelAdmin):
+    form = PatientForm
+    list_display = ("id", "first_name", "last_name", "dob", "center")
+    search_fields = ("first_name", "last_name", "email", "phone")
+    list_filter = ("dob", "center")
+    ordering = ("last_name",)
+
+
+@admin.register(Examination)
+class ExaminationAdmin(admin.ModelAdmin):
+    list_display = ("id", "name")
+    search_fields = ("name", )
+    list_filter = ("name",)
+    ordering = ("name",)
+
+
+class PatientFindingInterventionAdmin(admin.ModelAdmin):
+    change_list_template = "admin/patient_finding_intervention.html"
+
+    def changelist_view(self, request, extra_context=None):
+        """
+        Overrides the admin changelist view to provide additional context data for the template, including all patients, examinations, findings, classifications, and interventions relevant to patient finding interventions.
+        """
+        extra_context = {
+            "patients": Patient.objects.all(),
+            "examinations": Examination.objects.all(),
+            "findings": Finding.objects.all(),
+            "locations": FindingClassification.objects.filter(classification_types__name__iexact="location"),
+            "location_choices": FindingClassificationChoice.objects.none(),
+            "morphologies": FindingClassification.objects.filter(classification_types__name__iexact="morphology"),
+            "morphology_choices": FindingClassificationChoice.objects.none(),
+            "finding_interventions": FindingIntervention.objects.all(),
+        }
+        return super().changelist_view(request, extra_context=extra_context)
+
+    def get_location_choices_json(self, request):
+        """
+        Handles AJAX requests to retrieve location classification choices as JSON.
+        
+        Expects a "location" parameter in the GET request and returns a list of matching FindingClassificationChoice objects with their IDs and names. Returns an error message with appropriate HTTP status if the parameter is missing or an exception occurs.
+        """
+        location_id = request.GET.get("location")
+        if not location_id:
+            return JsonResponse({"error": "Location ID is required"}, status=400)
+
+        try:
+            choices = list(
+                FindingClassificationChoice.objects.filter(
+                    classifications__id=location_id,
+                    classifications__classification_types__name__iexact="location"
+                ).values("id", "name")
+            )
+            if not choices:
+                return JsonResponse([], safe=False)
+            return JsonResponse(choices, safe=False)
+        except Exception as e:
+            return JsonResponse({"error": str(e)}, status=500)
+
+    def get_urls(self):
+        """Register JSON endpoint inside Django Admin"""
+        urls = super().get_urls()
+        custom_urls = [
+            path(
+                "ajax/get-location-choices/",
+                self.admin_site.admin_view(self.get_location_choices_json),
+                name="ajax_get_location_choices",
+            ),
+        ]
+        return custom_urls + urls
+
+
+admin.site.register(PatientFindingIntervention, PatientFindingInterventionAdmin)

endoreg_db/api_urls.py

@@ -0,0 +1,4 @@
+# Export raw API URL patterns so the project-level router can add the single `/api/` prefix.
+from endoreg_db.urls import urlpatterns as api_urlpatterns
+
+urlpatterns = api_urlpatterns

endoreg_db/apps.py

@@ -4,3 +4,15 @@ from django.apps import AppConfig
 class EndoregDbConfig(AppConfig):
     default_auto_field = 'django.db.models.BigAutoField'
     name = 'endoreg_db'
+    
+    def ready(self):
+        """
+        Performs application startup tasks when the Django app is fully loaded.
+        
+        This method imports media-related model modules to ensure they are registered
+        and ready for use when the application starts.
+        """
+        import endoreg_db.models.media.video
+        import endoreg_db.models.media.frame
+        import endoreg_db.models.media.pdf
+        pass

endoreg_db/assets/dummy_model.ckpt

@@ -0,0 +1 @@
+dummy weights content

endoreg_db/codemods/readme.md

@@ -0,0 +1,88 @@
+### install
+```
+uv add bowler fissix
+```
+
+### Dry-run (preview only)
+```
+python endoreg_db/codemods/rename_datetime_fields.py
+```
+
+### Run it (models only):
+```
+python endoreg_db/codemods/rename_datetime_fields.py --yes
+```
+
+### Verify what changed
+```
+git status
+git diff --name-only
+git diff
+```
+
+### Then check for any leftover legacy names in models (excluding migration history):
+```
+grep -RIn --exclude-dir='migrations' \
+  -E '\b(date_created|date_modified|last_update|date_updated|createdOn|updatedOn|modified_at|lastModified)\b' \
+  endoreg_db/models || true
+
+grep -RIn --exclude-dir='migrations' -E '\b(date_created|date_modified|last_update|date_updated|createdOn|updatedOn|modified_at|lastModified)\b' endoreg_db/models || true
+```
+
+### Fix F(..) expressions (if any)
+```
+# date_created -> created_at
+grep -Rl --exclude-dir='migrations' "F('date_created')" endoreg_db/models | xargs -r sed -i "s/F('date_created')/F('created_at')/g"
+grep -Rl --exclude-dir='migrations' 'F("date_created")' endoreg_db/models | xargs -r sed -i 's/F("date_created")/F("created_at")/g'
+
+# date_modified -> updated_at
+grep -Rl --exclude-dir='migrations' "F('date_modified')" endoreg_db/models | xargs -r sed -i "s/F('date_modified')/F('updated_at')/g"
+grep -Rl --exclude-dir='migrations' 'F("date_modified")' endoreg_db/models | xargs -r sed -i 's/F("date_modified")/F("updated_at")/g'
+```
+
+
+### Example: Fix f-strings / attribute refs (seen in raw_pdf.py, video.py)
+```
+# self.date_created -> self.created_at
+grep -Rl --exclude-dir='migrations' 'self\.date_created' endoreg_db/models | xargs -r sed -i 's/self\.date_created/self.created_at/g'
+# self.date_modified -> self.updated_at
+grep -Rl --exclude-dir='migrations' 'self\.date_modified' endoreg_db/models | xargs -r sed -i 's/self\.date_modified/self.updated_at/g'
+```
+
+### Fix string literals in .values(), .order_by(), dict keys
+```
+# double-quoted keys
+grep -Rl --exclude-dir='migrations' --include="*.py" '"date_created"' endoreg_db/models | xargs -r sed -i 's/"date_created"/"created_at"/g'
+grep -Rl --exclude-dir='migrations' --include="*.py" '"date_modified"' endoreg_db/models | xargs -r sed -i 's/"date_modified"/"updated_at"/g'
+grep -Rl --exclude-dir='migrations' --include="*.py" '"last_update"'  endoreg_db/models | xargs -r sed -i 's/"last_update"/"updated_at"/g'
+grep -Rl --exclude-dir='migrations' --include="*.py" '"date_updated"' endoreg_db/models | xargs -r sed -i 's/"date_updated"/"updated_at"/g'
+
+# single-quoted keys
+grep -Rl --exclude-dir='migrations' --include="*.py" "'date_created'" endoreg_db/models | xargs -r sed -i "s/'date_created'/'created_at'/g"
+grep -Rl --exclude-dir='migrations' --include="*.py" "'date_modified'" endoreg_db/models | xargs -r sed -i "s/'date_modified'/'updated_at'/g"
+grep -Rl --exclude-dir='migrations' --include="*.py" "'last_update'"  endoreg_db/models | xargs -r sed -i "s/'last_update'/'updated_at'/g"
+grep -Rl --exclude-dir='migrations' --include="*.py" "'date_updated'" endoreg_db/models | xargs -r sed -i "s/'date_updated'/'updated_at'/g"
+
+# order_by("-...") style
+grep -Rl --exclude-dir='migrations' --include="*.py" '"-date_created"' endoreg_db/models | xargs -r sed -i 's/"-date_created"/"-created_at"/g'
+grep -Rl --exclude-dir='migrations' --include="*.py" "'-date_created'" endoreg_db/models | xargs -r sed -i "s/'-date_created'/'-created_at'/g"
+grep -Rl --exclude-dir='migrations' --include="*.py" '"-date_modified"' endoreg_db/models | xargs -r sed -i 's/"-date_modified"/"-updated_at"/g'
+grep -Rl --exclude-dir='migrations' --include="*.py" "'-date_modified'" endoreg_db/models | xargs -r sed -i "s/'-date_modified'/'-updated_at'/g"
+```
+
+
+### Re-scan to confirm
+```
+grep -RIn --exclude-dir='migrations' \
+  -E '\b(date_created|date_modified|last_update|date_updated|createdOn|updatedOn|modified_at|lastModified)\b' \
+  endoreg_db/models || true
+```
+
+
+### Migrations
+```
+python manage.py makemigrations endoreg_db
+python manage.py migrate
+```
+
+

endoreg_db/codemods/rename_datetime_fields.py

@@ -0,0 +1,92 @@
+# endoreg_db/codemods/rename_datetime_fields.py
+from bowler import Query
+from pathlib import Path
+import argparse, yaml, sys
+
+# Paths
+BASE = Path(__file__).resolve().parents[1]  # .../endoreg_db
+RENAMES_YML = BASE / "renames.yml"
+DEFAULT_TARGETS = ["endoreg_db/models"]  # safer default
+EXCLUDE_DIR_NAMES = {"migrations", "__pycache__"}
+
+def load_renames():
+    if not RENAMES_YML.exists():
+        print(f"ERROR: renames.yml not found at {RENAMES_YML}", file=sys.stderr)
+        sys.exit(2)
+    data = yaml.safe_load(RENAMES_YML.read_text()) or {}
+    if not isinstance(data, dict) or not data:
+        print("ERROR: renames.yml is empty or not a mapping.", file=sys.stderr)
+        sys.exit(2)
+    return data
+
+def iter_python_targets(paths):
+    """Yield *.py files under given paths, excluding migrations and caches."""
+    for p in map(Path, paths):
+        if p.is_file() and p.suffix == ".py":
+            if not any(part in EXCLUDE_DIR_NAMES for part in p.parts):
+                yield str(p)
+        elif p.is_dir():
+            for f in p.rglob("*.py"):
+                if any(part in EXCLUDE_DIR_NAMES for part in f.parts):
+                    continue
+                yield str(f)
+
+def build_query(files):
+    # Bowler can take a list of files; we’ve already filtered them
+    return Query(list(files))
+
+def main(argv=None):
+    parser = argparse.ArgumentParser(
+        description="Rename legacy datetime fields to standardized names."
+    )
+    parser.add_argument(
+        "targets",
+        nargs="*",
+        help="Files/dirs to process. Default: endoreg_db/models",
+    )
+    parser.add_argument(
+        "--yes",
+        action="store_true",
+        help="Apply changes (write). Omit for a dry run.",
+    )
+    parser.add_argument(
+        "--silent",
+        action="store_true",
+        help="Reduce output verbosity.",
+    )
+    args = parser.parse_args(argv)
+
+    targets = args.targets or DEFAULT_TARGETS
+    if args.targets == []:
+        print(
+            "NOTICE: Using default target 'endoreg_db/models'. "
+            "Pass explicit paths to broaden scope.",
+            file=sys.stderr,
+        )
+
+    files = list(iter_python_targets(targets))
+    if not files:
+        print("No Python files found to process.", file=sys.stderr)
+        return 0
+
+    renames = load_renames()
+    q = build_query(files)
+
+    # Build transforms
+    for old, new in renames.items():
+        # obj.date_created  -> obj.created_at
+        q.select_attribute(old).rename(new)
+        # LHS or bare names: date_created = models.DateTimeField(...)
+        q.select_var(old).rename(new)
+
+    # Execute (dry-run by default)
+    q.execute(write=args.yes, silent=args.silent)
+    if not args.yes:
+        print(
+            "\nDry run complete. Re-run with --yes to apply changes.",
+            file=sys.stderr,
+        )
+    return 0
+
+if __name__ == "__main__":
+    raise SystemExit(main())

endoreg_db/config/env.py

@@ -0,0 +1,101 @@
+"""
+Centralized environment configuration for EndoReg-DB.
+
+This module is the single place to read environment variables and .env files.
+It avoids loading .env during pytest, and provides typed helpers.
+No Django imports here to prevent early settings configuration.
+"""
+from __future__ import annotations
+
+from pathlib import Path
+import os
+from typing import Optional, Dict, Any
+
+# Detect pytest early to avoid loading .env in test runs
+IS_PYTEST = bool(os.environ.get("PYTEST_CURRENT_TEST"))
+
+# Compute repository BASE_DIR (repo root). This file is endoreg_db/config/env.py.
+BASE_DIR = Path(__file__).resolve().parents[2]
+
+# Optional: load .env only when not under pytest
+_DOTENV_LOADED = False
+try:
+    if not IS_PYTEST:
+        import dotenv  # type: ignore
+        dotenv.load_dotenv()
+        _DOTENV_LOADED = True
+except Exception:
+    # dotenv is optional, ignore errors
+    _DOTENV_LOADED = False
+
+
+def _get(key: str, default: Optional[str] = None) -> Optional[str]:
+    return os.environ.get(key, default)
+
+
+def env_str(key: str, default: str = "") -> str:
+    val = _get(key)
+    return val if val is not None else default
+
+
+def env_bool(key: str, default: bool = False) -> bool:
+    val = _get(key)
+    if val is None:
+        return default
+    return str(val).strip().lower() in {"1", "true", "yes", "on"}
+
+
+def env_int(key: str, default: int = 0) -> int:
+    val = _get(key)
+    if val is None:
+        return default
+    try:
+        return int(str(val).strip())
+    except Exception:
+        return default
+
+
+def env_path(key: str, default_relative: str) -> Path:
+    """Return an absolute path. If env is relative, resolve under BASE_DIR."""
+    val = _get(key)
+    if not val:
+        p = BASE_DIR / default_relative
+    else:
+        p = Path(val)
+        if not p.is_absolute():
+            p = (BASE_DIR / p).resolve()
+    return p
+
+
+def snapshot() -> Dict[str, Any]:
+    """Return a snapshot of relevant config for debugging/logging."""
+    keys = [
+        # Core
+        "DJANGO_SETTINGS_MODULE",
+        "TIME_ZONE",
+        # Paths
+        "STORAGE_DIR",
+        "ASSET_DIR",
+        "STATIC_URL",
+        "MEDIA_URL",
+        # Dev DB
+        "DEV_DB_ENGINE",
+        "DEV_DB_NAME",
+        # Test DB
+        "TEST_DB_ENGINE",
+        "TEST_DB_NAME",
+        "TEST_DB_FILE",
+        # Flags
+        "RUN_VIDEO_TESTS",
+        "SKIP_EXPENSIVE_TESTS",
+    ]
+    data: Dict[str, Any] = {k: os.environ.get(k) for k in keys}
+    data.update({
+        "IS_PYTEST": IS_PYTEST,
+        "DOTENV_LOADED": _DOTENV_LOADED,
+        "BASE_DIR": str(BASE_DIR),
+    })
+    return data
+
+# Back-compat short aliases used by settings modules
+ENV = os.environ.get

endoreg_db/data/__init__.py

@@ -3,19 +3,60 @@ from pathlib import Path
 DATA_DIR = Path(__file__).parent
 
 INFORMATION_DATA_DIR = DATA_DIR / "information_source"
+INFORMATION_SOURCE_TYPE_DATA_DIR = DATA_DIR / "information_source_type"
 UNIT_DATA_DIR = DATA_DIR / "unit"
 CENTER_DATA_DIR = DATA_DIR / "center"
 EXAMINATION_DATA_DIR = DATA_DIR / "examination"
-LABEL_DATA_DIR = DATA_DIR / "label"
-MODEL_TYPE_DATA_DIR = DATA_DIR / "model_type"
+MODEL_TYPE_DATA_DIR = DATA_DIR / "ai_model_type"
 ENDOSCOPY_PROCESSOR_DATA_DIR = DATA_DIR / "endoscopy_processor"
 ENDOSCOPE_TYPE_DATA_DIR = DATA_DIR / "endoscope_type"
+ENDOSCOPE_DATA_DIR = DATA_DIR / "endoscope"
 ACTIVE_MODEL_DATA_DIR = DATA_DIR / "active_model"
 PROFESSION_DATA_DIR = DATA_DIR / "profession"
 PDF_TYPE_DATA_DIR = DATA_DIR / "pdf_type"
 REPORT_READER_FLAG_DATA_DIR = DATA_DIR / "report_reader_flag"
 EVENT_DATA_DIR = DATA_DIR / "event"
 GENDER_DATA_DIR = DATA_DIR / "gender"
+ORGAN_DATA_DIR = DATA_DIR / "organ"
+NAMES_FIRST_DATA_DIR = DATA_DIR / "names_first"
+NAMES_LAST_DATA_DIR = DATA_DIR / "names_last"
+
+
+# Contraindication
+CONTRAINDICATION_DATA_DIR = DATA_DIR / "contraindication"
+
+# Examination
+EXAMINATION_INDICATION_DATA_DIR = DATA_DIR / "examination_indication"
+EXAMINATION_INDICATION_CLASSIFICATION_DATA_DIR = (
+    DATA_DIR / "examination_indication_classification"
+)
+EXAMINATION_INDICATION_CLASSIFICATION_CHOICE_DATA_DIR = (
+    DATA_DIR / "examination_indication_classification_choice"
+)
+
+# Finding
+FINDING_TYPE_DATA_DIR = DATA_DIR / "finding_type"
+FINDING_DATA_DIR = DATA_DIR / "finding"
+FINDING_LOCATION_CLASSIFICATION_DATA_DIR = DATA_DIR / "finding_location_classification"
+FINDING_LOCATION_CLASSIFICATION_CHOICE_DATA_DIR = (
+    DATA_DIR / "finding_location_classification_choice"
+)
+FINDING_MORPHOLOGY_CLASSIFICATION_DATA_DIR = (
+    DATA_DIR / "finding_morphology_classification"
+)
+FINDING_MORPHOLOGY_CLASSIFICATION_CHOICE_DATA_DIR = (
+    DATA_DIR / "finding_morphology_classification_choice"
+)
+FINDING_MORPGOLOGY_CLASSIFICATION_TYPE_DATA_DIR = (
+    DATA_DIR / "finding_morphology_classification_type"
+)
+FINDING_INTERVETION_DATA_DIR = DATA_DIR / "finding_intervention"
+FINIDNG_INTERVENTION_TYPE_DATA_DIR = DATA_DIR / "finding_intervention_type"
+
+
+FINDING_CLASSIFICATION_TYPE_DATA_DIR = DATA_DIR / "finding_classification_type"
+FINDING_CLASSIFICATION_DATA_DIR = DATA_DIR / "finding_classification"
+FINDING_CLASSIFICATION_CHOICE_DATA_DIR = DATA_DIR / "finding_classification_choice"
 
 # Green Endoscopy Dashboard Stuffs
 CENTER_RESOURCE_DATA_DIR = DATA_DIR / "center_resource"
@@ -40,7 +81,9 @@ MEDICATION_INDICATION_TYPE_DATA_DIR = DATA_DIR / "medication_indication_type"
 DISTRIBUTION_BASE_DIR = DATA_DIR / "distribution"
 DISTRIBUTION_SINGLE_CATEGORICAL_DATA_DIR = DISTRIBUTION_BASE_DIR / "single_categorical"
 DISTRIBUTION_NUMERIC_DATA_DIR = DISTRIBUTION_BASE_DIR / "numeric"
-DISTRIBUTION_MULTIPLE_CATEGORICAL_DATA_DIR = DISTRIBUTION_BASE_DIR / "multiple_categorical"
+DISTRIBUTION_MULTIPLE_CATEGORICAL_DATA_DIR = (
+    DISTRIBUTION_BASE_DIR / "multiple_categorical"
+)
 DISTRIBUTION_DATE_DATA_DIR = DISTRIBUTION_BASE_DIR / "date"
 
 # G-Play Stuffs
@@ -60,6 +103,25 @@ AGL_SERVICE_DATA_DIR = DATA_DIR / "agl_service"
 # LOGS
 LOG_TYPE_DATA_DIR = DATA_DIR / "log_type"
 
+# AI STUFFS
+AI_MODEL_DATA_DIR = DATA_DIR / "ai_model"
+AI_MODEL_TYPE_DATA_DIR = DATA_DIR / "ai_model_type"
+AI_MODEL_META_DATA_DIR = DATA_DIR / "ai_model_meta"
+LABEL_DATA_DIR = DATA_DIR / "ai_model_label"
+VIDEO_SEGMENTATION_LABEL_DATA_DIR = DATA_DIR / "ai_model_video_segmentation_label"
+VIDEO_SEGMENTATION_LABELSET_DATA_DIR = DATA_DIR / "ai_model_video_segmentation_labelset"
+
+# RISK
+RISK_DATA_DIR = DATA_DIR / "risk"
+RISK_TYPE_DATA_DIR = DATA_DIR / "risk_type"
+
+# REQUIREMENT
+REQUIREMENT_DATA_DIR = DATA_DIR / "requirement"
+REQUIREMENT_OPERATOR_DATA_DIR = DATA_DIR / "requirement_operator"
+REQUIREMENT_SET_DATA_DIR = DATA_DIR / "requirement_set"
+REQUIREMENT_SET_TYPE_DATA_DIR = DATA_DIR / "requirement_set_type"
+REQUIREMENT_TYPE_DATA_DIR = DATA_DIR / "requirement_type"
+
 # To Do
 LAB_VALUE_DATA_DIR = DATA_DIR / "lab_value"
 PATIENT_LAB_SAMPLE_TYPE_DATA_DIR = DATA_DIR / "patient_lab_sample_type"
@@ -69,4 +131,14 @@ MEDICATION_INTAKE_TIME_DATA_DIR = DATA_DIR / "medication_intake_time"
 
 RULE_TYPE_DATA_DIR = DATA_DIR / "rule_type"
 RULE_ATTRIBUTE_DTYPE_DATA_DIR = DATA_DIR / "rule_attribute_dtype"
-RULESET_DATA_DIR = DATA_DIR / "ruleset"
+RULESET_DATA_DIR = DATA_DIR / "ruleset"
+
+LX_PERMISSION_DATA_DIR = DATA_DIR / "lx_permission"
+LX_CLIENT_TAG_DATA_DIR = DATA_DIR / "lx_client_tag"
+LX_CLIENT_TYPE_DATA_DIR = DATA_DIR / "lx_client_type"
+
+QUALIFICATION_DATA_DIR = DATA_DIR / "qualification"
+
+TAG_DATA_DIR = DATA_DIR / "tag"
+
+EXAMINATION_REQUIREMENT_SET_DATA_DIR = DATA_DIR / "examination_requirement_set"

endoreg_db/data/ai_model/data.yaml

@@ -0,0 +1,7 @@
+- model: endoreg_db.ai_model
+  fields:
+    name: "image_multilabel_classification_colonoscopy_default"
+    description: "Image multilabel classification model"
+    model_type: "multilabel_classification"
+    model_subtype: "multilabel"
+    video_segmentation_labelset: "colonoscopy_video_segmentation_raw"

endoreg_db/data/{label → ai_model_label}/label/data.yaml

@@ -7,7 +7,7 @@
 
 - model: endoreg_db.Label
   fields:
-    name: "ileocecal_valve"
+    name: "ileocaecalvalve"
     label_type: "classification"
 
 - model: endoreg_db.Label
@@ -60,3 +60,29 @@
     name: "outside"
     label_type: "classification"
 
+- model: endoreg_db.Label
+  fields:
+    name: "diverticule"
+    label_type: "classification"
+
+- model: endoreg_db.Label
+  fields:
+    name: "grasper"
+    label_type: "classification"
+
+- model: endoreg_db.Label
+  fields:
+    name: "nbi"
+    label_type: "classification"
+
+# needle
+- model: endoreg_db.Label
+  fields:
+    name: "needle"
+    label_type: "classification"
+
+# snare
+- model: endoreg_db.Label
+  fields:
+    name: "snare"
+    label_type: "classification"

endoreg_db/data/ai_model_label/label/polyp_classification.yaml

@@ -0,0 +1,52 @@
+# Paris
+- model: endoreg_db.Label
+  fields:
+    name: "paris_1s"
+    label_type: "classification"
+
+- model: endoreg_db.Label
+  fields:
+    name: "paris_1p"
+    label_type: "classification"
+
+- model: endoreg_db.Label
+  fields:
+    name: "paris_2a"
+    label_type: "classification"
+
+- model: endoreg_db.Label
+  fields:
+    name: "paris_2b"
+    label_type: "classification"
+
+- model: endoreg_db.Label
+  fields:
+    name: "paris_2c"
+    label_type: "classification"
+
+- model: endoreg_db.Label
+  fields:
+    name: "paris_3"
+    label_type: "classification"
+
+# NICE Classification
+- model: endoreg_db.Label
+  fields:
+    name: "nice_1"
+    label_type: "classification"
+
+- model: endoreg_db.Label
+  fields:
+    name: "nice_2a"
+    label_type: "classification"
+
+- model: endoreg_db.Label
+  fields:
+    name: "nice_2b"
+    label_type: "classification"
+
+- model: endoreg_db.Label
+  fields:
+    name: "nice_3"
+    label_type: "classification"
+    

endoreg_db/data/ai_model_label/label-set/data.yaml

@@ -0,0 +1,40 @@
+- model: endoreg_db.label_set
+  fields:
+    name: "multilabel_classification_colonoscopy_default"
+    version: 1
+    labels:
+      [
+        "appendix",
+        "blood",
+        "diverticule",
+        "grasper",
+        "ileocaecalvalve",
+        "ileum",
+        "low_quality",
+        "nbi",
+        "needle",
+        "outside",
+        "polyp",
+        "snare",
+        "water_jet",
+        "wound",
+      ]
+
+- model: endoreg_db.label_set
+  fields:
+    name: "multilabel_classification_colonoscopy_default"
+    version: 2
+    labels:
+      [
+        "appendix",
+        "blood",
+        "ileocaecalvalve",
+        "ileum",
+        "low_quality",
+        "instrument",
+        "digital_chromo_endoscopy",
+        "outside",
+        "polyp",
+        "water_jet",
+        "wound",
+      ]

endoreg_db/data/ai_model_label/label-set/polyp_classifications.yaml

@@ -0,0 +1,25 @@
+- model: endoreg_db.LabelSet
+  fields:
+    name: "endoscopy_paris_classification"
+    version: 1
+    labels:
+      [
+        "paris_1s",
+        "paris_1p",
+        "paris_2a",
+        "paris_2b",
+        "paris_2c",
+        "paris_3",
+      ]
+
+- model: endoreg_db.LabelSet
+  fields:
+    name: "endoscopy_nice_classification"
+    version: 1
+    labels:
+      [
+        "nice_1",
+        "nice_2a",
+        "nice_2b",
+        "nice_3",
+      ]