diff --git a/notebooks/AGIPD/AGIPD_Correct_and_Verify.ipynb b/notebooks/AGIPD/AGIPD_Correct_and_Verify.ipynb
index 166257f9e088a5ee95c16d1b341ac4d6acf3c11f..fdd307e5e444070c59edb11b683841d927f0a1e6 100644
--- a/notebooks/AGIPD/AGIPD_Correct_and_Verify.ipynb
+++ b/notebooks/AGIPD/AGIPD_Correct_and_Verify.ipynb
@@ -183,6 +183,7 @@
     "from cal_tools.step_timing import StepTimer\n",
     "from cal_tools.tools import (\n",
     "    calcat_creation_time,\n",
+    "    latex_warning,\n",
     "    map_modules_from_folder,\n",
     "    module_index_to_qm,\n",
     "    write_constants_fragment,\n",
@@ -327,6 +328,21 @@
     "print(f\"Instrument {instrument}\")"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# Before going further check if there are any trains\n",
+    "# to correction in AGIPD detector modules INSTRUMENT sources.\n",
+    "channels_to_correct = [instrument_src.format(m) for m in modules]\n",
+    "total_trains = len(dc.select(channels_to_correct, require_all=True).train_ids)\n",
+    "if total_trains == 0:\n",
+    "    latex_warning(f\"No trains available to correct in: {channels_to_correct}\")\n",
+    "    sys.exit(0)"
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,
@@ -523,11 +539,12 @@
     "        print(f\"{cell_sel.msg()}\\n\")\n",
     "        cell_sel.print_report()\n",
     "        if np.count_nonzero(r.nLitFrame.value) == 0:  # No lit frames.\n",
-    "            warning(\n",
-    "                \"Warning: No lit frames identified using AGIPD LitFrameFinder.\"\n",
+    "            latex_warning(\n",
+    "                \"No lit frames identified using AGIPD LitFrameFinder.\"\n",
     "                \" Offline correction will not be performed.\")\n",
     "            sys.exit(0)\n",
     "    except LitFrameFinderError as err:\n",
+    "        print(cell_sel.msg())\n",
     "        warning(f\"Cannot use AgipdLitFrameFinder due to:\\n{err}\")\n",
     "        cell_sel = CellRange(max_pulses, max_cells=mem_cells)\n",
     "else:\n",
diff --git a/src/cal_tools/tools.py b/src/cal_tools/tools.py
index cc81c81a4e0156eb6c60bdfe00b2b0671c566efb..0161fb8922db89a638ac1c8e70ca27e88da2882d 100644
--- a/src/cal_tools/tools.py
+++ b/src/cal_tools/tools.py
@@ -3,6 +3,7 @@ import json
 import os
 import re
 import zlib
+from IPython.display import Latex, display
 from collections import OrderedDict
 from glob import glob
 from multiprocessing.pool import ThreadPool
@@ -1151,3 +1152,9 @@ def exit_notebook(message):
     import sys
     print(message)
     sys.exit(0)
+
+
+def latex_warning(message):
+    """Show latex warning custom command parsed correctly
+    when reports are generated."""
+    display(Latex("\warningbox{" + message + "}"))
diff --git a/src/xfel_calibrate/finalize.py b/src/xfel_calibrate/finalize.py
index 82f6235c8afa9c03ed165f776092dbe83f8258cd..78efb19b04947a42388f39aed4b94586b62851a9 100644
--- a/src/xfel_calibrate/finalize.py
+++ b/src/xfel_calibrate/finalize.py
@@ -245,6 +245,20 @@ def make_report(run_path: Path, cal_work_dir: Path, project: str,
         latex_elements = {'extraclassoptions': ',openany, oneside',
                           'preamble': r'\usepackage{longtable}',
                           'maketitle': r'\input{titlepage.tex.txt}'}
+
+        # Define a new LaTeX command for warning boxes within the preamble
+        custom_latex_preamble = r'''
+        \newcommand{\warningbox}[1]{%
+            \noindent\fcolorbox{orange}{yellow}{%
+                \parbox{\textwidth}{%
+                    \textbf{Warning:} #1%
+                }%
+            }
+        }
+        '''
+
+        # Append the custom preamble
+        latex_elements['preamble'] += custom_latex_preamble
         mf.write("latex_elements = {}\n".format(latex_elements))
         mf.write("latex_logo = '{}/{}'\n".format(module_path,
                                                  logo_path))