通义万相AI绘画教程

import randomimport hashlibimport timefrom typing import List, Dict, Optionalfrom dataclasses import dataclass, field  @dataclassclass WanxConfig:    """通义万相API配置"""    api_key: str = ""    endpoint: str = "https://dashscope.aliyuncs.com/api/v1"    model: str = "wanx-v1"    timeout: int = 60  @dataclassclass ImageGenerationRequest:    """图像生成请求"""    prompt: str    negative_prompt: str = ""    style: str = "default"    resolution: str = "1024*1024"    count: int = 1    seed: int = -1  @dataclassclass GeneratedImage:    """生成的图像结果"""    image_url: str    width: int    height: int    style: str    seed: int    task_id: str  class WanxClient:    """通义万相API客户端（模拟）"""     AVAILABLE_STYLES = [        "default", "chinese_painting", "watercolor",        "3d_render", "realistic", "anime",        "oil_painting", "sketch", "pixel_art"    ]     RESOLUTIONS = [        "512*512", "768*768", "1024*1024",        "1024*768", "768*1024"    ]     def __init__(self, config: WanxConfig):        self.config = config        self.task_counter = 0        self.generated_images = []     def _validate_request(self, req: ImageGenerationRequest) -> bool:        """验证请求参数合法性"""        if not req.prompt or len(req.prompt.strip()) == 0:            raise ValueError("提示词不能为空")        if req.style not in self.AVAILABLE_STYLES:            raise ValueError(                f"不支持的风格: {req.style}，"                f"可选: {self.AVAILABLE_STYLES}"            )        if req.resolution not in self.RESOLUTIONS:            raise ValueError(                f"不支持的分辨率: {req.resolution}，"                f"可选: {self.RESOLUTIONS}"            )        if req.count < 1 or req.count > 4:            raise ValueError("生成数量需在1-4之间")        return True     def generate(self, req: ImageGenerationRequest) -> List[GeneratedImage]:        """调用通义万相API生成图像"""        self._validate_request(req)        self.task_counter += 1        task_id = f"task_{self.task_counter:04d}"         results = []        for i in range(req.count):            seed = req.seed if req.seed >= 0 else random.randint(0, 999999)            url_hash = hashlib.md5(                f"{req.prompt}{seed}{i}".encode()            ).hexdigest()[:12]             w, h = req.resolution.split("*")            img = GeneratedImage(                image_url=f"https://wanx-output.aliyuncs.com/{task_id}/{url_hash}.png",                width=int(w),                height=int(h),                style=req.style,                seed=seed,                task_id=task_id            )            results.append(img)            self.generated_images.append(img)         return results     def style_transfer(self, content_prompt: str,                       style_reference: str) -> GeneratedImage:        """风格迁移：将内容图应用参考图的风格"""        self.task_counter += 1        task_id = f"task_{self.task_counter:04d}"        seed = random.randint(0, 999999)        url_hash = hashlib.md5(            f"{content_prompt}{style_reference}{seed}".encode()        ).hexdigest()[:12]         img = GeneratedImage(            image_url=f"https://wanx-output.aliyuncs.com/{task_id}/{url_hash}.png",            width=1024,            height=1024,            style=f"transfer_{style_reference}",            seed=seed,            task_id=task_id        )        self.generated_images.append(img)        return img     def batch_generate(self, prompts: List[str],                       style: str = "default") -> List[GeneratedImage]:        """批量生成图像：用于电商等场景的大批量素材生产"""        all_results = []        for prompt in prompts:            req = ImageGenerationRequest(                prompt=prompt, style=style, count=1            )            results = self.generate(req)            all_results.extend(results)        return all_results     def get_stats(self) -> Dict:        """获取生成统计"""        style_counts = {}        for img in self.generated_images:            style_counts[img.style] = style_counts.get(img.style, 0) + 1        return {            "total_generated": len(self.generated_images),            "total_tasks": self.task_counter,            "style_distribution": style_counts        }  def demo_wanx_workflow():    """演示通义万相AI绘画完整工作流"""    random.seed(42)     print("=" * 60)    print("通义万相 AI绘画 - 工作流演示")    print("=" * 60)     config = WanxConfig(        api_key="sk-demo-key-xxxxx",        model="wanx-v1"    )    client = WanxClient(config)     print("\n--- 1. 文生图：中文语义理解 ---")    req1 = ImageGenerationRequest(        prompt="水墨山水画，远山如黛，云雾缭绕，一叶扁舟泛于湖上",        style="chinese_painting",        resolution="1024*768",        count=2    )    results1 = client.generate(req1)    for img in results1:        print(f"  风格: {img.style} 尺寸: {img.width}x{img.height}")        print(f"  URL: {img.image_url}")     print("\n--- 2. 电商主图批量生成 ---")    product_prompts = [        "简约白色陶瓷茶杯，日式风格，柔和自然光",        "高端黑色皮质钱包，商务风格，暗调背景",        "清新绿色植物盆栽，北欧风格，明亮背景",    ]    batch_results = client.batch_generate(        product_prompts, style="realistic"    )    print(f"  批量生成 {len(batch_results)} 张商品主图")    for img in batch_results:        print(f"  [{img.task_id}] {img.image_url}")     print("\n--- 3. 风格迁移 ---")    transfer = client.style_transfer(        content_prompt="城市夜景照片",        style_reference="梵高星空风格"    )    print(f"  风格迁移结果: {transfer.image_url}")    print(f"  应用风格: {transfer.style}")     print("\n--- 4. 生成统计 ---")    stats = client.get_stats()    print(f"  总生成数: {stats['total_generated']}")    print(f"  总任务数: {stats['total_tasks']}")    print(f"  风格分布:")    for style, count in stats["style_distribution"].items():        print(f"    {style}: {count}张")     print("\n演示完成!")  if __name__ == "__main__":    demo_wanx_workflow()