FLUX Pro 专业图像生成完整指南

import requests
import json
import base64
from PIL import Image
from io import BytesIO

# API 配置
API_URL = "https://api.n1n.ai/v1/images/generations"
API_KEY = "your-api-key"

def generate_image(prompt, model="flux-pro", size="1024x1024"):
    """使用 FLUX Pro 生成图像"""
    headers = {
        "Authorization": f"Bearer {API_KEY}",
        "Content-Type": "application/json"
    }
    
    payload = {
        "model": model,
        "prompt": prompt,
        "n": 1,
        "size": size,
        "quality": "hd",
        "response_format": "b64_json"
    }
    
    response = requests.post(API_URL, headers=headers, json=payload)
    
    if response.status_code == 200:
        result = response.json()
        image_data = result["data"][0]["b64_json"]
        
        # 将 base64 转换为图像
        image = Image.open(BytesIO(base64.b64decode(image_data)))
        image.save(f"flux_{size}.png")
        
        print(f"图像已保存")
        return image
    else:
        print(f"错误: {response.status_code}")
        return None

# 使用示例
prompt = "cyberpunk city at night, neon lights, flying cars, ultra detailed, 8k"
image = generate_image(prompt, model="flux-pro")

class FluxProPromptBuilder:
    """FLUX Pro 提示词构建器"""
    
    def __init__(self):
        self.elements = {
            "subject": "",
            "style": "",
            "lighting": "",
            "camera": "",
            "quality": []
        }
    
    def subject(self, description):
        self.elements["subject"] = description
        return self
    
    def style(self, *styles):
        self.elements["style"] = ", ".join(styles)
        return self
    
    def lighting(self, lighting_type):
        self.elements["lighting"] = lighting_type
        return self
    
    def camera(self, angle, focal_length=None):
        camera_str = angle
        if focal_length:
            camera_str += f", {focal_length} lens"
        self.elements["camera"] = camera_str
        return self
    
    def quality_tags(self, *tags):
        self.elements["quality"].extend(tags)
        return self
    
    def build(self):
        prompt_parts = []
        
        if self.elements["subject"]:
            prompt_parts.append(self.elements["subject"])
        
        if self.elements["style"]:
            prompt_parts.append(self.elements["style"])
        
        if self.elements["lighting"]:
            prompt_parts.append(self.elements["lighting"])
        
        if self.elements["camera"]:
            prompt_parts.append(self.elements["camera"])
        
        if self.elements["quality"]:
            prompt_parts.extend(self.elements["quality"])
        
        return ", ".join(prompt_parts)

# 使用示例
builder = FluxProPromptBuilder()
prompt = builder\
    .subject("majestic dragon soaring through clouds")\
    .style("fantasy art", "digital painting")\
    .lighting("dramatic sunset lighting")\
    .camera("wide angle shot", "24mm")\
    .quality_tags("8k resolution", "highly detailed", "masterpiece")\
    .build()

print("生成的提示词:", prompt)

# FLUX Pro 风格控制
class FluxProStyleControl:
    """风格精确控制"""
    
    STYLE_PRESETS = {
        "photorealistic": {
            "prompt_suffix": "photorealistic, ultra realistic, professional photography",
            "guidance_scale": 7.5,
            "steps": 50
        },
        "anime": {
            "prompt_suffix": "anime style, manga art, cel shaded",
            "guidance_scale": 12,
            "steps": 40
        },
        "oil_painting": {
            "prompt_suffix": "oil painting, traditional art, brush strokes",
            "guidance_scale": 8,
            "steps": 45
        }
    }
    
    @classmethod
    def apply_style(cls, prompt, style_name):
        if style_name not in cls.STYLE_PRESETS:
            raise ValueError(f"Unknown style: {style_name}")
        
        style = cls.STYLE_PRESETS[style_name]
        full_prompt = f"{prompt}, {style['prompt_suffix']}"
        
        return {
            "prompt": full_prompt,
            "guidance_scale": style["guidance_scale"],
            "steps": style["steps"]
        }

# 风格混合
def blend_styles(prompt, style1, style2, ratio=0.7):
    """混合两种风格"""
    s1 = FluxProStyleControl.STYLE_PRESETS[style1]
    s2 = FluxProStyleControl.STYLE_PRESETS[style2]
    
    blended_prompt = f"{prompt}, [{s1['prompt_suffix']}:{ratio}], [{s2['prompt_suffix']}:{1-ratio}]"
    
    return {
        "prompt": blended_prompt,
        "guidance_scale": s1["guidance_scale"] * ratio + s2["guidance_scale"] * (1-ratio),
        "steps": int(s1["steps"] * ratio + s2["steps"] * (1-ratio))
    }

import asyncio
import aiohttp

class FluxProBatchGenerator:
    """批量生成优化"""
    
    def __init__(self, api_key, max_concurrent=3):
        self.api_key = api_key
        self.api_url = "https://api.n1n.ai/v1/images/generations"
        self.max_concurrent = max_concurrent
    
    async def generate_single(self, session, prompt, **kwargs):
        headers = {
            "Authorization": f"Bearer {self.api_key}",
            "Content-Type": "application/json"
        }
        
        payload = {
            "model": "flux-pro",
            "prompt": prompt,
            "n": 1,
            **kwargs
        }
        
        async with session.post(self.api_url, headers=headers, json=payload) as response:
            if response.status == 200:
                return await response.json()
            else:
                return {"error": f"Status {response.status}"}
    
    async def generate_batch(self, prompts_with_params):
        async with aiohttp.ClientSession() as session:
            semaphore = asyncio.Semaphore(self.max_concurrent)
            
            async def generate_with_limit(prompt_params):
                async with semaphore:
                    prompt = prompt_params["prompt"]
                    params = prompt_params.get("params", {})
                    return await self.generate_single(session, prompt, **params)
            
            tasks = [generate_with_limit(pp) for pp in prompts_with_params]
            return await asyncio.gather(*tasks)