import requests
from flask import jsonify

def generate_image(prompt, API_KEY):
    try:
        url = "https://api.siliconflow.cn/v1/images/generations"
        payload = {
            "model": "black-forest-labs/FLUX.1-schnell",
            "prompt": prompt,
            "negative_prompt": "",
            "image_size": "768x1024",
            "batch_size": 1,
            "num_inference_steps": 20,
            "guidance_scale": 7.5,
            "seed": 42746333
        }
        
        headers = {
            "Authorization": f"Bearer {API_KEY}",
            "Content-Type": "application/json"
        }

        response = requests.post(url, json=payload, headers=headers)
        response_data = response.json()
        
        # 返回图片URL和其他相关信息
        return jsonify({
            'success': True,
            'image_url': response_data['images'][0]['url'],
            'created': response_data['created'],
            'inference_time': response_data['timings']['inference']
        })
        
    except Exception as e:
        return jsonify({
            'success': False,
            'error': str(e)
        }), 500