feat(gui): replace AI studio with product suite

2026-07-14 09:53:13 +08:00
parent fb873aae90
commit bc115ba0d7
27 changed files with 3453 additions and 81 deletions
@@ -0,0 +1,74 @@
+import os
+import sys
+import unittest
+from types import SimpleNamespace
+
+sys.path.insert(0, os.path.dirname(__file__))
+
+from app import product_suite
+
+
+class ProductSuiteTests(unittest.TestCase):
+    def test_defaults_and_per_image_total(self):
+        settings = product_suite.default_suite_settings()
+
+        self.assertEqual(["白底图", "场景图", "卖点图"], product_suite.category_order(settings))
+        self.assertEqual(5, product_suite.suite_total_count(settings, 3))
+
+        settings["per_image_primary"] = True
+
+        self.assertEqual(13, product_suite.suite_total_count(settings, 3))
+
+    def test_custom_category_validation_and_normalization(self):
+        self.assertEqual("分类名称不能为空", product_suite.suite_name_error(""))
+        self.assertEqual("分类名称不能包含空格", product_suite.suite_name_error("使用 场景"))
+        self.assertEqual("分类名称不能超过10个字", product_suite.suite_name_error("一二三四五六七八九十甲"))
+        self.assertEqual("分类名称已存在", product_suite.suite_name_error("场景图", {"场景图"}))
+
+        settings = product_suite.normalize_suite_settings(
+            {
+                "ratio": "3:4",
+                "categories": {"白底图": 2, "场景图": 0, "卖点图": 1, "尺寸图": 2},
+                "custom_category_order": ["尺寸图"],
+            }
+        )
+
+        self.assertEqual("3:4", settings["ratio"])
+        self.assertEqual(["白底图", "场景图", "卖点图", "尺寸图"], product_suite.category_order(settings))
+        self.assertEqual(5, product_suite.suite_total_count(settings, 1))
+
+    def test_job_specs_include_selected_context_and_source_assignment(self):
+        settings = product_suite.default_suite_settings()
+        settings.update(
+            {
+                "platform": "Shopee",
+                "country": "中国台湾",
+                "language": "繁体中文",
+                "ratio": "4:3",
+                "per_image_primary": True,
+                "categories": {"白底图": 1, "场景图": 1, "卖点图": 0},
+            }
+        )
+        assets = [SimpleNamespace(id=11), SimpleNamespace(id=12)]
+
+        specs = product_suite.build_job_specs(
+            assets,
+            "40小时续航，适合通勤",
+            settings,
+            "51100639510",
+        )
+
+        self.assertEqual(3, len(specs))
+        self.assertEqual([11, 11, 12], [spec["source_asset_id"] for spec in specs])
+        self.assertEqual(["白底图", "场景图", "场景图"], [spec["job_type"] for spec in specs])
+        for spec in specs:
+            self.assertIn("平台：Shopee", spec["prompt"])
+            self.assertIn("国家地区：中国台湾", spec["prompt"])
+            self.assertIn("输出语言：繁体中文", spec["prompt"])
+            self.assertIn("图片比例：4:3", spec["prompt"])
+            self.assertIn("商品ID：51100639510", spec["prompt"])
+            self.assertIn("40小时续航", spec["prompt"])
+
+
+if __name__ == "__main__":
+    unittest.main()