#!/usr/bin/env python3
"""
测试图片生成功能
"""

import sys
import os
sys.path.insert(0, "/home/ubuntu/wewrite")

try:
    from toolkit.image_gen import generate_image
    print("✅ 成功导入图片生成模块")
    
    # 测试生成一张简单的图片
    prompt = "一只可爱的橘猫在阳光下睡觉，卡通风格"
    output_path = "/tmp/test_cat_simple.jpg"
    
    print(f"🎨 生成图片: {prompt}")
    path = generate_image(prompt, output_path, size="cover")
    print(f"✅ 图片生成成功: {path}")
    
    # 检查文件
    if os.path.exists(path):
        size = os.path.getsize(path) / 1024
        print(f"📊 图片大小: {size:.1f}KB")
        
        # 尝试用Pillow获取图片信息
        try:
            from PIL import Image
            with Image.open(path) as img:
                print(f"📐 图片尺寸: {img.width}x{img.height}")
                print(f"🎨 图片格式: {img.format}")
                print(f"🌈 图片模式: {img.mode}")
        except ImportError:
            print("⚠️  Pillow未安装，无法获取详细图片信息")
    else:
        print("❌ 图片文件不存在")
        
except Exception as e:
    print(f"❌ 测试失败: {e}")
    import traceback
    traceback.print_exc()