From 9870c903ca1b4c72f7cbb162c04f9a0b4e579380 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=A1=9C=E8=8F=AF?= <huan0016@gmail.com>
Date: Sat, 13 May 2023 04:39:47 +0800
Subject: [PATCH] model

---
 models/Model.go | 26 +++++++++++++-------------
 1 file changed, 13 insertions(+), 13 deletions(-)

diff --git a/models/Model.go b/models/Model.go
index fa5e07d..0aa8bb0 100644
--- a/models/Model.go
+++ b/models/Model.go
@@ -8,13 +8,13 @@ import (
 type Model struct {
 	ID           int    `json:"id"`
 	Name         string `json:"name"`
-	Type         string `json:"type"`          // (lora|ckp|hyper|ti)
-	TriggerWords string `json:"trigger_words"` // 觸發詞
-	BaseModel    string `json:"base_model"`    // (SD1.5|SD2)
-	ModelPath    string `json:"model_path"`    // 模型路徑
-	Status       string `json:"status"`        // (initial|ready|waiting|running|success|error)
-	Progress     int    `json:"progress"`      // (0-100)
-	Image        string `json:"image"`         // 封面圖片實際地址
+	Type         string `json:"type"`                     // (lora|ckp|hyper|ti)
+	TriggerWords string `json:"trigger_words"`            // 觸發詞
+	BaseModel    string `json:"base_model"`               // (SD1.5|SD2)
+	ModelPath    string `json:"model_path"`               // 模型路徑
+	Status       string `json:"status" default:"initial"` // (initial|ready|waiting|running|success|error)
+	Progress     int    `json:"progress"`                 // (0-100)
+	Image        string `json:"image"`                    // 封面圖片實際地址
 	Tags         string `json:"tags"`
 	CreatedAt    string `json:"created_at"`
 	UpdatedAt    string `json:"updated_at"`
@@ -51,13 +51,13 @@ func (model *Model) Create() error {
 		return err
 	}
 	defer db.Close()
-	stmt, err := db.Prepare("INSERT INTO models(name, type, trigger_words, base_model, model_path, status, progress, tags, created_at, updated_at, user_id) values(?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)")
+	stmt, err := db.Prepare("INSERT INTO models(name, type, trigger_words, base_model, model_path, status, progress, image, tags, created_at, updated_at, user_id) values(?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)")
 	if err != nil {
 		log.Println(err)
 		return err
 	}
 	defer stmt.Close()
-	result, err := stmt.Exec(model.Name, model.Type, model.TriggerWords, model.BaseModel, model.ModelPath, model.Status, model.Progress, model.Tags, model.CreatedAt, model.UpdatedAt, model.UserID)
+	result, err := stmt.Exec(model.Name, model.Type, model.TriggerWords, model.BaseModel, model.ModelPath, model.Status, model.Progress, model.Image, model.Tags, model.CreatedAt, model.UpdatedAt, model.UserID)
 	if err != nil {
 		log.Println(err)
 		return err
@@ -119,7 +119,7 @@ func (model *Model) Get() error {
 		return err
 	}
 	defer db.Close()
-	err = db.QueryRow("SELECT * FROM models WHERE id = ?", model.ID).Scan(&model.ID, &model.Name, &model.Type, &model.CreatedAt, &model.UpdatedAt)
+	err = db.QueryRow("SELECT id,name,type,trigger_words,base_model,model_path,status,progress,image,tags,creaated_at,updated_at,user_id FROM models WHERE id = ?", model.ID).Scan(&model.ID, &model.Name, &model.Type, &model.TriggerWords, &model.BaseModel, &model.ModelPath, &model.Status, &model.Progress, &model.Image, &model.Tags, &model.CreatedAt, &model.UpdatedAt, &model.UserID)
 	if err != nil {
 		log.Println(err)
 		return err
@@ -135,7 +135,7 @@ func QueryModel(id int) (model Model) {
 		return
 	}
 	defer db.Close()
-	err = db.QueryRow("SELECT id, name, type, trigger_words, base_model, model_path, status, progress, tags, created_at, updated_at, user_id FROM models WHERE id = ?", id).Scan(&model.ID, &model.Name, &model.Type, &model.TriggerWords, &model.BaseModel, &model.ModelPath, &model.Status, &model.Progress, &model.Tags, &model.CreatedAt, &model.UpdatedAt, &model.UserID)
+	err = db.QueryRow("SELECT id,name,type,trigger_words,base_model,model_path,status,progress,image,tags,created_at,updated_at,user_id FROM models WHERE id = ?", id).Scan(&model.ID, &model.Name, &model.Type, &model.TriggerWords, &model.BaseModel, &model.ModelPath, &model.Status, &model.Progress, &model.Image, &model.Tags, &model.CreatedAt, &model.UpdatedAt, &model.UserID)
 	if err != nil {
 		log.Println(err)
 		return
@@ -150,7 +150,7 @@ func QueryModels(page int, pagesize int) (models []interface{}) {
 		return
 	}
 	defer db.Close()
-	rows, err := db.Query("SELECT id, name, type, trigger_words, base_model, model_path, status, progress, tags, created_at, updated_at, user_id FROM models LIMIT ?, ?", (page-1)*pagesize, pagesize)
+	rows, err := db.Query("SELECT id, name, type, trigger_words, base_model, model_path, status, progress, image, tags, created_at, updated_at, user_id FROM models LIMIT ?, ?", (page-1)*pagesize, pagesize)
 	if err != nil {
 		log.Println(err)
 		return
@@ -158,7 +158,7 @@ func QueryModels(page int, pagesize int) (models []interface{}) {
 	defer rows.Close()
 	for rows.Next() {
 		var model Model
-		err = rows.Scan(&model.ID, &model.Name, &model.Type, &model.TriggerWords, &model.BaseModel, &model.ModelPath, &model.Status, &model.Progress, &model.Tags, &model.CreatedAt, &model.UpdatedAt, &model.UserID)
+		err = rows.Scan(&model.ID, &model.Name, &model.Type, &model.TriggerWords, &model.BaseModel, &model.ModelPath, &model.Status, &model.Progress, &model.Image, &model.Tags, &model.CreatedAt, &model.UpdatedAt, &model.UserID)
 		if err != nil {
 			log.Println(err)
 			return