feat(ai-studio): add main workspace tab

app/gui/__init__.py

@@ -24,12 +24,16 @@ if QT_IMPORT_ERROR is None:
         ApplyWorker,
         CollectWorker,
         GenerateWorker,
+        ImageStudioDownloadOriginalWorker,
+        ImageStudioGenerateJobsWorker,
+        ImageStudioPullImagesWorker,
         WriteBackWorker,
     )
     from .tabs.accounts import AccountDialog, AccountsTab
     from .tabs.apply import ApplyTab
     from .tabs.collect import CollectTab
     from .tabs.generate import GenerateTab
+    from .tabs.image_studio import ImageStudioPreviewDialog, ImageStudioTab
     from .tabs.settings import SettingsTab
     from .main_window import MainWindow
 else:

app/gui/main_window.py

@@ -7,6 +7,7 @@ from .tabs.accounts import AccountsTab
 from .tabs.apply import ApplyTab
 from .tabs.collect import CollectTab
 from .tabs.generate import GenerateTab
+from .tabs.image_studio import ImageStudioTab
 from .tabs.settings import SettingsTab
 from .widgets import *
 
@@ -72,7 +73,7 @@ def _fit_and_center_window(
 
 
 class MainWindow(QMainWindow):
-    """Main application window with the fixed five-tab workflow."""
+    """Main application window with the fixed workflow tabs."""
 
     def __init__(
         self,
@@ -158,11 +159,19 @@ class MainWindow(QMainWindow):
                 config=self.config,
                 status_callback=self.show_status,
             )
-        return SettingsTab(
+        if title == "⑤ 设置":
+            return SettingsTab(
+                config=self.config,
+                config_path=self.config_path,
+                ai_models_path=self.ai_models_path,
+                status_callback=self.show_status,
+            )
+        return ImageStudioTab(
+            db_path=self.db_path,
             config=self.config,
             config_path=self.config_path,
-            ai_models_path=self.ai_models_path,
             status_callback=self.show_status,
+            prompts_dir=appconfig.image_studio_prompts_dir(self.config),
         )
 
     def refresh_task_tabs(self):

app/gui/tabs/image_studio.py

@@ -0,0 +1,984 @@
+"""Tab 6: AI image studio UI."""
+
+from __future__ import annotations
+
+import os
+
+from ... import accounts, appconfig, db, image_studio, prompts
+from .. import file_manager
+from ..widgets import *
+from ..workers import (
+    ImageStudioDownloadOriginalWorker as _RealImageStudioDownloadOriginalWorker,
+)
+from ..workers import ImageStudioGenerateJobsWorker as _RealImageStudioGenerateJobsWorker
+from ..workers import ImageStudioPullImagesWorker as _RealImageStudioPullImagesWorker
+
+
+def ImageStudioPullImagesWorker(*args, **kwargs):
+    return _call_package_attr(
+        "ImageStudioPullImagesWorker",
+        _RealImageStudioPullImagesWorker,
+        *args,
+        **kwargs,
+    )
+
+
+def ImageStudioDownloadOriginalWorker(*args, **kwargs):
+    return _call_package_attr(
+        "ImageStudioDownloadOriginalWorker",
+        _RealImageStudioDownloadOriginalWorker,
+        *args,
+        **kwargs,
+    )
+
+
+def ImageStudioGenerateJobsWorker(*args, **kwargs):
+    return _call_package_attr(
+        "ImageStudioGenerateJobsWorker",
+        _RealImageStudioGenerateJobsWorker,
+        *args,
+        **kwargs,
+    )
+
+
+class ImageStudioPreviewDialog(QDialog):
+    """Simple large image preview used by original and pool tables."""
+
+    def __init__(self, asset, parent=None):
+        super().__init__(parent)
+        self.asset = asset
+        self.setWindowTitle(self._title_for_asset(asset))
+        layout = QVBoxLayout(self)
+        scroll = QScrollArea()
+        scroll.setWidgetResizable(False)
+        image_label = QLabel()
+        image_label.setAlignment(Qt.AlignCenter)
+        path = str(getattr(asset, "local_path", "") or "")
+        image = QImage(path) if path and os.path.isfile(path) else QImage()
+        if image.isNull():
+            image_label.setText("图片尚未下载或读取失败")
+            image_label.setMinimumSize(420, 260)
+        else:
+            image_label.setPixmap(QPixmap.fromImage(image))
+            image_label.resize(image.size())
+            self.setWindowTitle(
+                f"{self._title_for_asset(asset)} · {image.width()}x{image.height()}"
+            )
+        scroll.setWidget(image_label)
+        layout.addWidget(scroll, 1)
+        buttons = QHBoxLayout()
+        buttons.addStretch(1)
+        close_button = QPushButton("关闭")
+        close_button.clicked.connect(self.reject)
+        buttons.addWidget(close_button)
+        layout.addLayout(buttons)
+        self.resize(720, 520)
+
+    def _title_for_asset(self, asset):
+        badge = _asset_badge(getattr(asset, "kind", ""))
+        asset_id = getattr(asset, "id", "")
+        return f"AI工场图片预览：{badge} #{asset_id}"
+
+
+class ImageStudioTab(QWidget):
+    """Sixth tab: project-based AI image studio."""
+
+    PROJECT_COLUMNS = ["项目", "账号", "商品ID", "更新时间"]
+    ORIGINAL_COLUMNS = ["序号", "状态", "远程地址"]
+    POOL_COLUMNS = ["类型", "比例", "状态", "来源", "本地文件"]
+    JOB_STATUS_LABELS = {
+        "pending": "排队中",
+        "submitted": "已提交",
+        "running": "生成中",
+        "succeeded": "成功",
+        "failed": "失败",
+        "expired": "已过期",
+        "cancelled": "已停止",
+    }
+
+    def __init__(
+        self,
+        parent=None,
+        db_path=None,
+        config=None,
+        config_path=None,
+        status_callback=None,
+        prompts_dir=None,
+    ):
+        super().__init__(parent)
+        self.setObjectName("imageStudioTab")
+        self.config = appconfig.load_config(config_path or appconfig.CONFIG_PATH) if config is None else config
+        self.db_path = db_path or appconfig.db_path(self.config)
+        self.config_path = config_path or self.config.get("config_path") or appconfig.CONFIG_PATH
+        self.prompts_dir = prompts_dir or appconfig.image_studio_prompts_dir(self.config)
+        self.cmhub_config_path = self.config.get("cmhub_config_path") or appconfig.cmhub_config_file_path(self.config)
+        self.status_callback = status_callback
+        self.projects = []
+        self.accounts = []
+        self.current_project = None
+        self.assets = []
+        self.jobs = []
+        self.selections = []
+        self.selected_source_asset_id = None
+        self._running_worker = None
+        self._running_thread = None
+        self._download_open_after = {}
+
+        self._build_ui()
+        self._connect_signals()
+        self.refresh_accounts()
+        self.refresh_templates()
+        self.refresh_projects()
+
+    def _build_ui(self):
+        root = QVBoxLayout(self)
+        root.setContentsMargins(10, 8, 10, 8)
+        root.setSpacing(8)
+
+        splitter = QSplitter(Qt.Horizontal)
+        splitter.setObjectName("imageStudioMainSplitter")
+        splitter.addWidget(self._build_project_panel())
+        splitter.addWidget(self._build_pool_panel())
+        splitter.addWidget(self._build_generation_panel())
+        splitter.setStretchFactor(0, 1)
+        splitter.setStretchFactor(1, 3)
+        splitter.setStretchFactor(2, 2)
+        root.addWidget(splitter, 1)
+        root.addWidget(self._build_final_panel(), 0)
+
+    def _build_project_panel(self):
+        panel = QWidget()
+        panel.setObjectName("imageStudioProjectPanel")
+        layout = QVBoxLayout(panel)
+        layout.setContentsMargins(0, 0, 0, 0)
+        layout.setSpacing(8)
+
+        project_form = QFormLayout()
+        project_form.setLabelAlignment(Qt.AlignRight)
+        self.account_combo = QComboBox()
+        self.account_combo.setObjectName("imageStudioAccountCombo")
+        self.item_id_edit = QLineEdit()
+        self.item_id_edit.setObjectName("imageStudioItemIdEdit")
+        self.item_id_edit.setPlaceholderText("商品ID")
+        project_form.addRow("账号", self.account_combo)
+        project_form.addRow("商品ID", self.item_id_edit)
+        layout.addLayout(project_form)
+
+        toolbar = QHBoxLayout()
+        self.open_project_button = QPushButton("打开项目")
+        self.open_project_button.setObjectName("imageStudioOpenProjectButton")
+        self.pull_images_button = QPushButton("拉取主图")
+        self.pull_images_button.setObjectName("imageStudioPullImagesButton")
+        self.open_folder_button = QPushButton("打开项目文件夹")
+        self.open_folder_button.setObjectName("imageStudioOpenFolderButton")
+        toolbar.addWidget(self.open_project_button)
+        toolbar.addWidget(self.pull_images_button)
+        toolbar.addWidget(self.open_folder_button)
+        layout.addLayout(toolbar)
+
+        self.project_table = QTableWidget(0, len(self.PROJECT_COLUMNS))
+        self.project_table.setObjectName("imageStudioProjectList")
+        self.project_table.setHorizontalHeaderLabels(self.PROJECT_COLUMNS)
+        self.project_table.setSelectionBehavior(QAbstractItemView.SelectRows)
+        self.project_table.setSelectionMode(QAbstractItemView.SingleSelection)
+        self.project_table.setEditTriggers(QAbstractItemView.NoEditTriggers)
+        self.project_table.horizontalHeader().setStretchLastSection(True)
+        self.project_table.verticalHeader().setVisible(False)
+        layout.addWidget(self.project_table, 1)
+        return panel
+
+    def _build_pool_panel(self):
+        panel = QWidget()
+        panel.setObjectName("imageStudioPoolPanel")
+        layout = QVBoxLayout(panel)
+        layout.setContentsMargins(0, 0, 0, 0)
+        layout.setSpacing(8)
+
+        original_header = QHBoxLayout()
+        original_header.addWidget(QLabel("蝦皮原主图"))
+        original_header.addStretch(1)
+        self.original_hint_label = QLabel("单击下载并加入照片池，双击查看大图")
+        self.original_hint_label.setObjectName("imageStudioOriginalHintLabel")
+        original_header.addWidget(self.original_hint_label)
+        layout.addLayout(original_header)
+
+        self.original_table = QTableWidget(0, len(self.ORIGINAL_COLUMNS))
+        self.original_table.setObjectName("imageStudioOriginalTable")
+        self.original_table.setHorizontalHeaderLabels(self.ORIGINAL_COLUMNS)
+        self.original_table.setSelectionBehavior(QAbstractItemView.SelectRows)
+        self.original_table.setSelectionMode(QAbstractItemView.SingleSelection)
+        self.original_table.setEditTriggers(QAbstractItemView.NoEditTriggers)
+        self.original_table.horizontalHeader().setStretchLastSection(True)
+        self.original_table.verticalHeader().setVisible(False)
+        layout.addWidget(self.original_table, 1)
+
+        pool_header = QHBoxLayout()
+        pool_header.addWidget(QLabel("照片池"))
+        pool_header.addStretch(1)
+        self.source_label = QLabel("源图：未选择")
+        self.source_label.setObjectName("imageStudioSourceLabel")
+        pool_header.addWidget(self.source_label)
+        layout.addLayout(pool_header)
+
+        self.pool_table = QTableWidget(0, len(self.POOL_COLUMNS))
+        self.pool_table.setObjectName("imageStudioPoolTable")
+        self.pool_table.setHorizontalHeaderLabels(self.POOL_COLUMNS)
+        self.pool_table.setSelectionBehavior(QAbstractItemView.SelectRows)
+        self.pool_table.setSelectionMode(QAbstractItemView.SingleSelection)
+        self.pool_table.setEditTriggers(QAbstractItemView.NoEditTriggers)
+        self.pool_table.setContextMenuPolicy(Qt.CustomContextMenu)
+        self.pool_table.horizontalHeader().setStretchLastSection(True)
+        self.pool_table.verticalHeader().setVisible(False)
+        layout.addWidget(self.pool_table, 2)
+        return panel
+
+    def _build_generation_panel(self):
+        panel = QWidget()
+        panel.setObjectName("imageStudioGenerationPanel")
+        layout = QVBoxLayout(panel)
+        layout.setContentsMargins(0, 0, 0, 0)
+        layout.setSpacing(8)
+
+        template_layout = QGridLayout()
+        self.template_combo = QComboBox()
+        self.template_combo.setObjectName("imageStudioTemplateCombo")
+        self.template_new_button = QPushButton("新建")
+        self.template_new_button.setObjectName("imageStudioTemplateNewButton")
+        self.template_rename_button = QPushButton("重命名")
+        self.template_rename_button.setObjectName("imageStudioTemplateRenameButton")
+        self.template_save_button = QPushButton("保存")
+        self.template_save_button.setObjectName("imageStudioTemplateSaveButton")
+        self.template_delete_button = QPushButton("删除")
+        self.template_delete_button.setObjectName("imageStudioTemplateDeleteButton")
+        template_layout.addWidget(self.template_combo, 0, 0, 1, 2)
+        template_layout.addWidget(self.template_new_button, 0, 2)
+        template_layout.addWidget(self.template_rename_button, 1, 0)
+        template_layout.addWidget(self.template_save_button, 1, 1)
+        template_layout.addWidget(self.template_delete_button, 1, 2)
+        layout.addLayout(template_layout)
+
+        self.prompt_edit = QPlainTextEdit()
+        self.prompt_edit.setObjectName("imageStudioPromptEdit")
+        self.prompt_edit.setPlaceholderText("输入完整图片生成提示词")
+        self.prompt_edit.setMinimumHeight(160)
+        layout.addWidget(self.prompt_edit, 2)
+
+        form = QFormLayout()
+        self.job_type_combo = QComboBox()
+        self.job_type_combo.setObjectName("imageStudioJobTypeCombo")
+        self.job_type_combo.addItem("主图", "main")
+        self.job_type_combo.addItem("详情图", "detail")
+        self.count_spin = QSpinBox()
+        self.count_spin.setObjectName("imageStudioCountSpin")
+        self.count_spin.setRange(1, 12)
+        self.count_spin.setValue(4)
+        self.aspect_combo = QComboBox()
+        self.aspect_combo.setObjectName("imageStudioAspectCombo")
+        for value in ("1:1", "3:4", "4:3", "9:16", "16:9"):
+            self.aspect_combo.addItem(value, value)
+        form.addRow("类型", self.job_type_combo)
+        form.addRow("数量", self.count_spin)
+        form.addRow("比例", self.aspect_combo)
+        layout.addLayout(form)
+
+        self.billing_label = QLabel("cmhub 托管模型：扣点以返回结果为准")
+        self.billing_label.setObjectName("imageStudioBillingLabel")
+        self.billing_label.setWordWrap(True)
+        layout.addWidget(self.billing_label)
+
+        action_layout = QHBoxLayout()
+        self.start_button = QPushButton("开始生成")
+        self.start_button.setObjectName("imageStudioStartButton")
+        self.stop_button = QPushButton("停止")
+        self.stop_button.setObjectName("imageStudioStopButton")
+        self.stop_button.setEnabled(False)
+        action_layout.addWidget(self.start_button)
+        action_layout.addWidget(self.stop_button)
+        layout.addLayout(action_layout)
+
+        self.progress_bar = QProgressBar()
+        self.progress_bar.setObjectName("imageStudioProgressBar")
+        self.progress_bar.setRange(0, 1)
+        self.progress_bar.setValue(0)
+        layout.addWidget(self.progress_bar)
+
+        self.log_view = QPlainTextEdit()
+        self.log_view.setObjectName("imageStudioLogView")
+        self.log_view.setReadOnly(True)
+        self.log_view.setPlaceholderText("运行日志会在开始后显示")
+        layout.addWidget(self.log_view, 1)
+        return panel
+
+    def _build_final_panel(self):
+        panel = QWidget()
+        panel.setObjectName("imageStudioFinalPanel")
+        layout = QHBoxLayout(panel)
+        layout.setContentsMargins(0, 0, 0, 0)
+        self.main_selection_label = QLabel("主图终选 0/9（拖放排序将在 T-592 接入）")
+        self.main_selection_label.setObjectName("imageStudioMainSelectionLabel")
+        self.detail_selection_label = QLabel("详情图终选 0/12（拖放排序将在 T-592 接入）")
+        self.detail_selection_label.setObjectName("imageStudioDetailSelectionLabel")
+        layout.addWidget(self.main_selection_label)
+        layout.addWidget(self.detail_selection_label)
+        layout.addStretch(1)
+        return panel
+
+    def _connect_signals(self):
+        self.open_project_button.clicked.connect(self.open_project)
+        self.pull_images_button.clicked.connect(self.pull_main_images)
+        self.open_folder_button.clicked.connect(self.open_project_folder)
+        self.project_table.itemSelectionChanged.connect(self._on_project_selection_changed)
+        self.original_table.cellClicked.connect(self._on_original_clicked)
+        self.original_table.cellDoubleClicked.connect(self._on_original_double_clicked)
+        self.pool_table.cellClicked.connect(self._on_pool_clicked)
+        self.pool_table.cellDoubleClicked.connect(self._on_pool_double_clicked)
+        self.pool_table.customContextMenuRequested.connect(self._show_pool_context_menu)
+        self.template_combo.currentIndexChanged.connect(self.load_selected_template)
+        self.template_new_button.clicked.connect(self.create_template)
+        self.template_rename_button.clicked.connect(self.rename_template)
+        self.template_save_button.clicked.connect(self.save_template)
+        self.template_delete_button.clicked.connect(self.delete_template)
+        self.prompt_edit.textChanged.connect(self._save_project_prompt)
+        self.start_button.clicked.connect(self.start_generation)
+        self.stop_button.clicked.connect(self.stop_generation)
+
+    def refresh_accounts(self):
+        self.account_combo.clear()
+        try:
+            self.accounts = accounts.list_accounts(path=self.db_path, config=self.config)
+        except Exception as exc:
+            self.accounts = []
+            self._status(f"账号读取失败：{exc}", "danger")
+        for account in self.accounts:
+            self.account_combo.addItem(
+                f"{account.account_name}（{account.alias}）",
+                account.alias,
+            )
+        if not self.accounts:
+            self.account_combo.addItem("暂无账号，请先到④账号管理添加", "")
+
+    def refresh_projects(self):
+        try:
+            db.init_db(self.db_path)
+            self.projects = image_studio.list_projects(path=self.db_path)
+        except Exception as exc:
+            self.projects = []
+            self._status(f"AI工场项目读取失败：{exc}", "danger")
+        self._fill_project_table()
+        if self.current_project is None and self.projects:
+            self._select_project(self.projects[0].id)
+        elif self.current_project is not None:
+            self._select_project(self.current_project.id, quiet=True)
+
+    def _fill_project_table(self):
+        self.project_table.setRowCount(len(self.projects))
+        for row, project in enumerate(self.projects):
+            values = [
+                f"{project.account_alias} / {project.item_id}",
+                project.account_name or project.account_alias,
+                project.item_id,
+                project.updated_at,
+            ]
+            for column, value in enumerate(values):
+                item = QTableWidgetItem(str(value or ""))
+                item.setData(Qt.UserRole, int(project.id))
+                self.project_table.setItem(row, column, item)
+        self.project_table.resizeColumnsToContents()
+
+    def open_project(self, checked=False):
+        alias = str(self.account_combo.currentData() or "").strip()
+        item_id = self.item_id_edit.text().strip()
+        if not alias:
+            self._message("账号未选择", "请先在④账号管理添加并选择账号。")
+            return
+        if not item_id:
+            self._message("商品ID不能为空", "请输入要打开的蝦皮商品ID。")
+            return
+        account = accounts.get_account(alias, path=self.db_path, config=self.config)
+        try:
+            project = image_studio.create_or_get_project(
+                account,
+                item_id=item_id,
+                path=self.db_path,
+            )
+        except Exception as exc:
+            self._message("打开项目失败", str(exc))
+            self._status(f"打开AI工场项目失败：{exc}", "danger")
+            return
+        self.current_project = project
+        self.item_id_edit.setText(project.item_id)
+        self._set_account_combo(project.account_alias)
+        self.refresh_projects()
+        self._select_project(project.id)
+        self._status("AI工场项目已打开", "success")
+
+    def pull_main_images(self, checked=False):
+        alias = str(self.account_combo.currentData() or "").strip()
+        item_id = self.item_id_edit.text().strip()
+        if self.current_project is not None:
+            alias = self.current_project.account_alias
+            item_id = self.current_project.item_id
+        if not alias or not item_id:
+            self._message("项目未打开", "请先选择账号和商品ID并打开项目。")
+            return
+        worker = ImageStudioPullImagesWorker(
+            alias,
+            item_id,
+            db_path=self.db_path,
+            config=self.config,
+        )
+        worker.log.connect(self._append_log)
+        worker.finished.connect(self._on_pull_finished)
+        worker.failed.connect(self._on_worker_failed)
+        self._start_worker(worker, "AI工场拉取主图")
+        self._append_log("[AI工场] 拉取主图开始")
+
+    def _on_pull_finished(self, summary):
+        if summary.get("ok") is False:
+            if self._running_worker is not None:
+                self._on_worker_failed(-1, summary.get("error") or "拉取主图失败")
+            return
+        project = summary.get("project")
+        if project is not None:
+            self.current_project = project
+            self._set_account_combo(project.account_alias)
+            self.item_id_edit.setText(project.item_id)
+        self._finish_worker()
+        self.refresh_projects()
+        if project is not None:
+            self._select_project(project.id)
+        self._status(f"已拉取 {summary.get('count', 0)} 张蝦皮原主图", "success")
+
+    def open_project_folder(self, checked=False):
+        if self.current_project is None:
+            self._message("项目未打开", "请先打开一个AI工场项目。")
+            return
+        dirs = image_studio.default_project_image_dirs(self.current_project, self.config)
+        try:
+            os.makedirs(dirs["root"], exist_ok=True)
+            opened = file_manager.open_in_file_manager(dirs["root"])
+        except Exception as exc:
+            self._message("打开项目文件夹失败", str(exc))
+            self._status(f"打开项目文件夹失败：{exc}", "warning")
+            return
+        self._status(f"已打开项目文件夹：{opened}", "success")
+
+    def _on_project_selection_changed(self):
+        items = self.project_table.selectedItems()
+        if not items:
+            return
+        project_id = items[0].data(Qt.UserRole)
+        if project_id is not None:
+            self._select_project(project_id)
+
+    def _select_project(self, project_id, quiet=False):
+        try:
+            project = image_studio.get_project(project_id, path=self.db_path)
+        except Exception as exc:
+            self._status(f"读取AI工场项目失败：{exc}", "danger")
+            return
+        if project is None:
+            return
+        self.current_project = project
+        self._set_account_combo(project.account_alias)
+        self.item_id_edit.setText(project.item_id)
+        self.prompt_edit.blockSignals(True)
+        try:
+            self.prompt_edit.setPlainText(project.draft_prompt or "")
+        finally:
+            self.prompt_edit.blockSignals(False)
+        self.selected_source_asset_id = None
+        self.refresh_project_assets()
+        self._sync_project_selection(project.id)
+        if not quiet:
+            self._status(f"当前AI工场项目：{project.account_alias} / {project.item_id}", "muted")
+
+    def _sync_project_selection(self, project_id):
+        for row in range(self.project_table.rowCount()):
+            item = self.project_table.item(row, 0)
+            if item is not None and item.data(Qt.UserRole) == int(project_id):
+                if not self.project_table.item(row, 0).isSelected():
+                    self.project_table.selectRow(row)
+                break
+
+    def refresh_project_assets(self):
+        if self.current_project is None:
+            self.assets = []
+            self.jobs = []
+            self.selections = []
+        else:
+            self.assets = image_studio.list_assets(self.current_project.id, path=self.db_path)
+            self.jobs = self._list_project_jobs(self.current_project.id)
+            self.selections = image_studio.list_selections(self.current_project.id, path=self.db_path)
+        self._fill_original_table()
+        self._fill_pool_table()
+        self._refresh_selection_labels()
+        self._refresh_source_label()
+
+    def _fill_original_table(self):
+        originals = [asset for asset in self.assets if asset.kind == image_studio.ASSET_KIND_ORIGINAL]
+        self.original_table.setRowCount(len(originals))
+        for row, asset in enumerate(originals):
+            values = [
+                str(asset.source_order or row + 1),
+                _asset_status_text(asset),
+                asset.remote_url or "",
+            ]
+            for column, value in enumerate(values):
+                item = QTableWidgetItem(str(value or ""))
+                item.setData(Qt.UserRole, {"type": "asset", "asset_id": int(asset.id)})
+                self.original_table.setItem(row, column, item)
+        self.original_table.resizeColumnsToContents()
+
+    def _fill_pool_table(self):
+        rows = []
+        for asset in self.assets:
+            if asset.status == image_studio.ASSET_STATUS_MISSING:
+                continue
+            if asset.kind not in {"original", "generated_main", "generated_detail"}:
+                continue
+            rows.append(("asset", asset))
+        for job in self.jobs:
+            if job.status in {"pending", "submitted", "running", "failed", "expired", "cancelled"}:
+                rows.append(("job", job))
+        self.pool_table.setRowCount(len(rows))
+        for row, (row_type, obj) in enumerate(rows):
+            if row_type == "asset":
+                values = [
+                    _asset_badge(obj.kind),
+                    obj.aspect_ratio or "未知",
+                    _asset_status_text(obj),
+                    _source_text(obj, self.assets),
+                    obj.local_path or "",
+                ]
+                data = {"type": "asset", "asset_id": int(obj.id)}
+            else:
+                values = [
+                    "任务",
+                    "-",
+                    self.JOB_STATUS_LABELS.get(obj.status, obj.status),
+                    f"源图 #{obj.source_asset_id or '-'}",
+                    obj.error or "",
+                ]
+                data = {"type": "job", "job_id": int(obj.id)}
+            for column, value in enumerate(values):
+                item = QTableWidgetItem(str(value or ""))
+                item.setData(Qt.UserRole, data)
+                self.pool_table.setItem(row, column, item)
+        self.pool_table.resizeColumnsToContents()
+
+    def _on_original_clicked(self, row, column):
+        asset = self._asset_from_table_row(self.original_table, row)
+        if asset is not None:
+            self._ensure_original_in_pool(asset, open_after=False)
+
+    def _on_original_double_clicked(self, row, column):
+        asset = self._asset_from_table_row(self.original_table, row)
+        if asset is not None:
+            self._ensure_original_in_pool(asset, open_after=True)
+
+    def _on_pool_clicked(self, row, column):
+        data = self._row_data(self.pool_table, row)
+        if not data or data.get("type") != "asset":
+            return
+        asset = self._asset_by_id(data.get("asset_id"))
+        if asset is not None:
+            self._select_source_asset(asset)
+
+    def _on_pool_double_clicked(self, row, column):
+        data = self._row_data(self.pool_table, row)
+        if not data or data.get("type") != "asset":
+            return
+        asset = self._asset_by_id(data.get("asset_id"))
+        if asset is not None:
+            self._open_preview(asset)
+
+    def _ensure_original_in_pool(self, asset, open_after=False):
+        if str(asset.local_path or "").strip() and os.path.isfile(asset.local_path):
+            self._select_source_asset(asset)
+            if open_after:
+                self._open_preview(asset)
+            return
+        worker = ImageStudioDownloadOriginalWorker(
+            asset.id,
+            db_path=self.db_path,
+            config=self.config,
+            open_after=open_after,
+        )
+        worker.log.connect(self._append_log)
+        worker.finished.connect(self._on_download_finished)
+        worker.failed.connect(self._on_worker_failed)
+        self._start_worker(worker, "AI工场下载原图")
+
+    def _on_download_finished(self, summary):
+        if summary.get("ok") is False:
+            if self._running_worker is not None:
+                self._on_worker_failed(-1, summary.get("error") or "下载原图失败")
+            return
+        asset = summary.get("asset")
+        self._finish_worker()
+        self.refresh_project_assets()
+        if asset is not None:
+            refreshed = self._asset_by_id(asset.id) or asset
+            self._select_source_asset(refreshed)
+            if summary.get("open_after"):
+                self._open_preview(refreshed)
+        self._status("原图已加入照片池", "success")
+
+    def _select_source_asset(self, asset):
+        self.selected_source_asset_id = int(asset.id)
+        self._refresh_source_label()
+        self._status(f"已选择源图：{_asset_badge(asset.kind)} #{asset.id}", "success")
+
+    def _refresh_source_label(self):
+        asset = self._asset_by_id(self.selected_source_asset_id)
+        if asset is None:
+            self.source_label.setText("源图：未选择")
+            return
+        self.source_label.setText(f"源图：{_asset_badge(asset.kind)} #{asset.id}")
+
+    def _open_preview(self, asset):
+        dialog = ImageStudioPreviewDialog(asset, self)
+        dialog.exec()
+
+    def _show_pool_context_menu(self, position):
+        row = self.pool_table.rowAt(position.y())
+        data = self._row_data(self.pool_table, row)
+        if not data or data.get("type") != "asset":
+            return
+        asset_id = data.get("asset_id")
+        menu = QMenu(self)
+        try:
+            counts = image_studio.asset_reference_counts(asset_id, path=self.db_path)
+            referenced = bool(counts.get("total"))
+        except Exception:
+            referenced = True
+        remove_action = menu.addAction(
+            "移除照片" if not referenced else "移除照片（已被任务或终选引用）"
+        )
+        remove_action.setEnabled(not referenced)
+        action = menu.exec(self.pool_table.viewport().mapToGlobal(position))
+        if action is remove_action and not referenced:
+            self.remove_asset(asset_id)
+
+    def remove_asset(self, asset_id):
+        if not self._confirm("移除照片", "只从AI工场照片池移除记录，不删除本地图片文件。"):
+            return
+        try:
+            image_studio.remove_asset_if_unused(asset_id, path=self.db_path)
+        except Exception as exc:
+            self._message("不能移除照片", str(exc))
+            return
+        if self.selected_source_asset_id == int(asset_id):
+            self.selected_source_asset_id = None
+        self.refresh_project_assets()
+        self._status("照片已从池中移除", "success")
+
+    def refresh_templates(self, selected=None):
+        current = selected or self.template_combo.currentData()
+        self.template_combo.blockSignals(True)
+        try:
+            self.template_combo.clear()
+            self.template_combo.addItem("选择模板", "")
+            for name in prompts.list_image_studio_templates(self.prompts_dir):
+                self.template_combo.addItem(name, name)
+            if current:
+                index = self.template_combo.findData(current)
+                if index >= 0:
+                    self.template_combo.setCurrentIndex(index)
+        finally:
+            self.template_combo.blockSignals(False)
+
+    def load_selected_template(self, index=None):
+        name = self.template_combo.currentData()
+        if not name:
+            return
+        try:
+            self.prompt_edit.setPlainText(prompts.load_image_studio_template(name, self.prompts_dir))
+        except Exception as exc:
+            self._message("加载模板失败", str(exc))
+
+    def create_template(self, checked=False):
+        name, ok = QInputDialog.getText(self, "新建模板", "模板名称")
+        if not ok:
+            return
+        try:
+            prompts.save_image_studio_template(name, self.prompt_edit.toPlainText(), self.prompts_dir)
+        except Exception as exc:
+            self._message("新建模板失败", str(exc))
+            return
+        self.refresh_templates(selected=name)
+        self._status("AI工场模板已新建", "success")
+
+    def rename_template(self, checked=False):
+        old = self.template_combo.currentData()
+        if not old:
+            self._message("未选择模板", "请先选择要重命名的模板。")
+            return
+        new, ok = QInputDialog.getText(self, "重命名模板", "新模板名称", text=old)
+        if not ok:
+            return
+        try:
+            prompts.rename_image_studio_template(old, new, self.prompts_dir)
+        except Exception as exc:
+            self._message("重命名模板失败", str(exc))
+            return
+        self.refresh_templates(selected=new)
+        self._status("AI工场模板已重命名", "success")
+
+    def save_template(self, checked=False):
+        name = self.template_combo.currentData()
+        if not name:
+            name, ok = QInputDialog.getText(self, "保存模板", "模板名称")
+            if not ok:
+                return
+        try:
+            prompts.save_image_studio_template(name, self.prompt_edit.toPlainText(), self.prompts_dir)
+        except Exception as exc:
+            self._message("保存模板失败", str(exc))
+            return
+        self.refresh_templates(selected=name)
+        self._status("AI工场模板已保存", "success")
+
+    def delete_template(self, checked=False):
+        name = self.template_combo.currentData()
+        if not name:
+            self._message("未选择模板", "请先选择要删除的模板。")
+            return
+        if not self._confirm("删除模板", f"确定删除模板「{name}」吗？"):
+            return
+        try:
+            prompts.delete_image_studio_template(name, self.prompts_dir)
+        except Exception as exc:
+            self._message("删除模板失败", str(exc))
+            return
+        self.refresh_templates()
+        self._status("AI工场模板已删除", "success")
+
+    def _save_project_prompt(self):
+        if self.current_project is None:
+            return
+        try:
+            self.current_project = image_studio.update_project_prompt(
+                self.current_project.id,
+                self.prompt_edit.toPlainText(),
+                path=self.db_path,
+            )
+        except Exception as exc:
+            self._status(f"保存AI工场草稿提示词失败：{exc}", "warning")
+
+    def start_generation(self, checked=False):
+        if self.current_project is None:
+            self._message("项目未打开", "请先打开一个AI工场项目。")
+            return
+        source = self._asset_by_id(self.selected_source_asset_id)
+        if source is None:
+            self._message("未选择源图", "请先在照片池单击选择一张源图。")
+            return
+        if not str(source.local_path or "").strip() or not os.path.isfile(source.local_path):
+            self._message("源图不可用", "请先单击蝦皮原主图下载到本地后再生成。")
+            return
+        prompt_text = self.prompt_edit.toPlainText().strip()
+        if not prompt_text:
+            self._message("提示词不能为空", "请输入完整图片生成提示词。")
+            return
+        count = self.count_spin.value()
+        self.progress_bar.setRange(0, count)
+        self.progress_bar.setValue(0)
+        self.log_view.clear()
+        self._append_log(f"[AI工场] 本轮生图开始：{count} 张，来源 cmhub 托管模型")
+        worker = ImageStudioGenerateJobsWorker(
+            self.current_project.id,
+            source.id,
+            prompt_text,
+            count,
+            job_type=self.job_type_combo.currentData(),
+            aspect_ratio=self.aspect_combo.currentData(),
+            db_path=self.db_path,
+            config=self.config,
+            cmhub_config_path=self.cmhub_config_path,
+        )
+        worker.progress.connect(self._on_generate_progress)
+        worker.log.connect(self._append_log)
+        worker.finished.connect(self._on_generation_finished)
+        worker.failed.connect(self._on_worker_failed)
+        self._start_worker(worker, "AI工场生成图片")
+
+    def stop_generation(self, checked=False):
+        if self._running_worker is not None and hasattr(self._running_worker, "cancel"):
+            self._running_worker.cancel()
+            self._append_log("[AI工场] 已请求停止，正在等待安全边界")
+            self._status("AI工场生成已请求停止", "warning")
+
+    def _on_generate_progress(self, payload):
+        total = max(1, int(payload.get("total") or self.progress_bar.maximum() or 1))
+        done = min(total, int(payload.get("done") or 0))
+        self.progress_bar.setRange(0, total)
+        self.progress_bar.setValue(done)
+        if payload.get("points_balance") is not None:
+            text = f"cmhub 托管模型：余额 {payload.get('points_balance')}"
+            if payload.get("points_cost") is not None:
+                text += f"，本张扣点 {payload.get('points_cost')}"
+            self.billing_label.setText(text)
+
+    def _on_generation_finished(self, summary):
+        if summary.get("ok") is False:
+            if self._running_worker is not None:
+                self._on_worker_failed(-1, summary.get("error") or "AI工场生成失败")
+            return
+        self._finish_worker()
+        self.refresh_project_assets()
+        total = int(summary.get("total") or 0)
+        success = int(summary.get("success") or 0)
+        failed = int(summary.get("failed") or 0)
+        cancelled = int(summary.get("cancelled") or 0)
+        self._append_log(f"[AI工场] 本轮完成：总数{total}，成功{success}，失败{failed}，停止{cancelled}")
+        level = "warning" if failed or cancelled else "success"
+        self._status(f"AI工场生成完成：成功{success}，失败{failed}，停止{cancelled}", level)
+
+    def _on_worker_failed(self, row, error):
+        self._finish_worker()
+        message = diagnostics.redact_log_text(str(error or "未知错误"))
+        self._append_log(f"[AI工场] 失败：{message}")
+        self._status(f"AI工场任务失败：{message}", "danger")
+        self._message("AI工场任务失败", message)
+        self.refresh_project_assets()
+
+    def _start_worker(self, worker, thread_name):
+        self._set_running(True)
+        self._running_worker = worker
+        self._running_thread = run_worker(worker, thread_name)
+
+    def _finish_worker(self):
+        self._running_worker = None
+        self._running_thread = None
+        self._set_running(False)
+
+    def _set_running(self, running):
+        self.open_project_button.setEnabled(not running)
+        self.pull_images_button.setEnabled(not running)
+        self.open_folder_button.setEnabled(not running)
+        self.project_table.setEnabled(not running)
+        self.original_table.setEnabled(not running)
+        self.pool_table.setEnabled(not running)
+        self.template_combo.setEnabled(not running)
+        self.template_new_button.setEnabled(not running)
+        self.template_rename_button.setEnabled(not running)
+        self.template_save_button.setEnabled(not running)
+        self.template_delete_button.setEnabled(not running)
+        self.prompt_edit.setEnabled(not running)
+        self.job_type_combo.setEnabled(not running)
+        self.count_spin.setEnabled(not running)
+        self.aspect_combo.setEnabled(not running)
+        self.start_button.setEnabled(not running)
+        self.stop_button.setEnabled(running)
+
+    def _set_account_combo(self, alias):
+        index = self.account_combo.findData(alias)
+        if index >= 0:
+            self.account_combo.setCurrentIndex(index)
+
+    def _list_project_jobs(self, project_id):
+        conn = db.connect(self.db_path)
+        try:
+            rows = conn.execute(
+                """
+                SELECT * FROM image_studio_jobs
+                WHERE project_id = ?
+                ORDER BY updated_at DESC, id DESC
+                """,
+                (int(project_id),),
+            ).fetchall()
+            return [image_studio.ImageStudioJob(**dict(row)) for row in rows]
+        finally:
+            conn.close()
+
+    def _refresh_selection_labels(self):
+        main_count = sum(1 for item in self.selections if item.selection_type == "main")
+        detail_count = sum(1 for item in self.selections if item.selection_type == "detail")
+        main_target = getattr(self.current_project, "target_main_count", 9) if self.current_project else 9
+        detail_target = getattr(self.current_project, "target_detail_count", 12) if self.current_project else 12
+        self.main_selection_label.setText(f"主图终选 {main_count}/{main_target}（拖放排序将在 T-592 接入）")
+        self.detail_selection_label.setText(f"详情图终选 {detail_count}/{detail_target}（拖放排序将在 T-592 接入）")
+
+    def _asset_from_table_row(self, table, row):
+        data = self._row_data(table, row)
+        if not data or data.get("type") != "asset":
+            return None
+        return self._asset_by_id(data.get("asset_id"))
+
+    def _row_data(self, table, row):
+        if row < 0 or row >= table.rowCount():
+            return None
+        item = table.item(row, 0)
+        if item is None:
+            return None
+        return item.data(Qt.UserRole)
+
+    def _asset_by_id(self, asset_id):
+        if asset_id is None:
+            return None
+        for asset in self.assets:
+            if int(asset.id) == int(asset_id):
+                return asset
+        try:
+            return image_studio.get_asset(asset_id, path=self.db_path)
+        except Exception:
+            return None
+
+    def _append_log(self, message):
+        text = diagnostics.redact_log_text(str(message or ""))
+        self.log_view.appendPlainText(text)
+        scrollbar = self.log_view.verticalScrollBar()
+        scrollbar.setValue(scrollbar.maximum())
+
+    def _message(self, title, text):
+        box = QMessageBox(self)
+        box.setWindowTitle(str(title or "提示"))
+        box.setText(str(text or ""))
+        ok_button = box.addButton("确定", QMessageBox.AcceptRole)
+        box.setDefaultButton(ok_button)
+        box.exec()
+
+    def _confirm(self, title, text):
+        box = QMessageBox(self)
+        box.setWindowTitle(str(title or "确认"))
+        box.setText(str(text or ""))
+        yes_button = box.addButton("确定", QMessageBox.AcceptRole)
+        box.addButton("取消", QMessageBox.RejectRole)
+        box.setDefaultButton(yes_button)
+        box.exec()
+        return box.clickedButton() is yes_button
+
+    def _status(self, message, level="muted"):
+        _emit_status(self.status_callback, message, level=level)
+
+
+def _asset_badge(kind):
+    return {
+        "original": "原图",
+        "generated_main": "主图",
+        "generated_detail": "详情图",
+    }.get(str(kind or ""), str(kind or "图片"))
+
+
+def _asset_status_text(asset):
+    status = str(getattr(asset, "status", "") or "")
+    local_path = str(getattr(asset, "local_path", "") or "")
+    if status == image_studio.ASSET_STATUS_MISSING:
+        return "文件缺失"
+    if local_path and os.path.isfile(local_path):
+        return "可用"
+    if getattr(asset, "remote_url", None):
+        return "远程待下载"
+    return "待生成"
+
+
+def _source_text(asset, assets):
+    parent_id = getattr(asset, "parent_asset_id", None)
+    if not parent_id:
+        return "原始来源"
+    for item in assets:
+        if int(item.id) == int(parent_id):
+            return f"{_asset_badge(item.kind)} #{item.id}"
+    return f"源图 #{parent_id}"

app/gui/widgets.py

@@ -60,6 +60,7 @@ TAB_TITLES = [
     "③ 更新蝦皮",
     "④ 账号管理",
     "⑤ 设置",
+    "⑥ AI工场",
 ]
 
 TAB_STYLE = """

app/gui/workers.py

@@ -4,7 +4,9 @@ from __future__ import annotations
 
 import datetime as _dt
 import re
+import threading
 
+from .. import image_studio, image_studio_generation, image_studio_images
 from .widgets import *
 
 
@@ -14,6 +16,182 @@ _USER_LOG_PATH_RE = re.compile(
 )
 
 
+def _image_studio_step_label(step):
+    return {
+        "ensure_chrome": "准备账号浏览器",
+        "login_check": "检测登录",
+        "open_product": "打开商品页",
+        "read_main_images": "读取蝦皮原主图",
+        "cover_submit": "提交生图任务",
+        "cover_poll": "查询生图结果",
+        "cover_download": "下载生成图片",
+        "job_done": "完成单张任务",
+    }.get(str(step or ""), str(step or "处理"))
+
+
+def _image_studio_result_label(result):
+    return {
+        "start": "开始",
+        "success": "成功",
+        "failed": "失败",
+        "cancelled": "已停止",
+        "blocked": "已阻断",
+        "reused": "已复用",
+        "launched": "已启动",
+        "resume": "继续查询",
+        "queued": "排队中",
+        "running": "生成中",
+        "expired": "已过期",
+    }.get(str(result or ""), str(result or ""))
+
+
+def _format_image_studio_event(event):
+    event = dict(event or {})
+    step = _image_studio_step_label(event.get("step"))
+    result = _image_studio_result_label(event.get("result"))
+    detail = str(event.get("detail") or "").strip()
+    job_id = event.get("job_id")
+    prefix = f"[AI工场] {step}"
+    if job_id is not None:
+        prefix += f" #{job_id}"
+    if result:
+        prefix += f"：{result}"
+    if detail:
+        prefix += f"，{diagnostics.redact_log_text(detail)}"
+    if event.get("points_cost") is not None:
+        prefix += f"，扣点 {event.get('points_cost')}"
+    if event.get("points_balance") is not None:
+        prefix += f"，余额 {event.get('points_balance')}"
+    return prefix
+
+
+class ImageStudioPullImagesWorker(BaseWorker):
+    """Read Shopee main image URLs for one AI studio project in background."""
+
+    def __init__(self, account_alias, item_id, *, db_path=None, config=None):
+        super().__init__()
+        self.account_alias = account_alias
+        self.item_id = item_id
+        self.db_path = db_path
+        self.config = config
+
+    def execute(self):
+        def on_step(payload):
+            self.log.emit(_format_image_studio_event(payload))
+
+        result = image_studio.pull_remote_main_image_urls(
+            self.account_alias,
+            self.item_id,
+            path=self.db_path,
+            config=self.config,
+            on_step=on_step,
+        )
+        project = result.get("project")
+        assets = result.get("assets") or []
+        return {
+            "project": project,
+            "assets": assets,
+            "count": len(assets),
+            "account": result.get("account"),
+        }
+
+
+class ImageStudioDownloadOriginalWorker(BaseWorker):
+    """Download one remote original image into the project pool."""
+
+    def __init__(self, asset_id, *, db_path=None, config=None, open_after=False):
+        super().__init__()
+        self.asset_id = int(asset_id)
+        self.db_path = db_path
+        self.config = config
+        self.open_after = bool(open_after)
+
+    def execute(self):
+        self.log.emit("[AI工场] 下载原图：开始")
+        asset = image_studio_images.download_original_asset(
+            self.asset_id,
+            path=self.db_path,
+            config=self.config,
+        )
+        self.log.emit("[AI工场] 下载原图：成功")
+        return {"asset": asset, "open_after": self.open_after}
+
+
+class ImageStudioGenerateJobsWorker(BaseWorker):
+    """Run cmhub hosted image generation jobs for the AI studio."""
+
+    def __init__(
+        self,
+        project_id,
+        source_asset_id,
+        prompt,
+        count,
+        *,
+        job_type="main",
+        aspect_ratio="1:1",
+        db_path=None,
+        config=None,
+        cmhub_config_path=None,
+    ):
+        super().__init__()
+        self.project_id = int(project_id)
+        self.source_asset_id = int(source_asset_id)
+        self.prompt = str(prompt or "")
+        self.count = int(count or 0)
+        self.job_type = str(job_type or "main")
+        self.aspect_ratio = str(aspect_ratio or "1:1")
+        self.db_path = db_path
+        self.config = config
+        self.cmhub_config_path = cmhub_config_path
+        self._done = 0
+        self._failed = 0
+        self._lock = threading.Lock()
+
+    def execute(self):
+        self.progress.emit({"total": self.count, "done": 0, "failed": 0})
+
+        def on_event(payload):
+            event = dict(payload or {})
+            self.log.emit(_format_image_studio_event(event))
+            if event.get("points_cost") is not None or event.get("points_balance") is not None:
+                self.progress.emit(
+                    {
+                        "total": self.count,
+                        "done": self._done,
+                        "failed": self._failed,
+                        "points_cost": event.get("points_cost"),
+                        "points_balance": event.get("points_balance"),
+                    }
+                )
+            if event.get("step") == "job_done":
+                with self._lock:
+                    self._done += 1
+                    if event.get("result") not in {"success"}:
+                        self._failed += 1
+                    progress = {
+                        "total": self.count,
+                        "done": self._done,
+                        "failed": self._failed,
+                    }
+                self.progress.emit(progress)
+
+        summary = image_studio_generation.generate_image_jobs(
+            self.project_id,
+            self.source_asset_id,
+            self.prompt,
+            self.count,
+            job_type=self.job_type,
+            aspect_ratio=self.aspect_ratio,
+            config=self.config,
+            cmhub_config_path=self.cmhub_config_path,
+            path=self.db_path,
+            should_stop=self.should_cancel,
+            on_event=on_event,
+        )
+        summary["project_id"] = self.project_id
+        return summary
+
+
 def _generation_mode_label(mode):
     mode = appconfig.normalize_generate_mode(mode)
     return {