package einograph

import (
	"bytes"
	"context"
	"encoding/json"
	"fmt"
	"io"
	"net/http"
	"time"

	"github.com/cloudwego/eino-ext/components/model/qwen"
	"github.com/cloudwego/eino/components/model"
	"github.com/gogf/gf/v2/util/gconv"
)

// NewChatModel component initialization function of node 'ChatModel1' in graph 'test'
func NewChatModel(ctx context.Context) (cm model.ChatModel, err error) {
	// TODO Modify component configuration here.
	config := &qwen.ChatModelConfig{
		APIKey:      "sk-4a8b82770bf74bc490eb3e4c5a8e2be9",
		Model:       "qwen-turbo",
		BaseURL:     "https://dashscope.aliyuncs.com/compatible-mode/v1", // 必须加！
		MaxTokens:   gconv.PtrInt(2000),
		Temperature: gconv.PtrFloat32(float32(0.7))}
	cm, err = qwen.NewChatModel(ctx, config)
	if err != nil {
		return nil, err
	}
	return cm, nil
}

func GenerateRealImage(prompt string) (string, error) {
	apiKey := "sk-4a8b82770bf74bc490eb3e4c5a8e2be9"
	url := "https://dashscope.aliyuncs.com/api/v1/services/aigc/multimodal-generation/generation"

	body := map[string]any{
		"model": "qwen-image",
		"input": map[string]any{
			"messages": []map[string]any{
				{
					"role": "user",
					"content": []map[string]string{
						{"type": "text", "text": prompt},
					},
				},
			},
		},
		"parameters": map[string]any{
			"size":      "1024*1364",
			"n":         1,
			"watermark": false,
		},
	}

	payload, _ := json.Marshal(body)
	req, _ := http.NewRequest("POST", url, bytes.NewBuffer(payload))
	req.Header.Set("Authorization", "Bearer "+apiKey)
	req.Header.Set("Content-Type", "application/json")

	client := &http.Client{Timeout: 30 * time.Second}
	resp, err := client.Do(req)
	if err != nil {
		return "", err
	}
	defer resp.Body.Close()

	data, _ := io.ReadAll(resp.Body)

	// ✅ 修复：plus 模型返回的字段是 image 不是 image_url
	var result struct {
		Output struct {
			Choices []struct {
				Message struct {
					Content []struct {
						Image string `json:"image"`
					} `json:"content"`
				} `json:"message"`
			} `json:"choices"`
		} `json:"output"`
		Code string `json:"code"`
	}

	err = gconv.Struct(data, &result)

	if len(result.Output.Choices) == 0 || result.Code != "" {
		return "", fmt.Errorf("生成失败: %s", string(data))
	}

	// ✅ 修复：直接取 base64，不再下载
	imgBase64 := result.Output.Choices[0].Message.Content[0].Image
	if imgBase64 == "" {
		return "", fmt.Errorf("图片为空")
	}

	// 解码 base64
	return imgBase64, nil
}