Ubuntu
42 lines
1.3 KiB
Python
42 lines
1.3 KiB
Python
#!/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() |