SeeDream 图片生成/编辑

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

"""
Seedream API 图片生成器 - 纯Python版本
修改 API_KEY 即可使用
"""

# =================== 配置区域 ===================
API_KEY = "sk-"                                              # 替换为您的API密钥
API_URL = "https://api.yelinai.com/v1/images/generations"    # API地址
PROMPT = "A beautiful sunset over mountains, realistic style"  # 图片描述
MODEL = "seedream-4-0-250828"                                # 模型名称
OUTPUT_DIR = "."                                             # 输出目录
# ==============================================

import requests
import os
import datetime

def generate_image(prompt, output_dir="."):
    """生成图片并保存到本地"""
    print(f"🎨 Seedream 图片生成器")
    print(f"📝 提示词: {prompt}")
    print("=" * 50)
    
    # API 请求参数
    headers = {
        "Content-Type": "application/json",
        "Authorization": f"Bearer {API_KEY}"
    }
    
    payload = {
        "model": MODEL,
        "prompt": prompt,
        "response_format": "url",
        "size": "2K",
        "watermark": False
    }
    
    try:
        print("⏳ 正在生成图片...")
        
        # 调用 API
        response = requests.post(API_URL, headers=headers, json=payload, timeout=60)
        
        if response.status_code != 200:
            return False, f"API错误 ({response.status_code}): {response.text}"
        
        # 解析响应
        result = response.json()
        if "data" not in result or len(result["data"]) == 0:
            return False, "API未返回图片数据"
        
        # 获取图片URL
        image_url = result["data"][0]["url"]
        print(f"🌐 获取图片URL成功")
        
        # 下载图片
        print("⬇️ 正在下载图片...")
        img_response = requests.get(image_url, timeout=30)
        img_response.raise_for_status()
        
        # 生成文件名（带时间戳）
        timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
        filename = os.path.join(output_dir, f"generated_{timestamp}.jpg")
        
        # 保存图片
        os.makedirs(output_dir, exist_ok=True)
        with open(filename, 'wb') as f:
            f.write(img_response.content)
        
        file_size = len(img_response.content)
        return True, f"✅ 生成成功: {filename} ({file_size // 1024}KB)"
        
    except Exception as e:
        return False, f"生成失败: {str(e)}"

def main():
    """主函数"""
    print("🚀 Seedream API 图片生成器 - Python版本")
    print("=" * 50)
    
    # 检查 API 密钥
    if API_KEY == "sk-" or not API_KEY:
        print("⚠️  请先修改代码顶部的 API_KEY")
        print("   将 'sk-' 替换为您的真实API密钥")
        print()
        print("📋 使用说明:")
        print("1. 修改 API_KEY 为您的真实密钥")
        print("2. 可选：修改 PROMPT 生成不同内容的图片")
        print("3. 运行: python3 seedream-text-to-image.py")
        return
    
    # 执行图片生成
    success, message = generate_image(PROMPT, OUTPUT_DIR)
    print(message)
    
    if success:
        print()
        print("🎉 任务完成！")
        print("💡 提示：修改 PROMPT 可以生成不同内容的图片")

if __name__ == "__main__":
    main()

curl -X POST https://api.yelinai.com/v1/images/generations \
  -H "Content-Type: application/json" \
  -H "Authorization: Bearer sk-YOUR_API_KEY" \
  -d '{
    "model": "seedream-4-0-250828",
    "prompt": "A beautiful sunset over mountains, realistic style",
    "response_format": "url",
    "size": "2K",
    "watermark": false
}'

curl -X POST https://api.yelinai.com/v1/images/generations \
  -H "Content-Type: application/json" \
  -H "Authorization: Bearer sk-YOUR_API_KEY" \
  -d '{"model": "seedream-4-0-250828","prompt": "A beautiful sunset over mountains, realistic style","response_format": "url","size": "2K","watermark": false}' \
  | python3 -c "import json,sys,os; data=json.loads(sys.stdin.read()); os.system(f'curl -L -o generated_\$(date +%Y%m%d_%H%M%S).jpg \"{data[\"data\"][0][\"url\"]}\"') if 'data' in data else print('Error')"

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

"""
Seedream API 图片编辑器 - Python版本
基于原图生成新图片，修改 API_KEY 即可使用
"""

# =================== 配置区域 ===================
API_KEY = "sk-"                                              # 替换为您的API密钥
API_URL = "https://api.yelinai.com/v1/images/generations"    # 图片编辑API地址
PROMPT = "Generate a close-up image of a dog lying on lush grass."  # 编辑提示词
IMAGE_URL = "https://ark-doc.tos-ap-southeast-1.bytepluses.com/doc_image/seedream4_imageToimage.png"  # 原图URL
MODEL = "seedream-4-0-250828"                                # 使用的模型
OUTPUT_DIR = "."                                             # 输出目录
# ==============================================

import requests
import os
import datetime

def edit_image(prompt, image_url, output_dir="."):
    """基于原图生成编辑后的图片"""
    print(f"🎨 Seedream 图片编辑器")
    print(f"📝 编辑提示: {prompt}")
    print(f"🖼️  原图链接: {image_url[:50]}...")
    print("=" * 50)
    
    # API 请求参数
    headers = {
        "Content-Type": "application/json",
        "Authorization": f"Bearer {API_KEY}"
    }
    
    payload = {
        "model": MODEL,
        "prompt": prompt,
        "image": image_url,
        "sequential_image_generation": "disabled",
        "response_format": "url",
        "size": "2K",
        "stream": False,
        "watermark": False
    }
    
    try:
        print("⏳ 正在编辑图片...")
        
        # 调用 API
        response = requests.post(API_URL, headers=headers, json=payload, timeout=60)
        
        if response.status_code != 200:
            return False, f"API错误 ({response.status_code}): {response.text}"
        
        # 解析响应
        result = response.json()
        if "data" not in result or len(result["data"]) == 0:
            return False, "API未返回图片数据"
        
        # 获取图片URL
        edited_image_url = result["data"][0]["url"]
        print(f"🌐 获取编辑图片URL成功")
        
        # 下载图片
        print("⬇️ 正在下载编辑后的图片...")
        img_response = requests.get(edited_image_url, timeout=30)
        img_response.raise_for_status()
        
        # 生成文件名（带时间戳）
        timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
        filename = os.path.join(output_dir, f"edited_{timestamp}.jpg")
        
        # 保存图片
        os.makedirs(output_dir, exist_ok=True)
        with open(filename, 'wb') as f:
            f.write(img_response.content)
        
        file_size = len(img_response.content)
        return True, f"✅ 编辑成功: {filename} ({file_size // 1024}KB)"
        
    except Exception as e:
        return False, f"编辑失败: {str(e)}"

def main():
    """主函数"""
    print("🚀 Seedream API 图片编辑器 - Python版本")
    print("=" * 50)
    
    # 检查 API 密钥
    if API_KEY == "sk-" or not API_KEY:
        print("⚠️  请先修改代码顶部的 API_KEY")
        print("   将 'sk-' 替换为您的真实API密钥")
        print()
        print("📋 使用说明:")
        print("1. 修改 API_KEY 为您的真实密钥")
        print("2. 修改 IMAGE_URL 为您要编辑的图片链接")
        print("3. 修改 PROMPT 描述您想要的编辑效果")
        print("4. 运行: python3 seedream-image-edit.py")
        return
    
    # 执行图片编辑
    success, message = edit_image(PROMPT, IMAGE_URL, OUTPUT_DIR)
    print(message)
    
    if success:
        print()
        print("🎉 编辑完成！")
        print("💡 提示：修改 PROMPT 和 IMAGE_URL 可以编辑不同的图片")

if __name__ == "__main__":
    main()

#!/bin/bash

# Seedream API 图片编辑器 - Curl版本
# 基于原图生成新图片，修改 API_KEY 即可使用

# =============== 配置区域 ===============
API_KEY="sk-YOUR_API_KEY"                                    # 请替换为您的API密钥
API_URL="https://api.yelinai.com/v1/images/generations"      # 图片编辑API地址
PROMPT="Generate a close-up image of a dog lying on lush grass."  # 编辑提示词
IMAGE_URL="https://ark-doc.tos-ap-southeast-1.bytepluses.com/doc_image/seedream4_imageToimage.png"  # 原图URL
MODEL="seedream-4-0-250828"                                  # 模型名称
# ====================================

# 检查API密钥
if [ "$API_KEY" = "sk-YOUR_API_KEY" ]; then
    echo "⚠️  请先修改脚本中的 API_KEY"
    echo "   将 'sk-YOUR_API_KEY' 替换为您的真实API密钥"
    echo ""
    echo "📋 使用说明:"
    echo "1. 修改 API_KEY 为您的真实密钥"
    echo "2. 修改 IMAGE_URL 为您要编辑的图片链接"
    echo "3. 修改 PROMPT 描述您想要的编辑效果"
    echo "4. 运行: ./seedream-image-edit.sh"
    exit 1
fi

# 生成输出文件名
TIMESTAMP=$(date +"%Y%m%d_%H%M%S")
OUTPUT_FILE="edited_\`$\{TIMESTAMP}.jpg"

echo "🎨 Seedream API 图片编辑器 - Curl版本"
echo "========================================"
echo "📝 编辑提示: $PROMPT"
echo "🖼️  原图链接: \`$\{IMAGE_URL:0:50}..."
echo "📁 输出文件: $OUTPUT_FILE"
echo "========================================"

# 步骤1: 调用API获取JSON响应
echo "⏳ 正在编辑图片..."

# 构建JSON请求体
JSON_DATA=$(cat <<EOF
{
  "model": "$MODEL",
  "prompt": "$PROMPT",
  "image": "$IMAGE_URL",
  "sequential_image_generation": "disabled",
  "response_format": "url",
  "size": "2K",
  "stream": false,
  "watermark": false
}
EOF
)

RESPONSE=$(curl -s -X POST "$API_URL" \
  -H "Content-Type: application/json" \
  -H "Authorization: Bearer $API_KEY" \
  -d "$JSON_DATA")

# 检查API响应
if [ -z "$RESPONSE" ]; then
    echo "❌ API无响应，请检查网络连接"
    exit 1
fi

echo "✅ API响应成功"

# 步骤2: 使用Python解析JSON并提取图片URL
echo "🔍 解析编辑后的图片URL..."
EDITED_IMAGE_URL=$(echo "$RESPONSE" | python3 -c "
import json
import sys

try:
    data = json.loads(sys.stdin.read())
    
    if 'data' in data and len(data['data']) > 0:
        url = data['data'][0]['url']
        print(url)
    else:
        print('ERROR: 未找到图片URL')
        sys.exit(1)
        
except Exception as e:
    print(f'ERROR: {e}')
    sys.exit(1)
")

# 检查URL解析结果
if [[ "$EDITED_IMAGE_URL" == ERROR* ]]; then
    echo "❌ URL解析失败: $EDITED_IMAGE_URL"
    echo "📄 原始API响应:"
    echo "$RESPONSE"
    exit 1
fi

echo "🌐 编辑图片URL获取成功"

# 步骤3: 下载编辑后的图片
echo "⬇️ 正在下载编辑后的图片..."
curl -s -L -o "$OUTPUT_FILE" "$EDITED_IMAGE_URL"

# 检查下载结果
if [ -f "$OUTPUT_FILE" ] && [ -s "$OUTPUT_FILE" ]; then
    FILE_SIZE=$(du -h "$OUTPUT_FILE" | cut -f1)
    echo "✅ 图片编辑成功!"
    echo "📁 文件: $OUTPUT_FILE"
    echo "📊 大小: $FILE_SIZE"
    echo ""
    echo "🎉 完成! 请查看编辑后的图片文件"
    echo "💡 提示: 修改 PROMPT 和 IMAGE_URL 可以编辑不同的图片"
else
    echo "❌ 图片下载失败"
    echo "🔗 您可以手动访问: $EDITED_IMAGE_URL"
fi

{
  "created": 1726051200,
  "data": [
    {
      "url": "https://example.com/generated_image.jpg",
      "revised_prompt": "..."
    }
  ]
}