新增：在构建过程中确保data目录存在，并在打包环境中正确设置数据目录路径；更新release工作流以创建和移动data目录。

Author: c1921

.github/workflows/release.yml

@@ -66,6 +66,10 @@ jobs:
         New-Item -ItemType Directory -Path "$VERSION"
         Move-Item "SD_Models_Manager_v*.exe" "$VERSION"
         
+        # 创建data目录
+        New-Item -ItemType Directory -Path "$VERSION\data"
+        Write-Host "Created data directory for storing prompt library"
+        
         # 只在 frontend 目录存在时才移动它
         if (Test-Path -Path "frontend") {
           Move-Item "frontend" "$VERSION"
@@ -78,6 +82,12 @@ jobs:
           Write-Host "Moved static directory to package"
         }
         
+        # 只在 data 目录存在时才移动它
+        if (Test-Path -Path "data") {
+          Move-Item "data" "$VERSION"
+          Write-Host "Moved data directory to package"
+        }
+        
         # 确保目录结构完整
         if (-not (Test-Path -Path "$VERSION\static")) {
           New-Item -ItemType Directory -Path "$VERSION\static"

build.py

@@ -158,6 +158,9 @@ def build_executable():
         # 设置环境变量以确保正确的编码
         os.environ['PYTHONIOENCODING'] = 'utf-8'
 
+        # 确保data目录存在
+        os.makedirs('data', exist_ok=True)
+        
         # 构建前端
         if not build_frontend():
             print("警告: 前端构建失败，将只包含API部分")
@@ -180,6 +183,10 @@ def build_executable():
         if os.path.exists('static'):
             pyinstaller_args.append('--add-data=static;static')
 
+        # 添加data目录
+        if os.path.exists('data'):
+            pyinstaller_args.append('--add-data=data;data')
+        
         # 添加前端构建
         if os.path.exists('dist/frontend'):
             pyinstaller_args.append('--add-data=dist/frontend;frontend')

src/api/prompt_library_api.py

@@ -6,6 +6,7 @@
 import os
 import json
 from datetime import datetime
+import sys
 
 # 提示词库项目模型
 class PromptLibraryItem(BaseModel):
@@ -36,7 +37,15 @@ class UpdatePromptLibraryItemRequest(BaseModel):
 class PromptLibraryManager:
     def __init__(self, data_dir):
         self.data_dir = data_dir
-        self.library_file = os.path.join(data_dir, "prompt_library.json")
+        
+        # 检查是否是PyInstaller打包环境
+        if getattr(sys, 'frozen', False) and hasattr(sys, '_MEIPASS'):
+            # 在打包环境中，使用应用程序所在目录下的data目录
+            app_dir = os.path.dirname(sys.executable)
+            self.data_dir = os.path.join(app_dir, "data")
+            print(f"检测到打包环境，设置数据目录为: {self.data_dir}")
+        
+        self.library_file = os.path.join(self.data_dir, "prompt_library.json")
         self.items = []
         self.load_library()
 
@@ -49,7 +58,7 @@ def load_library(self):
                 print(f"已加载提示词库，共{len(self.items)}个项目")
             else:
                 self.items = []
-                print("提示词库文件不存在，已初始化空库")
+                print(f"提示词库文件不存在，已初始化空库 (路径: {self.library_file})")
         except Exception as e:
             print(f"加载提示词库失败: {str(e)}")
             self.items = []