Merge pull request #39 from BrainLesion/31-feature-request-citation-r…

…eminder

31 feature request citation reminder

brainles_aurora/inferer/inferer.py

@@ -17,6 +17,7 @@
 from brainles_aurora.inferer.data import DataHandler
 from brainles_aurora.inferer.model import ModelHandler
 from brainles_aurora.utils import remove_path_suffixes
+from brainles_aurora.utils.console_decorators import citation_reminder
 
 logger = logging.getLogger(__name__)
 
@@ -148,6 +149,7 @@ def _configure_device(self) -> torch.device:
 
         return device
 
+    @citation_reminder
     def infer(
         self,
         t1: str | Path | np.ndarray | None = None,

brainles_aurora/utils/console_decorators.py

@@ -0,0 +1,22 @@
+from rich.console import Console
+
+CITATION_LINK = "https://github.com/BrainLesion/AURORA#citation"
+
+
+def citation_reminder(func):
+    def wrapper(*args, **kwargs):
+        console = Console()
+        console.rule("Thank you for using [bold]AURORA[/bold]")
+        console.print(
+            f"Please support our development by citing the papers listed here:",
+            justify="center",
+        )
+        console.print(
+            f"{CITATION_LINK} -- Thank you!",
+            justify="center",
+        )
+        console.rule()
+        console.line()
+        func(*args, **kwargs)
+
+    return wrapper

pyproject.toml

@@ -48,6 +48,7 @@ numpy = ">=1.23.0"
 tqdm = "^4.64.1"
 path = "^16.2.0"
 requests =">=2.0.0"
+rich = ">=13.0.0"
 
 [tool.poetry.dev-dependencies]
 pytest = "^6.2"