From 55a4720a9e72fb37def04357694f79dfd24dab6c Mon Sep 17 00:00:00 2001
From: gh-underfoot581 <313842205+gh-underfoot581@users.noreply.github.com>
Date: Mon, 31 Aug 2026 21:41:28 +1000
Subject: [PATCH] fix: find the exported engine next to the source model

Ultralytics writes the .engine beside the .pt it exported, which is not
always the output directory the caller asked for. Only that directory
was checked, so a perfectly good export was reported as "No engine file
was created".

Check beside the model too, and move the file into the output directory
when it turns up there. The failure message now names both paths it
looked in.
---
 .../utils/tensorrt_export_engine_model.py        | 16 ++++++++++++++--
 1 file changed, 14 insertions(+), 2 deletions(-)

diff --git a/application/utils/tensorrt_export_engine_model.py b/application/utils/tensorrt_export_engine_model.py
index dd2b48dd..ddc6eee2 100644
--- a/application/utils/tensorrt_export_engine_model.py
+++ b/application/utils/tensorrt_export_engine_model.py
@@ -44,7 +44,10 @@ def export_model(model_path, output_dir):
         logger.info("Searching for generated engine files...")
         # Find the specific engine file that should have been created
         model_basename = os.path.splitext(os.path.basename(model_path))[0]
-        expected_engine_path = os.path.join(output_dir, model_basename + ".engine")
+        engine_filename = model_basename + ".engine"
+        expected_engine_path = os.path.join(output_dir, engine_filename)
+        # Ultralytics writes the .engine next to the source .pt, not always output_dir
+        pt_engine_path = os.path.join(os.path.dirname(model_path), engine_filename)
         
         if os.path.exists(expected_engine_path):
             logger.info(f"Found engine file: {expected_engine_path}")
@@ -52,11 +55,20 @@ def export_model(model_path, output_dir):
                 'success': True,
                 'engine_file': expected_engine_path
             }
+        elif os.path.exists(pt_engine_path):
+            logger.info(f"Found engine file next to the model: {pt_engine_path}")
+            if os.path.abspath(pt_engine_path) != os.path.abspath(expected_engine_path):
+                import shutil
+                shutil.move(pt_engine_path, expected_engine_path)
+            result = {
+                'success': True,
+                'engine_file': expected_engine_path
+            }
         else:
             logger.error("No engine file was created!")
             result = {
                 'success': False,
-                'error': 'No engine file was created'
+                'error': f'No engine file was created. Checked: {expected_engine_path}, {pt_engine_path}'
             }
     except Exception as e:
         result = {