fix: randomize tuhui designer alias
This commit is contained in:
@@ -4,6 +4,7 @@ import json
|
|||||||
import logging
|
import logging
|
||||||
import mimetypes
|
import mimetypes
|
||||||
import os
|
import os
|
||||||
|
import random
|
||||||
import re
|
import re
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
from typing import Dict, List, Optional
|
from typing import Dict, List, Optional
|
||||||
@@ -24,7 +25,6 @@ logger = logging.getLogger("cs_agent")
|
|||||||
|
|
||||||
AUTO_PROCESS_PRICE = int(os.getenv("AUTO_PROCESS_DEFAULT_PRICE", "12"))
|
AUTO_PROCESS_PRICE = int(os.getenv("AUTO_PROCESS_DEFAULT_PRICE", "12"))
|
||||||
AUTO_PROCESS_CATEGORY = os.getenv("AUTO_PROCESS_CATEGORY", "设计素材")
|
AUTO_PROCESS_CATEGORY = os.getenv("AUTO_PROCESS_CATEGORY", "设计素材")
|
||||||
AUTO_PROCESS_DESIGNER_ALIAS = os.getenv("AUTO_PROCESS_DESIGNER_ALIAS", "图绘平台")
|
|
||||||
AUTO_PROCESS_ROOT = Path(
|
AUTO_PROCESS_ROOT = Path(
|
||||||
os.getenv("AUTO_PROCESS_ROOT", str(Path(__file__).resolve().parents[1] / "runtime" / "auto_processed"))
|
os.getenv("AUTO_PROCESS_ROOT", str(Path(__file__).resolve().parents[1] / "runtime" / "auto_processed"))
|
||||||
)
|
)
|
||||||
@@ -52,6 +52,8 @@ _CONTENT_TYPE_SUFFIX = {
|
|||||||
"image/avif": ".avif",
|
"image/avif": ".avif",
|
||||||
"image/gif": ".gif",
|
"image/gif": ".gif",
|
||||||
}
|
}
|
||||||
|
_DESIGNER_ALIAS_PREFIXES = ("青木", "星野", "白川", "南栀", "言川", "木也", "安可", "拾光", "云岸", "知禾")
|
||||||
|
_DESIGNER_ALIAS_SUFFIXES = ("设计", "studio", "视觉", "创意", "图像", "工坊", "素材", "像素")
|
||||||
|
|
||||||
|
|
||||||
def _safe_name(text: str, fallback: str = "image") -> str:
|
def _safe_name(text: str, fallback: str = "image") -> str:
|
||||||
@@ -95,6 +97,10 @@ def _build_upload_title(intent: str, analysis: Dict, requirement_text: str, idx:
|
|||||||
return base
|
return base
|
||||||
|
|
||||||
|
|
||||||
|
def _build_designer_alias() -> str:
|
||||||
|
return f"{random.choice(_DESIGNER_ALIAS_PREFIXES)}{random.choice(_DESIGNER_ALIAS_SUFFIXES)}"
|
||||||
|
|
||||||
|
|
||||||
class AutoImagePipelineService:
|
class AutoImagePipelineService:
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
self.customer_db = CustomerDatabase()
|
self.customer_db = CustomerDatabase()
|
||||||
@@ -300,7 +306,7 @@ class AutoImagePipelineService:
|
|||||||
price=AUTO_PROCESS_PRICE,
|
price=AUTO_PROCESS_PRICE,
|
||||||
category=AUTO_PROCESS_CATEGORY,
|
category=AUTO_PROCESS_CATEGORY,
|
||||||
tags="AI处理,自动转接",
|
tags="AI处理,自动转接",
|
||||||
designer_name=AUTO_PROCESS_DESIGNER_ALIAS,
|
designer_name=_build_designer_alias(),
|
||||||
)
|
)
|
||||||
if not upload_result.success:
|
if not upload_result.success:
|
||||||
if task_id:
|
if task_id:
|
||||||
|
|||||||
Reference in New Issue
Block a user