From 3de31cd06aeab33b7a0f507b3064486c8b4a7bce Mon Sep 17 00:00:00 2001
From: Will Miao <13051207myq@gmail.com>
Date: Fri, 25 Apr 2025 09:41:23 +0800
Subject: [PATCH] feat: Add functionality to move civitai.info file during
 model relocation

---
 py/services/model_scanner.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/py/services/model_scanner.py b/py/services/model_scanner.py
index 93f622b9..b3db2d33 100644
--- a/py/services/model_scanner.py
+++ b/py/services/model_scanner.py
@@ -736,6 +736,12 @@ class ModelScanner:
                 shutil.move(source_metadata, target_metadata)
                 metadata = await self._update_metadata_paths(target_metadata, target_file)
             
+            # Move civitai.info file if exists
+            source_civitai = os.path.join(source_dir, f"{base_name}.civitai.info")
+            if os.path.exists(source_civitai):
+                target_civitai = os.path.join(target_path, f"{base_name}.civitai.info")
+                shutil.move(source_civitai, target_civitai)
+            
             for ext in PREVIEW_EXTENSIONS:
                 source_preview = os.path.join(source_dir, f"{base_name}{ext}")
                 if os.path.exists(source_preview):