OmniGen2 API Integration Guide

pip install omnigen2

# 导入OmniGen2 SDK
import omnigen2
from omnigen2 import OmniGen2Client

# 初始化客户端
client = OmniGen2Client(
    api_key="your_api_key_here",
    base_url="https://api.omnigen2.pro/v1"
)

# 生成图像
response = client.text_to_image(
    prompt="A futuristic cityscape with flying cars",
    width=1024,
    height=1024,
    style="photorealistic",
    quality="high"
)

# 保存图像
with open("generated_image.png", "wb") as f:
    f.write(response.image_data)

print(f"Image generated successfully: {response.image_url}")

npm install omnigen2-js

// 导入OmniGen2 JavaScript SDK
import { OmniGen2Client } from 'omnigen2-js';

// 初始化客户端
const client = new OmniGen2Client({
    apiKey: 'your_api_key_here',
    baseUrl: 'https://api.omnigen2.pro/v1'
});

// 生成图像
async function generateImage() {
    try {
        const response = await client.textToImage({
            prompt: 'A beautiful landscape with mountains and a lake',
            width: 1024,
            height: 1024,
            style: 'photorealistic',
            quality: 'high'
        });
        
        console.log('Image generated:', response.imageUrl);
        return response;
    } catch (error) {
        console.error('Error generating image:', error);
    }
}

// 调用函数
generateImage();

// React组件示例
import React, { useState } from 'react';
import { OmniGen2Client } from 'omnigen2-js';

const ImageGenerator = () => {
    const [prompt, setPrompt] = useState('');
    const [imageUrl, setImageUrl] = useState('');
    const [loading, setLoading] = useState(false);
    
    const client = new OmniGen2Client({
        apiKey: process.env.REACT_APP_OMNIGEN2_API_KEY
    });
    
    const handleGenerate = async () => {
        setLoading(true);
        try {
            const response = await client.textToImage({
                prompt: prompt,
                width: 512,
                height: 512
            });
            setImageUrl(response.imageUrl);
        } catch (error) {
            console.error('Generation failed:', error);
        } finally {
            setLoading(false);
        }
    };
    
    return (
        <div className="image-generator">
            <input
                type="text"
                value={prompt}
                onChange={(e) => setPrompt(e.target.value)}
                placeholder="Enter your image description..."
            />
            <button onClick={handleGenerate} disabled={loading}>
                {loading ? 'Generating...' : 'Generate Image'}
            </button>
            {imageUrl && <img src={imageUrl} alt="Generated" />}
        </div>
    );
};

export default ImageGenerator;

# Python错误处理示例
import time
from omnigen2 import OmniGen2Client, OmniGen2Error

def generate_with_retry(client, prompt, max_retries=3):
    for attempt in range(max_retries):
        try:
            response = client.text_to_image(prompt=prompt)
            return response
        except OmniGen2Error as e:
            if e.status_code == 429:  # 速率限制
                wait_time = 2 ** attempt  # 指数退避
                print(f"Rate limited, waiting {wait_time}s...")
                time.sleep(wait_time)
            elif e.status_code == 500:  # 服务器错误
                print(f"Server error, retrying in {attempt + 1}s...")
                time.sleep(attempt + 1)
            else:
                raise e  # 其他错误直接抛出
    
    raise Exception("Max retries exceeded")

# .env文件示例
OMNIGEN2_API_KEY=your_production_api_key
OMNIGEN2_BASE_URL=https://api.omnigen2.pro/v1
REDIS_URL=redis://localhost:6379
DATABASE_URL=postgresql://user:pass@localhost/db

# Dockerfile示例
FROM node:18-alpine

WORKDIR /app

# 复制package.json和安装依赖
COPY package*.json ./
RUN npm ci --only=production

# 复制应用代码
COPY . .

# 设置环境变量
ENV NODE_ENV=production

# 暴露端口
EXPOSE 3000

# 启动应用
CMD ["npm", "start"]