package routers

import (
	"main/models"
	"main/utils"
	"net/http"
)

func ParamsListGet(w http.ResponseWriter, r *http.Request) {
	var listview models.ListView

	var item = make(map[string]interface{})
	item["id"] = "model"
	item["name"] = "模型"

	//listview.List = append(listview.List, item)
	listview.List = []interface{}{item}
	listview.Total = 1
	listview.Page = 1
	listview.PageSize = 10
	listview.Next = false

	w.Header().Set("Content-Type", "application/json; charset=utf-8")
	w.Write(utils.ToJSON(listview))
}

type Size struct {
	Width       int    `json:"width"`
	Height      int    `json:"height"`
	AspectRatio string `json:"aspect_ratio"`
}

func ParamsModelsGet(w http.ResponseWriter, r *http.Request) {
	var size_list []Size
	size_list = append(size_list, Size{Width: 512, Height: 512, AspectRatio: "1:1"})
	size_list = append(size_list, Size{Width: 768, Height: 768, AspectRatio: "1:1"})
	size_list = append(size_list, Size{Width: 1024, Height: 768, AspectRatio: "4:3"})
	size_list = append(size_list, Size{Width: 1024, Height: 1024, AspectRatio: "1:1"})
	size_list = append(size_list, Size{Width: 1280, Height: 720, AspectRatio: "16:9"})
	size_list = append(size_list, Size{Width: 1280, Height: 1024, AspectRatio: "5:4"})
	size_list = append(size_list, Size{Width: 1920, Height: 1080, AspectRatio: "16:9"})
	size_list = append(size_list, Size{Width: 1920, Height: 1200, AspectRatio: "16:10"})

	params := make(map[string]interface{})
	params["type"] = []string{"lora", "ckp", "hyper", "ti"}
	params["status"] = []string{"pending", "running", "finished", "failed"}
	params["base_model"] = []string{"SD1.5", "SD2"}
	params["size"] = size_list
	params["sampler_name"] = []string{"Euler a", "Euler", "LMS", "Heun", "DPM2", "DPM2 a", "DPM++ 2S a", "DPM++ 2M", "DPM++ SDE", "DPM++ 2M SDE", "DPM fast", "DDIM"}
	w.Header().Set("Content-Type", "application/json; charset=utf-8")
	w.Write(utils.ToJSON(params))
}