"""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}"