feat(logging): store application logs (#854)

* feat: Wrap all processes inside a try catch block

* chore: Modify files to use the custom logger

* lint: flake8 lint

---------

Co-authored-by: Richard Abrich <[email protected]>

Author: KIRA009

CONTRIBUTING.md

@@ -18,7 +18,7 @@ from pprint import pformat
 import json
 
 # external
-from loguru import logger
+from openadapt.custom_logger import logger
 import numpy as np
 
 # local

experiments/fastsamsom.py

@@ -2,12 +2,11 @@
 
 from pprint import pformat
 
-from loguru import logger
 from PIL import Image
 import numpy as np
 
 from openadapt import adapters, config, contrib, utils, vision
-
+from openadapt.custom_logger import logger
 
 CONTRAST_FACTOR = 10000
 DEBUG = False

experiments/gpt4o_seg.py

@@ -5,11 +5,11 @@
 import sys
 import time
 
-from loguru import logger
 from PIL import Image
 
 from openadapt import cache, config, models, plotting, utils
 from openadapt.adapters import openai
+from openadapt.custom_logger import logger
 
 
 @cache.cache(force_refresh=False)

experiments/handle_similar_segments.py

@@ -3,10 +3,9 @@
 import os
 
 from PIL import Image
-from loguru import logger
 
 from openadapt import adapters, cache, config, plotting, utils, vision
-
+from openadapt.custom_logger import logger
 
 DEBUG = True
 MIN_SEGMENT_SSIM = 0.95  # threshold for considering segments structurally similar

openadapt/adapters/prompt.py

@@ -1,13 +1,12 @@
 """Adapter for prompting foundation models."""
 
-from loguru import logger
 from typing import Type
-from PIL import Image
 
+from PIL import Image
 
+from openadapt.custom_logger import logger
 from openadapt.drivers import anthropic, google, openai
 
-
 # Define a list of drivers in the order they should be tried
 DRIVER_ORDER: list[Type] = [openai, google, anthropic]
 

openadapt/adapters/replicate.py

@@ -3,14 +3,14 @@
 import io
 import os
 
-from loguru import logger
 from PIL import Image
 import fire
 import replicate
 import requests
 
 from openadapt import cache, utils
 from openadapt.config import config
+from openadapt.custom_logger import logger
 
 
 @cache.cache()

openadapt/adapters/som.py

@@ -6,12 +6,12 @@
 import os
 import tempfile
 
-from loguru import logger
 from PIL import Image
 import fire
 import gradio_client
 
 from openadapt.config import config
+from openadapt.custom_logger import logger
 
 
 def save_image_to_temp_file(img: Image.Image) -> str:

openadapt/adapters/ultralytics.py

@@ -11,16 +11,16 @@
 import os
 import time
 
-from loguru import logger
 from PIL import Image, ImageColor
-import numpy as np
 
+# importing is required for use() to work
+from PySide6.QtCore import Qt  # noqa
 
 # use() required when invoked from tray
 import matplotlib
+import numpy as np
 
-# importing is required for use() to work
-from PySide6.QtCore import Qt  # noqa
+from openadapt.custom_logger import logger
 
 matplotlib.use("Qt5Agg")
 
@@ -34,7 +34,6 @@
 
 from openadapt import cache
 
-
 FASTSAM_MODEL_NAMES = (
     "FastSAM-x.pt",
     "FastSAM-s.pt",

openadapt/app/cards.py

@@ -12,6 +12,7 @@
 from openadapt.app.objects.local_file_picker import LocalFilePicker
 from openadapt.app.util import get_scrub, set_dark, set_scrub, sync_switch
 from openadapt.record import record
+from openadapt.utils import WrapStdout
 
 
 class RecordProc:
@@ -56,7 +57,7 @@ def is_running(self) -> bool:
     def start(self, func: callable, args: tuple, kwargs: dict) -> None:
         """Start the recording process."""
         self.record_proc = multiprocessing.Process(
-            target=func,
+            target=WrapStdout(func),
             args=args,
             kwargs=kwargs,
         )

openadapt/app/dashboard/api/action_events.py

@@ -1,8 +1,8 @@
 """API endpoints for recordings."""
 
 from fastapi import APIRouter
-from loguru import logger
 
+from openadapt.custom_logger import logger
 from openadapt.db import crud