feat: split generate title and image statuses

app/gui/models.py

@@ -155,7 +155,7 @@ class TaskTableModel(QAbstractTableModel):
 class GenerateTaskTableModel(QAbstractTableModel):
     """Table model for Tab 2 generation candidates."""
 
-    HEADERS = ["店铺", "商品ID", "旧标题", "新标题", "状态"]
+    HEADERS = ["店铺", "商品ID", "旧标题", "新标题", "标题状态", "图片状态"]
 
     STATUS_TEXT = {
         "running": "处理中",
@@ -178,6 +178,15 @@ class GenerateTaskTableModel(QAbstractTableModel):
         self.db_path = db_path
         self.status_callback = status_callback
         self.last_edit_error = None
+        self.generate_cover_enabled = False
+
+    def set_generate_cover_enabled(self, enabled):
+        self.generate_cover_enabled = bool(enabled)
+        if not self.tasks:
+            return
+        top_left = self.index(0, 4)
+        bottom_right = self.index(len(self.tasks) - 1, 5)
+        self.dataChanged.emit(top_left, bottom_right, [Qt.DisplayRole, Qt.ForegroundRole])
 
     def set_tasks(self, tasks, accounts):
         self.beginResetModel()
@@ -208,8 +217,8 @@ class GenerateTaskTableModel(QAbstractTableModel):
         task = self.tasks[index.row()]
         if role in (Qt.DisplayRole, Qt.EditRole):
             return self._display_value(task, index.column())
-        if role == Qt.ForegroundRole and index.column() == 4:
-            return self._status_color(task)
+        if role == Qt.ForegroundRole and index.column() in {4, 5}:
+            return self._component_status_color(self._display_value(task, index.column()))
         if role == Qt.ToolTipRole:
             if index.column() == 3 and self._can_edit_title(task):
                 return "双击可微调新标题，只修改本地待更新内容"
@@ -273,13 +282,6 @@ class GenerateTaskTableModel(QAbstractTableModel):
             return account.account_name
         return task.account_name or task.alias or ""
 
-    def _status_text(self, task):
-        if task.status == "failed":
-            return self._failed_status_text(task)
-        if task.status in self.STATUS_TEXT and task.status != "pending":
-            return self.STATUS_TEXT[task.status]
-        return self.STAGE_TEXT.get(task.stage, task.stage)
-
     def _failed_status_text(self, task):
         collect_attempts = self._attempt_count(task, "collect_attempts")
         generate_attempts = self._attempt_count(task, "generate_attempts")
@@ -298,12 +300,97 @@ class GenerateTaskTableModel(QAbstractTableModel):
         except (TypeError, ValueError):
             return 0
 
-    def _status_color(self, task):
-        base_color = _status_base_color(getattr(task, "status", None))
-        if base_color is not None:
-            return _qcolor(base_color)
-        if getattr(task, "stage", None) in {"generated", "applied"}:
+    def _has_new_title(self, task):
+        return bool(str(getattr(task, "new_title", "") or "").strip())
+
+    def _has_new_cover(self, task):
+        return bool(str(getattr(task, "new_cover_path", "") or "").strip())
+
+    def _generate_failed(self, task):
+        return (
+            getattr(task, "status", None) == "failed"
+            and self._attempt_count(task, "generate_attempts") > 0
+        )
+
+    def _last_error_text(self, task):
+        return str(getattr(task, "last_error", "") or "").lower()
+
+    def _is_title_failure(self, task):
+        if not self._generate_failed(task):
+            return False
+        error = self._last_error_text(task)
+        if any(marker in error for marker in ("title", "标题", "生文", "文本")):
+            return True
+        return not self._has_new_title(task) and not self._is_cover_failure(task)
+
+    def _is_cover_failure(self, task):
+        if not self._generate_failed(task):
+            return False
+        error = self._last_error_text(task)
+        return any(
+            marker in error
+            for marker in ("cover", "封面", "图片", "图像", "新封面")
+        )
+
+    def _is_collect_incomplete(self, task):
+        return (
+            getattr(task, "stage", None) == "imported"
+            and self._attempt_count(task, "generate_attempts") == 0
+            and not self._is_collect_failure(task)
+        )
+
+    def _is_collect_failure(self, task):
+        return (
+            getattr(task, "status", None) == "failed"
+            and self._attempt_count(task, "collect_attempts") > 0
+            and self._attempt_count(task, "generate_attempts") == 0
+        )
+
+    def _title_status_text(self, task):
+        if getattr(task, "status", None) == "skipped":
+            return "略过"
+        if self._is_collect_failure(task):
+            return "采集失败"
+        if self._is_collect_incomplete(task):
+            return "未采集"
+        if self._is_title_failure(task):
+            return "失败"
+        if self._has_new_title(task):
+            return "已生成"
+        if getattr(task, "status", None) == "running":
+            return "处理中"
+        if getattr(task, "stage", None) == "applied":
+            return "已生成" if self._has_new_title(task) else "未生成"
+        return "待生成"
+
+    def _cover_status_text(self, task):
+        if getattr(task, "status", None) == "skipped":
+            return "略过"
+        if self._is_collect_failure(task):
+            return "采集失败"
+        if self._is_collect_incomplete(task):
+            return "未采集"
+        if self._has_new_cover(task):
+            return "已生成"
+        if self._is_cover_failure(task):
+            return "失败"
+        if not self.generate_cover_enabled:
+            return "未启用"
+        if self._is_title_failure(task) or not self._has_new_title(task):
+            return "未开始"
+        if getattr(task, "status", None) == "running":
+            return "处理中"
+        return "待生成"
+
+    def _component_status_color(self, text):
+        if text in {"失败", "采集失败"}:
+            return _qcolor(COLOR_DANGER)
+        if text == "处理中":
+            return _qcolor(COLOR_INFO)
+        if text == "已生成":
             return _qcolor(COLOR_SUCCESS)
+        if text in {"未启用", "未采集", "未开始", "略过", "已取消"}:
+            return _qcolor(COLOR_MUTED)
         return _qcolor(COLOR_PENDING)
 
     def _display_value(self, task, column):
@@ -312,7 +399,8 @@ class GenerateTaskTableModel(QAbstractTableModel):
             task.item_id,
             task.old_title or "",
             task.new_title or "",
-            self._status_text(task),
+            self._title_status_text(task),
+            self._cover_status_text(task),
         ]
         return values[column] if 0 <= column < len(values) else None