endoreg-db 0.6.4__py3-none-any.whl → 0.8.2__py3-none-any.whl

endoreg_db/urls/upload.py

@@ -0,0 +1,20 @@
+from django.urls import path
+
+from endoreg_db.views import (
+    UploadFileView,
+    UploadStatusView,
+)
+
+urlpatterns = [
+    # Upload endpoints
+    path(
+        'upload/', 
+        UploadFileView.as_view(), 
+        name='video_upload'
+    ),
+    path(
+        'upload/<uuid:id>/status', 
+        UploadStatusView.as_view(), 
+        name='upload_status'
+    ),
+]

endoreg_db/urls/video.py

@@ -0,0 +1,61 @@
+from django.urls import path
+
+from endoreg_db.views import (
+    SensitiveMetaDetailView,
+    VideoLabelView,
+    # Note: VideoStreamView moved to modern media framework. See: endoreg_db/urls/media.py
+    # Note: All Video Correction Views moved to modern media framework. See: endoreg_db/urls/media.py
+)
+
+url_patterns = [
+    # Video Label segments API endpoint
+    # GET /api/videos/<int:video_id>/labels/<str:label_name>/
+    # Retrieves time segments and frame predictions for a specific label on a video
+    path(
+        'videos/<int:video_id>/labels/<str:label_name>/', 
+        VideoLabelView.as_view(), 
+        name='video_label_segments'
+    ),
+    
+    # ---------------------------------------------------------------------------------------
+    # VIDEO STREAMING - MOVED TO MODERN MEDIA FRAMEWORK
+    #
+    # Video streaming endpoint has been migrated to the media framework
+    # as of October 14, 2025. Please use the new endpoints:
+    #
+    # OLD → NEW:
+    # GET /api/videostream/<pk>/              → GET /api/media/videos/<pk>/
+    # GET /api/videostream/<pk>/              → GET /api/media/videos/<pk>/stream/
+    #
+    # See: endoreg_db/urls/media.py for new URL registrations
+    # ---------------------------------------------------------------------------------------
+    
+    # Note: Video Re-import moved to modern media framework
+    # See: endoreg_db/urls/media.py - POST /api/media/videos/<int:pk>/reimport/
+    
+    # Video Sensitive Meta endpoints (for video anonymization)
+    # GET /api/video/sensitivemeta/<int:sensitive_meta_id>/
+    # PATCH /api/video/sensitivemeta/<int:sensitive_meta_id>/
+    path(
+        'video/sensitivemeta/<int:sensitive_meta_id>/', 
+        SensitiveMetaDetailView.as_view(), 
+        name='video_sensitive_meta_detail'
+    ),
+    
+    # ---------------------------------------------------------------------------------------
+    # VIDEO CORRECTION API ENDPOINTS - MOVED TO MODERN MEDIA FRAMEWORK
+    #
+    # All video correction endpoints have been migrated to the modern media framework
+    # as of October 14, 2025. Please use the new endpoints:
+    #
+    # OLD → NEW:
+    # GET  /api/video-metadata/<id>/              → GET  /api/media/videos/<pk>/metadata/
+    # GET  /api/video-processing-history/<id>/    → GET  /api/media/videos/<pk>/processing-history/
+    # POST /api/video-analyze/<id>/               → POST /api/media/videos/<pk>/analyze/
+    # POST /api/video-apply-mask/<id>/            → POST /api/media/videos/<pk>/apply-mask/
+    # POST /api/video-remove-frames/<id>/         → POST /api/media/videos/<pk>/remove-frames/
+    # POST /api/video-reprocess/<id>/             → POST /api/media/videos/<pk>/reprocess/
+    #
+    # See: endoreg_db/urls/media.py for new URL registrations
+    # ---------------------------------------------------------------------------------------
+]

endoreg_db/urls.py

@@ -1,286 +1,9 @@
-from django.urls import path, include
-from rest_framework.routers import DefaultRouter
-from .views.patient_views import (
-    PatientViewSet,
-    start_examination,
-    get_location_choices,
-    get_morphology_choices, 
-)
-from .views.csrf import csrf_token_view
-# endoreg_db_production/endoreg_db/urls.py
-from endoreg_db.views.views import (
-    VideoView,
-    keycloak_login,
-    keycloak_callback,
-    public_home,
-)#from .views.feature_selection_view import FetchSingleFramePredictionView // its implemented in ando-ai other project need to add here
-from .views.video_segmentation_views import VideoView, VideoLabelView,UpdateLabelSegmentsView
-from .views.views_for_timeline import video_timeline_view
-from .views.raw_video_meta_validation_views import VideoFileForMetaView, VideoFileForMetaView
-from .views.raw_pdf_meta_validation_views import PDFFileForMetaView
-from .views.raw_pdf_meta_validation_views import UpdateSensitiveMetaView
-from .views.raw_pdf_anony_text_validation_views import RawPdfAnonyTextView, UpdateAnonymizedTextView
-router = DefaultRouter()
-router.register(r'patients', PatientViewSet)
+from django.urls import include, path
 
+# Import the package-level URL patterns (defined in endoreg_db/urls/__init__.py)
+from endoreg_db.urls import urlpatterns as api_urlpatterns
 
+# Expose API under /api/
 urlpatterns = [
-    path('start-examination/', start_examination, name="start_examination"),
-    path('get-location-choices/<int:location_id>/', get_location_choices, name="get_location_choices"),
-    path('get-morphology-choices/<int:morphology_id>/', get_morphology_choices, name="get_morphology_choices"),
-    path('api/', include(router.urls)),
-    path('api/conf/', csrf_token_view, name='csrf_token'),
-
-
-#--------------------------------------START : VIDEO SEGMENTATION END POINTS--------------------------------------
-
-       # The dropdown contains video names and their corresponding IDs, which are retrieved from the database(RawVideoFile). Additionally, this route(api/videos) also fetches labels along with their names and IDs from the label table.
-       # We will modify this implementation later as per our requirements.
-        #Based on the selected video ID, relevant data will be fetched from the database, and the video, along with its associated details, 
-        # will be displayed on the frontend using route(api/video/<int:video_id>/).
-        # once label is selected from the dropdown,using its name, details can be fetched from rawvideofile using route("api/video/<int:video_id>/label/<str:label_name>/) 
-        #If editing is required, a form will be available for each label. This form dynamically updates when the selected label changes. It will display all segments associated with the selected label, each with a delete option. Below these segments, there may be a button for adding more segments.
-        #If any values in the form are modified, the updated data will be saved in the dafrom .views import VideoView, keycloak_login, keycloak_callbacktabase table.
-    
-    
-    
-    
-    # Fetch all videos for the dropdown
-    # Fetch all available videos (for the video selection dropdown in frontend)
-    # Usage in frontend: Call this API to get a list of available videos.
-    # Example request: GET /api/videos/
-    # Response: Returns a JSON list of videos (id, file path, video_url, sequences, label_names).
-    # it also fetch the label from teh label table
-
-    # for kaycloak:-
-    # /api/videos/ , protected API
-    # /login/ , sends user to Keycloak login page
-    # /login/callback/ , where Keycloak sends the user after login
-    path('', public_home, name='public_home'),
-    path('api/videos/', VideoView.as_view(), name='video_list'),
-    path('login/', keycloak_login, name='keycloak_login'),
-    path('login/callback/', keycloak_callback, name='keycloak_callback'),
-    
-    #need to change this route
-    #path('api/prediction/', FetchSingleFramePredictionView.as_view(), name='fetch-single-frame-prediction'),#.as_view() converts the class-based view into a function that Django can use for routing.
-    
-    # Fetch video details + labels based on selected video
-    # Usage in frontend: Call this when a user selects a video from the dropdown.
-    # Example request: GET /api/video/1/
-    # Response: Returns metadata, file path, video URL, sequences, and available label names.
-    # Frontend should use this api for `label_names`, to display label selection (dropdown or tabs).
-    path("api/video/<int:video_id>/", VideoView.as_view(), name="video_handler"),
-    
-
-    # Fetch time segments along with frame predictions and filenames for a selected label
-    # 
-    # **Usage in frontend:**
-    # - Call this API when a user selects a label from the dropdown in the timeline.
-    # - The API returns time segments (start & end times in seconds) where the label appears in the video.
-    # - Each segment includes frame-wise predictions and the corresponding frame filenames.
-    #
-    # **Example request:**
-    #   GET /api/video/1/label/outside/
-    #
-    # **Example response:**
-    # {
-    #     "label": "outside",
-    #     "time_segments": [
-    #         {   "segment_start": 0,
-     #       "segment_end": 463,
-    #             "start_time": 0.0,
-    #             "end_time": 9.26,
-    #             "frames": {
-    #                 "0": {
-    #                     "frame_filename": "frame_0000000.jpg",
-    #                     "frame_file_path": "/home/admin/test-data/video_frames/abc123/frame_0000000.jpg",
-    #                     "predictions": {
-    #                         "appendix": 0.0150,
-    #                         and others
-    #                     }
-    #                 },
-    #                 "1": {
-    #                     "frame_filename": "frame_0000001.jpg",
-    #                     "frame_file_path": "/home/admin/test-data/video_frames/abc123/frame_0000001.jpg",
-    #                     "predictions": {
-    #                         "appendix": 0.0124,
-    #                         and otehrs
-    #                     }}}}]}
-    #                 
-    # start_time and end_time are in seconds, converted from frame indices.
-    # frames contains all frames in the segment with:
-    # frame_filename (e.g., frame_0000001.jpg)
-    # frame_file_path (full path for easy access)
-    #  predictions (frame-wise probability scores for each category)
-    #we need label name for this not id as we donot have id in rawvideofile table
-    path("api/video/<int:video_id>/label/<str:label_name>/", VideoLabelView.as_view(), name="video_label_times"),
-
-    # Update or create label segments for a specific video and label
-    # - Call this API when a user updates the start and end frames of a label segment(adding new , deleting or changing the values).
-    # - The API will:
-    #   - **Update** existing segments if they match the given `video_id` and `label_id`.
-    #   - **Create** new segments if they do not already exist.
-    #   - **Delete** segments from the database if they were removed from the frontend.
-    # - This ensures that the database reflects exactly what is sent from the frontend.
-    #
-    # **Example request (Updating segments for label ID 12 in video ID 1):**
-    #   PUT /api/video/1/label/12/update_segments/
-    #
-    # **Request body (JSON):**
-    # {
-    #     "video_id": 1,
-    #     "label_id": 12,
-    #     "segments": [
-    #         {
-    #             "start_frame_number": 0,
-    #             "end_frame_number": 463
-    #         },{
-    #             "start_frame_number": 500,
-    #             "end_frame_number": 700
-    #         }]}
-    #
-    # **Expected response (If successful):**
-    # {
-    #     "updated_segments": [
-    #         {
-    #             "id": 1,
-    #             "video_id": 1,
-    #             "label_id": 12,
-    #             "start_frame_number": 0,
-    #             "end_frame_number": 463
-    #         }
-    #     ],
-    #     "new_segments": [
-    #         {
-    #             "id": 2,
-    #             "video_id": 1,
-    #             "label_id": 12,
-    #             "start_frame_number": 500,
-    #             "end_frame_number": 700
-    #         }
-    #     ],
-    #     "deleted_segments": 1  # Number of deleted segments if any were removed
-    # }
- 
-    # **Frontend Integration:**
-    # - Vue.js should send this request when the user **saves changes** to segments.
-    # - The **"Add Segment"** button should allow users to add new segments dynamically.
-    # - The **"Delete Segment"** button should remove segments from the list before submitting.
-    # - The request should be triggered via an `axios.put` call:
-    #
-    path("api/video/<int:video_id>/label/<int:label_id>/update_segments/", UpdateLabelSegmentsView.as_view(), name="update_label_segments"),
-
-
-#----------------------------------START : SENSITIVE META AND RAWVIDEOFILE VIDEO PATIENT DETAILS-------------------------------
-
-    # API Endpoint for fetching video metadata or streaming the next available video
-    # This endpoint is used by the frontend to fetch:
-    #  - The first available video if `last_id` is NOT provided.
-    #  - The next available video where `id > last_id` if `last_id` is provided.
-    #  - If `Accept: application/json` is set in headers, it returns video metadata as JSON.
-    #  - If no videos are available, it returns {"error": "No more videos available."}.
-    #  const url = lastId ? `http://localhost:8000/api/video/meta/?last_id=${lastId}` : "http://localhost:8000/api/video/meta/";
-    path("api/video/sensitivemeta/", VideoFileForMetaView.as_view(), name="video_meta"),  # Single endpoint for both first and next video    
-
-
-
-    # This API endpoint allows updating specific patient details (SensitiveMeta)
-    # linked to a video. It is used to correct or modify the patient's first name,
-    # last name, date of birth, and examination date.
-    # Fetch video metadata and update patient details
-    # The frontend should send a JSON request body like this:
-    # {
-    #     "sensitive_meta_id": 2,          # The ID of the SensitiveMeta entry (REQUIRED)
-    #     "patient_first_name": "John",    # New first name (REQUIRED, cannot be empty)
-    #     "patient_last_name": "Doe",      # New last name (REQUIRED, cannot be empty)
-    #     "patient_dob": "1985-06-15",     # New Date of Birth (REQUIRED, format YYYY-MM-DD)
-    #     "examination_date": "2024-03-20" # New Examination Date (OPTIONAL, format YYYY-MM-DD)
-    # }
-    # - The frontend sends a PATCH request to this endpoint with updated patient data.
-    # - The backend validates the input using the serializer (`SensitiveMetaUpdateSerializer`).
-    # - If validation passes, the patient information is updated in the database.
-    # - If there are errors (e.g., missing fields, incorrect date format), 
-    #   the API returns structured error messages.
-    path("api/video/update_sensitivemeta/", VideoFileForMetaView.as_view(), name="update_patient_meta"),
-
-
-        
-        
-#----------------------------------START : SENSITIVE META AND RAWPDFOFILE PDF PATIENT DETAILS-------------------------------
-        
-    #The first request (without id) fetches the first available PDF metadata.
-    #The "Next" button (with id) fetches the next available PDF.
-    #If an id is provided, the API returns the actual PDF file instead of JSON.
-    path("api/pdf/sensitivemeta/", PDFFileForMetaView.as_view(), name="pdf_meta"),  
-
-
-
-
-    # This API endpoint allows updating specific patient details (SensitiveMeta)
-    # linked to a PDF record. It enables modifying the patient's first name,
-    # last name, date of birth, and examination date.
-
-    # The frontend should send a JSON request body like this:
-    # {
-    #     "sensitive_meta_id": 2,          # The ID of the SensitiveMeta entry (REQUIRED)
-    #     "patient_first_name": "John",    # New first name (OPTIONAL, if provided, cannot be empty)
-    #     "patient_last_name": "Doe",      # New last name (OPTIONAL, if provided, cannot be empty)
-    #     "patient_dob": "1985-06-15",     # New Date of Birth (OPTIONAL, format YYYY-MM-DD)
-    #     "examination_date": "2024-03-20" # New Examination Date (OPTIONAL, format YYYY-MM-DD)
-    # }
-
-    # - The frontend sends a PATCH request to this endpoint with the updated patient data.
-    # - The backend processes the request and updates only the fields that are provided.
-    # - If validation passes, the corresponding SensitiveMeta entry is updated in the database.
-    # - If errors occur (e.g., invalid ID, empty fields, incorrect date format), 
-    #   the API returns structured error messages.
-
-    path("api/pdf/update_sensitivemeta/", UpdateSensitiveMetaView.as_view(), name="update_pdf_meta"),
-
-
-
-
-
-    #  API Endpoint for Fetching PDF Data (Including Anonymized Text)
-    # - This endpoint is used when the page loads.
-    # - Fetches the first available PDF (if no `last_id` is provided).
-    # - If `last_id` is given, fetches the next available PDF.
-    # - The frontend calls this endpoint on **page load** and when clicking the **next button**.
-    # Example frontend usage:
-    #     const url = lastId ? `http://localhost:8000/api/pdf/anony_text/?last_id=${lastId}` 
-    #                        : "http://localhost:8000/api/pdf/anony_text/";
-    path("api/pdf/anony_text/", RawPdfAnonyTextView.as_view(), name="pdf_anony_text"),  
-
-    #  API Endpoint for Updating the `anonymized_text` Field in `RawPdfFile`
-    # - This endpoint is called when the user edits the anonymized text and clicks **Save**.
-    # - Updates only the `anonymized_text` field for the specified PDF `id`.
-    # - The frontend sends a **PATCH request** to this endpoint with the updated text.
-    # Example frontend usage:
-    #     axios.patch("http://localhost:8000/api/pdf/update_anony_text/", { id: 1, anonymized_text: "Updated text" });
-    path("api/pdf/update_anony_text/", UpdateAnonymizedTextView.as_view(), name="update_pdf_anony_text"),
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-    #this is for, to test the timeline
-    #need to delete this url and also endoreg_db_production/endoreg_db/views/views_for_timeline.py and endoreg_db_production/endoreg_db/templates/timeline.html
-    path('video/<int:video_id>/timeline/', video_timeline_view, name='video_timeline'),
-    ]
-
-    
-#https://biigle.de/manual/tutorials/videos/navigating-timeline#for time line example
-from django.conf import settings
-from django.conf.urls.static import static
-if settings.DEBUG:
-    urlpatterns += static(settings.MEDIA_URL, document_root=settings.MEDIA_ROOT)
+    path("api/", include((api_urlpatterns, "endoreg_db"), namespace="api")),
+]

endoreg_db/utils/__init__.py

@@ -1,79 +1,88 @@
 """Module for utility classes and functions."""
 
+# --- Imports from submodules ---
+
+# dataloader
+from endoreg_db.utils.video.ffmpeg_wrapper import assemble_video_from_frames, get_stream_info, transcode_video, transcode_videofile_if_required
 from .dataloader import load_model_data_from_yaml
-from .parse_and_generate_yaml import collect_center_names
-from .validate_endo_roi import validate_endo_roi
-from .pydantic_models import DbConfig
-from .dates import random_day_by_month_year, random_day_by_year
+
+# dates
+from .dates import ensure_aware_datetime, random_day_by_month_year, random_day_by_year
+
+# env
+from .env import DEBUG, DJANGO_SETTINGS_MODULE, get_env_var
+
+# file_operations
+from .file_operations import copy_with_progress, get_uuid_filename, rename_file_uuid
+
+# hashs
 from .hashs import (
+    DJANGO_NAME_SALT,
+    get_examiner_hash,
     get_hash_string,
+    get_patient_examination_hash,
     get_pdf_hash,
-    get_examiner_hash,
     get_video_hash,
-    get_patient_examination_hash,
-    DJANGO_NAME_SALT
 )
+
+# names
 from .names import (
-    create_mock_patient_name,
     create_mock_examiner_name,
+    create_mock_patient_name,
     guess_name_gender,
 )
 
-from .paths import (
-    STORAGE_DIR,
-    FRAME_DIR,
-    VIDEO_DIR,
-    RAW_VIDEO_DIR,
-    RAW_FRAME_DIR,
-    TEST_RUN,
-    TEST_RUN_FRAME_NUMBER,
-    FRAME_PROCESSING_BATCH_SIZE,
-    WEIGHTS_DIR,
-    WEIGHTS_DIR_NAME,
-    FRAME_DIR_NAME,
-    VIDEO_DIR_NAME,
-    STORAGE_DIR_NAME,
-    RAW_FRAME_DIR_NAME,
-    RAW_VIDEO_DIR_NAME,
-    PDF_DIR_NAME,
-    PDF_DIR,
-    RAW_PDF_DIR,
-    RAW_PDF_DIR_NAME,
+# parse_and_generate_yaml
+from .parse_and_generate_yaml import collect_center_names
+
+# paths
+from .paths import data_paths
+
+# pydantic_models
+from .pydantic_models import DbConfig
+
+# validate_endo_roi
+from .validate_endo_roi import validate_endo_roi
+
+# ffmpeg_wrapper
+from .video.ffmpeg_wrapper import (
+    extract_frames,
+)
+from .video import (
+    split_video
 )
 
+
+# --- Exports ---
+
 __all__ = [
-    "load_model_data_from_yaml",
     "collect_center_names",
-    "validate_endo_roi",
-    "get_examiner_hash",
-    "DbConfig",
-    "random_day_by_month_year",
-    "random_day_by_year",
-    "get_hash_string",
+    "copy_with_progress",
     "create_mock_examiner_name",
     "create_mock_patient_name",
-    "guess_name_gender",
+    "data_paths",
+    "DbConfig",
+    "DEBUG",
+    "DJANGO_NAME_SALT",
+    "DJANGO_SETTINGS_MODULE",
+    "ensure_aware_datetime",
+    "get_env_var",
+    "get_examiner_hash",
+    "get_hash_string",
+    "get_patient_examination_hash",
     "get_pdf_hash",
+    "get_uuid_filename",
     "get_video_hash",
-    "get_patient_examination_hash",
-    "STORAGE_DIR",
-    "FRAME_DIR",
-    "VIDEO_DIR",
-    "RAW_VIDEO_DIR",
-    "RAW_FRAME_DIR",
-    "TEST_RUN",
-    "TEST_RUN_FRAME_NUMBER",
-    "FRAME_PROCESSING_BATCH_SIZE",
-    "DJANGO_NAME_SALT",
-    "WEIGHTS_DIR",
-    "WEIGHTS_DIR_NAME",
-    "FRAME_DIR_NAME",
-    "VIDEO_DIR_NAME",
-    "STORAGE_DIR_NAME",
-    "RAW_FRAME_DIR_NAME",
-    "RAW_VIDEO_DIR_NAME",
-    "RAW_PDF_DIR_NAME",
-    "PDF_DIR_NAME",
-    "PDF_DIR",
-    "RAW_PDF_DIR",
+    "guess_name_gender",
+    "load_model_data_from_yaml",
+    "random_day_by_month_year",
+    "random_day_by_year",
+    "rename_file_uuid",
+    "validate_endo_roi",
+    "assemble_video_from_frames",  # Updated name
+    "get_stream_info",
+    "transcode_video",
+    "transcode_videofile_if_required",  # Added
+    "extract_frames",  # Added
 ]
+

endoreg_db/utils/ai/__init__.py

@@ -0,0 +1,9 @@
+from .inference_dataset import InferenceDataset
+from .multilabel_classification_net import MultiLabelClassificationNet
+from .predict import Classifier
+
+__all__ = [
+    "InferenceDataset",
+    "MultiLabelClassificationNet",
+    "Classifier",
+]

endoreg_db/{models/ai_model/utils.py → utils/ai/get.py}

@@ -1,8 +1,5 @@
-from .model_meta import ModelMeta
+from endoreg_db.models import ModelMeta
 
-# get latest model meta by model name
-
-# TODO MOVE THIS TO A QUERY FILE
 def get_latest_model_meta_by_model_name(model_name):
     model_meta = ModelMeta.objects.filter(name=model_name).order_by('-version').first()
     return model_meta

endoreg_db/{models/ai_model/lightning → utils/ai}/inference_dataset.py

@@ -1,5 +1,4 @@
 from torch.utils.data import Dataset
-import torch
 import numpy as np
 from PIL import Image
 from torchvision import transforms

endoreg_db/{models/ai_model/lightning → utils/ai}/multilabel_classification_net.py

@@ -44,7 +44,7 @@ def calculate_metrics(pred, target, threshold=0.5):
 class MultiLabelClassificationNet(LightningModule):
     def __init__(
         self,
-        labels=["ASD"],
+        labels=None,
         lr=6e-3,
         weight_decay=0.001,
         pos_weight=2,
@@ -76,13 +76,18 @@ class MultiLabelClassificationNet(LightningModule):
             pos_weight=torch.Tensor([self.pos_weight] * len(self.labels))
         )
 
-    def forward(self, x):
-        """Forward pass"""
+    @classmethod
+    def load_from_checkpoint(cls, checkpoint_path, *args, **kwargs):
+        instance = super(MultiLabelClassificationNet, cls).load_from_checkpoint(
+            checkpoint_path, *args, **kwargs
+        )
+        return instance
+
+    def forward(self, x):  # pylint: disable=arguments-differ
         x = self.model(x)
         return x
 
-    def training_step(self, batch, batch_idx):
-        """Training step"""
+    def training_step(self, batch, _batch_idx):  # pylint: disable=arguments-differ
         x, y = batch
         y_pred = self(x)
         loss = self.criterion(y_pred, y)
@@ -94,8 +99,7 @@ class MultiLabelClassificationNet(LightningModule):
 
         return {"loss": loss, "preds": preds, "targets": y}
 
-    def validation_step(self, batch, batch_idx):
-        """Validation step"""
+    def validation_step(self, batch, _batch_idx):  # pylint: disable=arguments-differ
         x, y = batch
         y_pred = self(x)
         loss = self.criterion(y_pred, y)
@@ -107,8 +111,8 @@ class MultiLabelClassificationNet(LightningModule):
 
         return {"loss": loss, "preds": preds, "targets": y}
 
-    def validation_epoch_end(self, outputs):
-        """Validation epoch end"""
+    def validation_epoch_end(self, _outputs):
+        """Called at the end of validation to aggregate outputs"""
         self.val_preds = np.concatenate([_ for _ in self.val_preds])
         self.val_targets = np.concatenate([_ for _ in self.val_targets])
 
@@ -145,7 +149,7 @@ class MultiLabelClassificationNet(LightningModule):
             self.parameters(), self.lr, momentum=0.5, weight_decay=self.weight_decay
         )
         lr_scheduler = torch.optim.lr_scheduler.CosineAnnealingWarmRestarts(
-            optimizer, T_0=20, verbose=True
+            optimizer, T_0=20,
         )
 
         return {

endoreg_db/{models/ai_model/lightning → utils/ai}/postprocess.py

@@ -1,7 +1,8 @@
 import numpy as np
 
+
 def concat_pred_dicts(pred_dicts):
-    '''Shoulkd be a list of dictionaries with the same keys'''
+    """Shoulkd be a list of dictionaries with the same keys"""
     assert len(pred_dicts) > 0
     keys = pred_dicts[0].keys()
 
@@ -15,11 +16,15 @@ def concat_pred_dicts(pred_dicts):
 
     return merged_predictions
 
-def make_smooth_preds(prediction_array, window_size_s=1, fps = 50):
+
+def make_smooth_preds(prediction_array, window_size_s=1, fps=50):
     window_size = int(window_size_s * fps)
-    smooth_prediction_array = np.convolve(prediction_array, np.ones(window_size)/window_size, mode='valid')
+    smooth_prediction_array = np.convolve(
+        prediction_array, np.ones(window_size) / window_size, mode="valid"
+    )
     return smooth_prediction_array
 
+
 def find_true_pred_sequences(predictions):
     """
     Efficiently finds sequences of 'outside' predictions in the binary predictions array using NumPy.
@@ -47,7 +52,12 @@ def find_true_pred_sequences(predictions):
         change_indices = np.append(change_indices, predictions.size)
 
     # Extract the 'outside' sequences by slicing the change_indices array in steps of two
-    outside_sequences = [(change_indices[i], change_indices[i + 1] - 1) for i in range(0, len(change_indices), 2)]
+    outside_sequences = [
+        (change_indices[i], change_indices[i + 1] - 1)
+        for i in range(0, len(change_indices), 2)
+    ]
 
-    return outside_sequences
+    # make sure the result is serializable
+    outside_sequences = [(int(start), int(stop)) for start, stop in outside_sequences]
 
+    return outside_sequences