chore: more work on moving to svelte front end
This commit is contained in:
parent
ce866725ff
commit
e990b832d3
6
DockerfileDev
Normal file
6
DockerfileDev
Normal file
@ -0,0 +1,6 @@
|
|||||||
|
# vi: ft=dockerfile
|
||||||
|
FROM docker.io/nginx
|
||||||
|
|
||||||
|
ADD nginx.dev.conf /nginx.conf
|
||||||
|
|
||||||
|
CMD ["nginx", "-c", "/nginx.conf", "-g", "daemon off;"]
|
@ -91,9 +91,79 @@ func handleAdd(handle *Handle) {
|
|||||||
if !CheckAuthLevel(1, w, r, c) {
|
if !CheckAuthLevel(1, w, r, c) {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
if c.Mode == JSON {
|
if c.Mode == JSON {
|
||||||
// TODO json
|
read_form, err := r.MultipartReader()
|
||||||
panic("TODO JSON")
|
if err != nil {
|
||||||
|
return c.JsonErrorBadRequest(err, "Please provide a valid multipart Reader request")
|
||||||
|
}
|
||||||
|
|
||||||
|
var name string
|
||||||
|
var file []byte
|
||||||
|
|
||||||
|
for {
|
||||||
|
part, err_part := read_form.NextPart()
|
||||||
|
if err_part == io.EOF {
|
||||||
|
break
|
||||||
|
} else if err_part != nil {
|
||||||
|
return &Error{Code: http.StatusBadRequest}
|
||||||
|
}
|
||||||
|
if part.FormName() == "name" {
|
||||||
|
buf := new(bytes.Buffer)
|
||||||
|
buf.ReadFrom(part)
|
||||||
|
name = buf.String()
|
||||||
|
}
|
||||||
|
if part.FormName() == "file" {
|
||||||
|
buf := new(bytes.Buffer)
|
||||||
|
buf.ReadFrom(part)
|
||||||
|
file = buf.Bytes()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if name == "" || len(file) == 0 {
|
||||||
|
return c.JsonBadRequest("Name is empty or file is empty")
|
||||||
|
}
|
||||||
|
|
||||||
|
row, err := handle.Db.Query("select id from models where name=$1 and user_id=$2;", name, c.User.Id)
|
||||||
|
if err != nil {
|
||||||
|
return c.Error500(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if row.Next() {
|
||||||
|
return c.JsonBadRequest("Model with that name already exists!")
|
||||||
|
}
|
||||||
|
|
||||||
|
row, err = handle.Db.Query("insert into models (user_id, name) values ($1, $2) returning id", c.User.Id, name)
|
||||||
|
if err != nil || !row.Next() {
|
||||||
|
return c.Error500(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
var id string
|
||||||
|
err = row.Scan(&id)
|
||||||
|
if err != nil {
|
||||||
|
return c.Error500(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// TODO mk this path configurable
|
||||||
|
dir_path := path.Join("savedData", id)
|
||||||
|
|
||||||
|
err = os.Mkdir(dir_path, os.ModePerm)
|
||||||
|
if err != nil {
|
||||||
|
return c.Error500(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
f, err := os.Create(path.Join(dir_path, "baseimage.png"))
|
||||||
|
if err != nil {
|
||||||
|
return c.Error500(err)
|
||||||
|
}
|
||||||
|
defer f.Close()
|
||||||
|
|
||||||
|
f.Write(file)
|
||||||
|
|
||||||
|
c.Logger.Warn("Created model with id %s! Started to proccess image!\n", "id", id)
|
||||||
|
go loadBaseImage(c, id)
|
||||||
|
|
||||||
|
return c.SendJSON(id)
|
||||||
}
|
}
|
||||||
|
|
||||||
read_form, err := r.MultipartReader()
|
read_form, err := r.MultipartReader()
|
||||||
|
@ -4,89 +4,145 @@ import (
|
|||||||
"net/http"
|
"net/http"
|
||||||
"strconv"
|
"strconv"
|
||||||
|
|
||||||
. "git.andr3h3nriqu3s.com/andr3/fyp/logic/utils"
|
|
||||||
. "git.andr3h3nriqu3s.com/andr3/fyp/logic/models/utils"
|
. "git.andr3h3nriqu3s.com/andr3/fyp/logic/models/utils"
|
||||||
|
"git.andr3h3nriqu3s.com/andr3/fyp/logic/utils"
|
||||||
|
. "git.andr3h3nriqu3s.com/andr3/fyp/logic/utils"
|
||||||
)
|
)
|
||||||
|
|
||||||
func HandleList(handle *Handle) {
|
func models_data_list_json(w http.ResponseWriter, r *http.Request, c *Context) *Error {
|
||||||
handle.Get("/models/data/list", func(w http.ResponseWriter, r *http.Request, c *Context) *Error {
|
id, err := GetIdFromUrl(r, "id")
|
||||||
if !CheckAuthLevel(1, w, r, c) {
|
if err != nil {
|
||||||
return nil
|
return c.JsonBadRequest("Model Class not found!")
|
||||||
}
|
}
|
||||||
if c.Mode == JSON {
|
|
||||||
panic("TODO JSON on /models/data/list")
|
|
||||||
}
|
|
||||||
|
|
||||||
id, err := GetIdFromUrl(r, "id")
|
page := 0
|
||||||
if err != nil {
|
if r.URL.Query().Has("page") {
|
||||||
return ErrorCode(err, 400, c.AddMap(nil))
|
page_url := r.URL.Query().Get("page")
|
||||||
}
|
page_url_number, err := strconv.Atoi(page_url)
|
||||||
|
if err != nil {
|
||||||
|
return c.JsonBadRequest("Page is not a number")
|
||||||
|
}
|
||||||
|
page = page_url_number
|
||||||
|
}
|
||||||
|
|
||||||
page := 0
|
var class_row struct {
|
||||||
if r.URL.Query().Has("page") {
|
Name string
|
||||||
page_url := r.URL.Query().Get("page")
|
Model_id string
|
||||||
page_url_number, err := strconv.Atoi(page_url)
|
}
|
||||||
if err != nil {
|
|
||||||
return ErrorCode(err, http.StatusBadRequest, c.AddMap(nil))
|
|
||||||
}
|
|
||||||
page = page_url_number
|
|
||||||
}
|
|
||||||
|
|
||||||
class_rows, err := handle.Db.Query("select name, model_id from model_classes where id=$1;", id)
|
err = utils.GetDBOnce(c, &class_row, "model_classes where id=$1", id)
|
||||||
if err != nil {
|
if err == NotFoundError {
|
||||||
return Error500(err)
|
return c.JsonBadRequest("Model Class not found!")
|
||||||
}
|
} else if err != nil {
|
||||||
defer class_rows.Close()
|
return c.Error500(err)
|
||||||
|
}
|
||||||
|
|
||||||
if !class_rows.Next() {
|
type baserow struct {
|
||||||
return ErrorCode(nil, 404, c.AddMap(nil))
|
Id string
|
||||||
}
|
File_Path string
|
||||||
|
Model_Mode int
|
||||||
|
Status int
|
||||||
|
}
|
||||||
|
|
||||||
name := ""
|
rows, err := utils.GetDbMultitple[baserow](c, "model_data_point where class_id=$1 limit 11 offset $2", id, page*10)
|
||||||
model_id := ""
|
if err != nil {
|
||||||
if err = class_rows.Scan(&name, &model_id); err != nil {
|
return c.Error500(err)
|
||||||
return Error500(nil)
|
}
|
||||||
}
|
|
||||||
|
|
||||||
model, err := GetBaseModel(c.Db, model_id)
|
type ReturnType struct {
|
||||||
if err != nil {
|
ImageList []*baserow `json:"image_list"`
|
||||||
return Error500(err)
|
Page int `json:"page"`
|
||||||
}
|
ShowNext bool `json:"showNext"`
|
||||||
|
}
|
||||||
|
|
||||||
rows, err := handle.Db.Query("select id, file_path, model_mode, status from model_data_point where class_id=$1 limit 11 offset $2;", id, page * 10)
|
max_len := min(11, len(rows))
|
||||||
if err != nil {
|
|
||||||
return Error500(err)
|
|
||||||
}
|
|
||||||
defer rows.Close()
|
|
||||||
|
|
||||||
type baserow struct {
|
return c.SendJSON(ReturnType{
|
||||||
Id string
|
ImageList: rows[0:max_len],
|
||||||
FilePath string
|
Page: page,
|
||||||
Mode int
|
ShowNext: len(rows) == 11,
|
||||||
Status int
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
got := []baserow{}
|
func HandleList(handle *Handle) {
|
||||||
|
handle.Get("/models/data/list", func(w http.ResponseWriter, r *http.Request, c *Context) *Error {
|
||||||
for rows.Next() {
|
if !CheckAuthLevel(1, w, r, c) {
|
||||||
nrow := baserow{}
|
return nil
|
||||||
err = rows.Scan(&nrow.Id, &nrow.FilePath, &nrow.Mode, &nrow.Status)
|
}
|
||||||
if err != nil {
|
if c.Mode == JSON {
|
||||||
return Error500(err)
|
return models_data_list_json(w, r, c)
|
||||||
}
|
}
|
||||||
got = append(got, nrow)
|
|
||||||
}
|
id, err := GetIdFromUrl(r, "id")
|
||||||
|
if err != nil {
|
||||||
max_len := min(11, len(got))
|
return ErrorCode(err, 400, c.AddMap(nil))
|
||||||
|
}
|
||||||
LoadDefineTemplate(w, "/models/edit.html", "data-model-create-class-table-table", c.AddMap(AnyMap{
|
|
||||||
"List": got[0:max_len],
|
page := 0
|
||||||
"Page": page,
|
if r.URL.Query().Has("page") {
|
||||||
"Id": id,
|
page_url := r.URL.Query().Get("page")
|
||||||
"Name": name,
|
page_url_number, err := strconv.Atoi(page_url)
|
||||||
"Model": model,
|
if err != nil {
|
||||||
"ShowNext": len(got) == 11,
|
return ErrorCode(err, http.StatusBadRequest, c.AddMap(nil))
|
||||||
}))
|
}
|
||||||
return nil
|
page = page_url_number
|
||||||
})
|
}
|
||||||
|
|
||||||
|
class_rows, err := handle.Db.Query("select name, model_id from model_classes where id=$1;", id)
|
||||||
|
if err != nil {
|
||||||
|
return Error500(err)
|
||||||
|
}
|
||||||
|
defer class_rows.Close()
|
||||||
|
|
||||||
|
if !class_rows.Next() {
|
||||||
|
return ErrorCode(nil, 404, c.AddMap(nil))
|
||||||
|
}
|
||||||
|
|
||||||
|
name := ""
|
||||||
|
model_id := ""
|
||||||
|
if err = class_rows.Scan(&name, &model_id); err != nil {
|
||||||
|
return Error500(nil)
|
||||||
|
}
|
||||||
|
|
||||||
|
model, err := GetBaseModel(c.Db, model_id)
|
||||||
|
if err != nil {
|
||||||
|
return Error500(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
rows, err := handle.Db.Query("select id, file_path, model_mode, status from model_data_point where class_id=$1 limit 11 offset $2;", id, page*10)
|
||||||
|
if err != nil {
|
||||||
|
return Error500(err)
|
||||||
|
}
|
||||||
|
defer rows.Close()
|
||||||
|
|
||||||
|
type baserow struct {
|
||||||
|
Id string
|
||||||
|
FilePath string
|
||||||
|
Mode int
|
||||||
|
Status int
|
||||||
|
}
|
||||||
|
|
||||||
|
got := []baserow{}
|
||||||
|
|
||||||
|
for rows.Next() {
|
||||||
|
nrow := baserow{}
|
||||||
|
err = rows.Scan(&nrow.Id, &nrow.FilePath, &nrow.Mode, &nrow.Status)
|
||||||
|
if err != nil {
|
||||||
|
return Error500(err)
|
||||||
|
}
|
||||||
|
got = append(got, nrow)
|
||||||
|
}
|
||||||
|
|
||||||
|
max_len := min(11, len(got))
|
||||||
|
|
||||||
|
LoadDefineTemplate(w, "/models/edit.html", "data-model-create-class-table-table", c.AddMap(AnyMap{
|
||||||
|
"List": got[0:max_len],
|
||||||
|
"Page": page,
|
||||||
|
"Id": id,
|
||||||
|
"Name": name,
|
||||||
|
"Model": model,
|
||||||
|
"ShowNext": len(got) == 11,
|
||||||
|
}))
|
||||||
|
return nil
|
||||||
|
})
|
||||||
}
|
}
|
||||||
|
@ -7,78 +7,82 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
type ModelClass struct {
|
type ModelClass struct {
|
||||||
Id string
|
Id string `json:"id"`
|
||||||
ModelId string
|
ModelId string `json:"model_id"`
|
||||||
Name string
|
Name string `json:"name"`
|
||||||
}
|
}
|
||||||
|
|
||||||
func ListClasses(db *sql.DB, model_id string) (cls []ModelClass, err error) {
|
func ListClasses(db *sql.DB, model_id string) (cls []ModelClass, err error) {
|
||||||
|
|
||||||
rows, err := db.Query("select id, model_id, name from model_classes where model_id=$1", model_id)
|
rows, err := db.Query("select id, model_id, name from model_classes where model_id=$1", model_id)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
defer rows.Close()
|
defer rows.Close()
|
||||||
|
|
||||||
for rows.Next() {
|
for rows.Next() {
|
||||||
var model ModelClass
|
var model ModelClass
|
||||||
err = rows.Scan(&model.Id, &model.ModelId, &model.Name)
|
err = rows.Scan(&model.Id, &model.ModelId, &model.Name)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
cls = append(cls, model)
|
cls = append(cls, model)
|
||||||
}
|
}
|
||||||
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
func ModelHasDataPoints(db *sql.DB, model_id string) (result bool, err error) {
|
func ModelHasDataPoints(db *sql.DB, model_id string) (result bool, err error) {
|
||||||
result = false
|
result = false
|
||||||
rows, err := db.Query("select mdp.id from model_data_point as mdp join model_classes as mc on mc.id = mdp.class_id where mc.model_id = $1 limit 1;", model_id)
|
rows, err := db.Query("select mdp.id from model_data_point as mdp join model_classes as mc on mc.id = mdp.class_id where mc.model_id = $1 limit 1;", model_id)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
defer rows.Close()
|
defer rows.Close()
|
||||||
|
|
||||||
return rows.Next(), nil
|
return rows.Next(), nil
|
||||||
}
|
}
|
||||||
|
|
||||||
var ClassAlreadyExists = errors.New("Class aready exists")
|
var ClassAlreadyExists = errors.New("Class aready exists")
|
||||||
|
|
||||||
func CreateClass(db *sql.DB, model_id string, order int, name string) (id string, err error) {
|
func CreateClass(db *sql.DB, model_id string, order int, name string) (id string, err error) {
|
||||||
id = ""
|
id = ""
|
||||||
rows, err := db.Query("select id from model_classes where model_id=$1 and name=$2;", model_id, name)
|
rows, err := db.Query("select id from model_classes where model_id=$1 and name=$2;", model_id, name)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
defer rows.Close()
|
defer rows.Close()
|
||||||
|
|
||||||
if rows.Next() {
|
if rows.Next() {
|
||||||
return id, ClassAlreadyExists
|
return id, ClassAlreadyExists
|
||||||
}
|
}
|
||||||
|
|
||||||
rows, err = db.Query("insert into model_classes (model_id, name, class_order) values ($1, $2, $3) returning id;", model_id, name, order)
|
rows, err = db.Query("insert into model_classes (model_id, name, class_order) values ($1, $2, $3) returning id;", model_id, name, order)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
defer rows.Close()
|
defer rows.Close()
|
||||||
|
|
||||||
if !rows.Next() {
|
if !rows.Next() {
|
||||||
return id, errors.New("Insert did not return anything")
|
return id, errors.New("Insert did not return anything")
|
||||||
}
|
}
|
||||||
|
|
||||||
err = rows.Scan(&id)
|
err = rows.Scan(&id)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
func GetNumberOfWrongDataPoints(db *sql.DB, model_id string) (number int, err error) {
|
func GetNumberOfWrongDataPoints(db *sql.DB, model_id string) (number int, err error) {
|
||||||
number = 0
|
number = 0
|
||||||
rows, err := db.Query("select count(mdp.id) from model_data_point as mdp join model_classes as mc on mc.id = mdp.class_id where mc.model_id=$1 and mdp.status=-1;", model_id)
|
rows, err := db.Query("select count(mdp.id) from model_data_point as mdp join model_classes as mc on mc.id = mdp.class_id where mc.model_id=$1 and mdp.status=-1;", model_id)
|
||||||
if err != nil { return }
|
if err != nil {
|
||||||
defer rows.Close()
|
return
|
||||||
// TODO not an error because if there is no result means that there is no need to count
|
}
|
||||||
if !rows.Next() { return }
|
defer rows.Close()
|
||||||
err = rows.Scan(&number)
|
// TODO not an error because if there is no result means that there is no need to count
|
||||||
return
|
if !rows.Next() {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
err = rows.Scan(&number)
|
||||||
|
return
|
||||||
}
|
}
|
||||||
|
@ -126,7 +126,7 @@ func processZipFile(c *Context, model *BaseModel) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
file_path := path.Join(base_path, data_point_id + "." + model.Format)
|
file_path := path.Join(base_path, data_point_id+"."+model.Format)
|
||||||
f, err := os.Create(file_path)
|
f, err := os.Create(file_path)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
fmt.Printf("Could not create file %s\n", file_path)
|
fmt.Printf("Could not create file %s\n", file_path)
|
||||||
@ -137,13 +137,13 @@ func processZipFile(c *Context, model *BaseModel) {
|
|||||||
f.Write(file_data)
|
f.Write(file_data)
|
||||||
|
|
||||||
if !testImgForModel(c, model, file_path) {
|
if !testImgForModel(c, model, file_path) {
|
||||||
c.Logger.Errorf("Image did not have valid format for model %s (in zip: %s)!", file_path, file.Name)
|
c.Logger.Errorf("Image did not have valid format for model %s (in zip: %s)!", file_path, file.Name)
|
||||||
c.Logger.Warn("Not failling updating data point to status -1")
|
c.Logger.Warn("Not failling updating data point to status -1")
|
||||||
message := "Image did not have valid format for the model"
|
message := "Image did not have valid format for the model"
|
||||||
if err = model_classes.UpdateDataPointStatus(c.Db, data_point_id, -1, &message); err != nil {
|
if err = model_classes.UpdateDataPointStatus(c.Db, data_point_id, -1, &message); err != nil {
|
||||||
c.Logger.Error("Failed to update data point status")
|
c.Logger.Error("Failed to update data point status")
|
||||||
ModelUpdateStatus(c, model.Id, FAILED_PREPARING_ZIP_FILE)
|
ModelUpdateStatus(c, model.Id, FAILED_PREPARING_ZIP_FILE)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -156,9 +156,58 @@ func handleDataUpload(handle *Handle) {
|
|||||||
if !CheckAuthLevel(1, w, r, c) {
|
if !CheckAuthLevel(1, w, r, c) {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
if c.Mode == JSON {
|
if c.Mode == JSON {
|
||||||
// TODO improve message
|
read_form, err := r.MultipartReader()
|
||||||
return ErrorCode(nil, 400, nil)
|
if err != nil {
|
||||||
|
return c.JsonBadRequest("Please provide a valid form data request!")
|
||||||
|
}
|
||||||
|
|
||||||
|
var id string
|
||||||
|
var file []byte
|
||||||
|
|
||||||
|
for {
|
||||||
|
part, err_part := read_form.NextPart()
|
||||||
|
if err_part == io.EOF {
|
||||||
|
break
|
||||||
|
} else if err_part != nil {
|
||||||
|
return c.JsonBadRequest("Please provide a valid form data request!")
|
||||||
|
}
|
||||||
|
if part.FormName() == "id" {
|
||||||
|
buf := new(bytes.Buffer)
|
||||||
|
buf.ReadFrom(part)
|
||||||
|
id = buf.String()
|
||||||
|
}
|
||||||
|
if part.FormName() == "file" {
|
||||||
|
buf := new(bytes.Buffer)
|
||||||
|
buf.ReadFrom(part)
|
||||||
|
file = buf.Bytes()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
model, err := GetBaseModel(handle.Db, id)
|
||||||
|
if err == ModelNotFoundError {
|
||||||
|
return c.SendJSONStatus(http.StatusNotFound, "Model not found")
|
||||||
|
} else if err != nil {
|
||||||
|
return Error500(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// TODO mk this path configurable
|
||||||
|
dir_path := path.Join("savedData", id)
|
||||||
|
|
||||||
|
f, err := os.Create(path.Join(dir_path, "base_data.zip"))
|
||||||
|
if err != nil {
|
||||||
|
return Error500(err)
|
||||||
|
}
|
||||||
|
defer f.Close()
|
||||||
|
|
||||||
|
f.Write(file)
|
||||||
|
|
||||||
|
ModelUpdateStatus(c, id, PREPARING_ZIP_FILE)
|
||||||
|
|
||||||
|
go processZipFile(c, model)
|
||||||
|
|
||||||
|
return c.SendJSON(model.Id)
|
||||||
}
|
}
|
||||||
|
|
||||||
read_form, err := r.MultipartReader()
|
read_form, err := r.MultipartReader()
|
||||||
@ -223,7 +272,45 @@ func handleDataUpload(handle *Handle) {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
if c.Mode == JSON {
|
if c.Mode == JSON {
|
||||||
panic("Handle delete zip file json")
|
|
||||||
|
type ModelData struct {
|
||||||
|
Id string `json:"id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
var dat ModelData
|
||||||
|
|
||||||
|
if err := c.ToJSON(r, &dat); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
model, err := GetBaseModel(handle.Db, dat.Id)
|
||||||
|
if err == ModelNotFoundError {
|
||||||
|
return c.SendJSONStatus(http.StatusNotFound, "Model not found");
|
||||||
|
} else if err != nil {
|
||||||
|
return Error500(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if model.Status != FAILED_PREPARING_ZIP_FILE {
|
||||||
|
return c.SendJSONStatus(http.StatusNotFound, "Model not in the correct status")
|
||||||
|
}
|
||||||
|
|
||||||
|
err = os.Remove(path.Join("savedData", model.Id, "base_data.zip"))
|
||||||
|
if err != nil {
|
||||||
|
return Error500(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
err = os.RemoveAll(path.Join("savedData", model.Id, "data"))
|
||||||
|
if err != nil {
|
||||||
|
return Error500(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
_, err = handle.Db.Exec("delete from model_classes where model_id=$1;", model.Id)
|
||||||
|
if err != nil {
|
||||||
|
return Error500(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
ModelUpdateStatus(c, model.Id, CONFIRM_PRE_TRAINING)
|
||||||
|
return c.SendJSON(model.Id)
|
||||||
}
|
}
|
||||||
|
|
||||||
f, err := MyParseForm(r)
|
f, err := MyParseForm(r)
|
||||||
|
@ -6,14 +6,238 @@ import (
|
|||||||
|
|
||||||
model_classes "git.andr3h3nriqu3s.com/andr3/fyp/logic/models/classes"
|
model_classes "git.andr3h3nriqu3s.com/andr3/fyp/logic/models/classes"
|
||||||
. "git.andr3h3nriqu3s.com/andr3/fyp/logic/models/utils"
|
. "git.andr3h3nriqu3s.com/andr3/fyp/logic/models/utils"
|
||||||
|
"git.andr3h3nriqu3s.com/andr3/fyp/logic/utils"
|
||||||
. "git.andr3h3nriqu3s.com/andr3/fyp/logic/utils"
|
. "git.andr3h3nriqu3s.com/andr3/fyp/logic/utils"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
func handleJson(w http.ResponseWriter, r *http.Request, c *Context) *Error {
|
||||||
|
if !CheckAuthLevel(1, w, r, c) {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
id, err := GetIdFromUrl(r, "id")
|
||||||
|
if err != nil {
|
||||||
|
return c.JsonBadRequest("Model not found")
|
||||||
|
}
|
||||||
|
|
||||||
|
type rowmodel struct {
|
||||||
|
Name string `json:"name"`
|
||||||
|
Status int `json:"status"`
|
||||||
|
Id string `json:"id"`
|
||||||
|
Width *int `json:"width"`
|
||||||
|
Height *int `json:"height"`
|
||||||
|
Color_mode *string `json:"color_mode"`
|
||||||
|
Format string `json:"format"`
|
||||||
|
Model_type int `json:"model_type"`
|
||||||
|
}
|
||||||
|
|
||||||
|
var model rowmodel = rowmodel{}
|
||||||
|
err = utils.GetDBOnce(c, &model, "models where id=$1 and user_id=$2", id, c.User.Id)
|
||||||
|
if err == NotFoundError {
|
||||||
|
return c.SendJSONStatus(404, "Model not found")
|
||||||
|
} else if err != nil {
|
||||||
|
return c.Error500(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return c.SendJSON(model)
|
||||||
|
|
||||||
|
/*
|
||||||
|
|
||||||
|
// Handle errors
|
||||||
|
// All errors will be negative
|
||||||
|
if model.Status < 0 {
|
||||||
|
LoadBasedOnAnswer(c.Mode, w, "/models/edit.html", c.AddMap(AnyMap{
|
||||||
|
"Model": model,
|
||||||
|
}))
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
switch model.Status {
|
||||||
|
case READY:
|
||||||
|
LoadBasedOnAnswer(c.Mode, w, "/models/edit.html", c.AddMap(AnyMap{
|
||||||
|
"Model": model,
|
||||||
|
}))
|
||||||
|
case TRAINING:
|
||||||
|
|
||||||
|
type defrow struct {
|
||||||
|
Id string
|
||||||
|
Status int
|
||||||
|
EpochProgress int
|
||||||
|
Epoch int
|
||||||
|
Accuracy float64
|
||||||
|
}
|
||||||
|
|
||||||
|
defs := []defrow{}
|
||||||
|
|
||||||
|
if model.Type == 2 {
|
||||||
|
def_rows, err := c.Db.Query("select md.id, md.status, md.epoch, h.epoch_progress, h.accuracy from model_definition as md inner join exp_model_head as h on h.def_id = md.id where md.model_id=$1 order by md.created_on asc", model.Id)
|
||||||
|
if err != nil {
|
||||||
|
return c.Error500(err)
|
||||||
|
}
|
||||||
|
defer def_rows.Close()
|
||||||
|
|
||||||
|
for def_rows.Next() {
|
||||||
|
var def defrow
|
||||||
|
err = def_rows.Scan(&def.Id, &def.Status, &def.Epoch, &def.EpochProgress, &def.Accuracy)
|
||||||
|
if err != nil {
|
||||||
|
return c.Error500(err)
|
||||||
|
}
|
||||||
|
defs = append(defs, def)
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
def_rows, err := c.Db.Query("select id, status, epoch, epoch_progress, accuracy from model_definition where model_id=$1 order by created_on asc", model.Id)
|
||||||
|
if err != nil {
|
||||||
|
return c.Error500(err)
|
||||||
|
}
|
||||||
|
defer def_rows.Close()
|
||||||
|
|
||||||
|
for def_rows.Next() {
|
||||||
|
var def defrow
|
||||||
|
err = def_rows.Scan(&def.Id, &def.Status, &def.Epoch, &def.EpochProgress, &def.Accuracy)
|
||||||
|
if err != nil {
|
||||||
|
return c.Error500(err)
|
||||||
|
}
|
||||||
|
defs = append(defs, def)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
type layerdef struct {
|
||||||
|
id string
|
||||||
|
LayerType int
|
||||||
|
Shape string
|
||||||
|
}
|
||||||
|
|
||||||
|
layers := []layerdef{}
|
||||||
|
|
||||||
|
for _, def := range defs {
|
||||||
|
if def.Status == MODEL_DEFINITION_STATUS_TRAINING {
|
||||||
|
rows, err := c.Db.Query("select id, layer_type, shape from model_definition_layer where def_id=$1 order by layer_order asc;", def.Id)
|
||||||
|
if err != nil {
|
||||||
|
return c.Error500(err)
|
||||||
|
}
|
||||||
|
defer rows.Close()
|
||||||
|
|
||||||
|
for rows.Next() {
|
||||||
|
var layerdef layerdef
|
||||||
|
err = rows.Scan(&layerdef.id, &layerdef.LayerType, &layerdef.Shape)
|
||||||
|
if err != nil {
|
||||||
|
return c.Error500(err)
|
||||||
|
}
|
||||||
|
layers = append(layers, layerdef)
|
||||||
|
}
|
||||||
|
|
||||||
|
if model.Type == 2 {
|
||||||
|
|
||||||
|
type lastLayerType struct {
|
||||||
|
Id string
|
||||||
|
Range_start int
|
||||||
|
Range_end int
|
||||||
|
}
|
||||||
|
|
||||||
|
var lastLayer lastLayerType
|
||||||
|
|
||||||
|
err := GetDBOnce(c, &lastLayer, "exp_model_head where def_id=$1 and status=3;", def.Id)
|
||||||
|
if err != nil {
|
||||||
|
return c.Error500(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
layers = append(layers, layerdef{
|
||||||
|
id: lastLayer.Id,
|
||||||
|
LayerType: LAYER_DENSE,
|
||||||
|
Shape: fmt.Sprintf("%d, 1", lastLayer.Range_end-lastLayer.Range_start+1),
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
sep_mod := 100
|
||||||
|
if len(layers) > 8 {
|
||||||
|
sep_mod = 100 - (len(layers)-8)*10
|
||||||
|
}
|
||||||
|
|
||||||
|
if sep_mod < 10 {
|
||||||
|
sep_mod = 10
|
||||||
|
}
|
||||||
|
|
||||||
|
LoadBasedOnAnswer(c.Mode, w, "/models/edit.html", c.AddMap(AnyMap{
|
||||||
|
"Model": model,
|
||||||
|
"Defs": defs,
|
||||||
|
"Layers": layers,
|
||||||
|
"SepMod": sep_mod,
|
||||||
|
}))
|
||||||
|
case PREPARING_ZIP_FILE:
|
||||||
|
LoadBasedOnAnswer(c.Mode, w, "/models/edit.html", c.AddMap(AnyMap{
|
||||||
|
"Model": model,
|
||||||
|
}))
|
||||||
|
default:
|
||||||
|
fmt.Printf("Unkown Status: %d\n", model.Status)
|
||||||
|
return Error500(nil)
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
return nil
|
||||||
|
*/
|
||||||
|
}
|
||||||
|
|
||||||
func handleEdit(handle *Handle) {
|
func handleEdit(handle *Handle) {
|
||||||
handle.GetHTML("/models/edit", func(w http.ResponseWriter, r *http.Request, c *Context) *Error {
|
handle.Get("/models/edit/classes", func(w http.ResponseWriter, r *http.Request, c *Context) *Error {
|
||||||
if !CheckAuthLevel(1, w, r, c) {
|
if !CheckAuthLevel(1, w, r, c) {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
if c.Mode != JSON {
|
||||||
|
return c.ErrorCode(nil, 400, AnyMap{})
|
||||||
|
}
|
||||||
|
|
||||||
|
id, err := GetIdFromUrl(r, "id")
|
||||||
|
if err != nil {
|
||||||
|
return c.SendJSONStatus(http.StatusNotFound, "Model not found")
|
||||||
|
}
|
||||||
|
|
||||||
|
model, err := GetBaseModel(c.Db, id)
|
||||||
|
if err == ModelNotFoundError {
|
||||||
|
return c.SendJSONStatus(http.StatusNotFound, "Model not found")
|
||||||
|
} else if err != nil {
|
||||||
|
return c.Error500(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
wrong_number, err := model_classes.GetNumberOfWrongDataPoints(c.Db, model.Id)
|
||||||
|
if err != nil {
|
||||||
|
return c.Error500(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
cls, err := model_classes.ListClasses(handle.Db, id)
|
||||||
|
if err != nil {
|
||||||
|
return c.Error500(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
has_data, err := model_classes.ModelHasDataPoints(handle.Db, id)
|
||||||
|
if err != nil {
|
||||||
|
return Error500(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
type ReturnType struct {
|
||||||
|
Classes []model_classes.ModelClass `json:"classes"`
|
||||||
|
HasData bool `json:"has_data"`
|
||||||
|
NumberOfInvalidImages int `json:"number_of_invalid_images"`
|
||||||
|
}
|
||||||
|
|
||||||
|
return c.SendJSON(ReturnType{
|
||||||
|
Classes: cls,
|
||||||
|
HasData: has_data,
|
||||||
|
NumberOfInvalidImages: wrong_number,
|
||||||
|
})
|
||||||
|
})
|
||||||
|
|
||||||
|
handle.Get("/models/edit", func(w http.ResponseWriter, r *http.Request, c *Context) *Error {
|
||||||
|
if !CheckAuthLevel(1, w, r, c) {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
if c.Mode == JSON {
|
||||||
|
return handleJson(w, r, c)
|
||||||
|
}
|
||||||
|
|
||||||
id, err := GetIdFromUrl(r, "id")
|
id, err := GetIdFromUrl(r, "id")
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@ -109,38 +333,37 @@ func handleEdit(handle *Handle) {
|
|||||||
|
|
||||||
defs := []defrow{}
|
defs := []defrow{}
|
||||||
|
|
||||||
if model.Type == 2 {
|
if model.Type == 2 {
|
||||||
def_rows, err := c.Db.Query("select md.id, md.status, md.epoch, h.epoch_progress, h.accuracy from model_definition as md inner join exp_model_head as h on h.def_id = md.id where md.model_id=$1 order by md.created_on asc", model.Id)
|
def_rows, err := c.Db.Query("select md.id, md.status, md.epoch, h.epoch_progress, h.accuracy from model_definition as md inner join exp_model_head as h on h.def_id = md.id where md.model_id=$1 order by md.created_on asc", model.Id)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return c.Error500(err)
|
return c.Error500(err)
|
||||||
}
|
}
|
||||||
defer def_rows.Close()
|
defer def_rows.Close()
|
||||||
|
|
||||||
for def_rows.Next() {
|
for def_rows.Next() {
|
||||||
var def defrow
|
var def defrow
|
||||||
err = def_rows.Scan(&def.Id, &def.Status, &def.Epoch, &def.EpochProgress, &def.Accuracy)
|
err = def_rows.Scan(&def.Id, &def.Status, &def.Epoch, &def.EpochProgress, &def.Accuracy)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return c.Error500(err)
|
return c.Error500(err)
|
||||||
}
|
}
|
||||||
defs = append(defs, def)
|
defs = append(defs, def)
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
def_rows, err := c.Db.Query("select id, status, epoch, epoch_progress, accuracy from model_definition where model_id=$1 order by created_on asc", model.Id)
|
def_rows, err := c.Db.Query("select id, status, epoch, epoch_progress, accuracy from model_definition where model_id=$1 order by created_on asc", model.Id)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return c.Error500(err)
|
return c.Error500(err)
|
||||||
}
|
}
|
||||||
defer def_rows.Close()
|
defer def_rows.Close()
|
||||||
|
|
||||||
for def_rows.Next() {
|
|
||||||
var def defrow
|
|
||||||
err = def_rows.Scan(&def.Id, &def.Status, &def.Epoch, &def.EpochProgress, &def.Accuracy)
|
|
||||||
if err != nil {
|
|
||||||
return c.Error500(err)
|
|
||||||
}
|
|
||||||
defs = append(defs, def)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
|
for def_rows.Next() {
|
||||||
|
var def defrow
|
||||||
|
err = def_rows.Scan(&def.Id, &def.Status, &def.Epoch, &def.EpochProgress, &def.Accuracy)
|
||||||
|
if err != nil {
|
||||||
|
return c.Error500(err)
|
||||||
|
}
|
||||||
|
defs = append(defs, def)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
type layerdef struct {
|
type layerdef struct {
|
||||||
id string
|
id string
|
||||||
@ -175,7 +398,7 @@ func handleEdit(handle *Handle) {
|
|||||||
Range_end int
|
Range_end int
|
||||||
}
|
}
|
||||||
|
|
||||||
var lastLayer lastLayerType
|
var lastLayer lastLayerType
|
||||||
|
|
||||||
err := GetDBOnce(c, &lastLayer, "exp_model_head where def_id=$1 and status=3;", def.Id)
|
err := GetDBOnce(c, &lastLayer, "exp_model_head where def_id=$1 and status=3;", def.Id)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@ -183,9 +406,9 @@ func handleEdit(handle *Handle) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
layers = append(layers, layerdef{
|
layers = append(layers, layerdef{
|
||||||
id: lastLayer.Id,
|
id: lastLayer.Id,
|
||||||
LayerType: LAYER_DENSE,
|
LayerType: LAYER_DENSE,
|
||||||
Shape: fmt.Sprintf("%d, 1", lastLayer.Range_end-lastLayer.Range_start + 1),
|
Shape: fmt.Sprintf("%d, 1", lastLayer.Range_end-lastLayer.Range_start+1),
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -219,4 +442,5 @@ func handleEdit(handle *Handle) {
|
|||||||
|
|
||||||
return nil
|
return nil
|
||||||
})
|
})
|
||||||
|
|
||||||
}
|
}
|
||||||
|
@ -3,25 +3,43 @@ package models
|
|||||||
import (
|
import (
|
||||||
"net/http"
|
"net/http"
|
||||||
|
|
||||||
|
"git.andr3h3nriqu3s.com/andr3/fyp/logic/utils"
|
||||||
. "git.andr3h3nriqu3s.com/andr3/fyp/logic/utils"
|
. "git.andr3h3nriqu3s.com/andr3/fyp/logic/utils"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
func handleList(handle *Handle) {
|
func handleList(handle *Handle) {
|
||||||
// TODO json
|
handle.Get("/models", func(w http.ResponseWriter, r *http.Request, c *Context) *Error {
|
||||||
handle.GetHTML("/models", func(w http.ResponseWriter, r *http.Request, c *Context) *Error {
|
|
||||||
if c.Mode == JSON {
|
|
||||||
panic("TODO JSON")
|
|
||||||
}
|
|
||||||
if !CheckAuthLevel(1, w, r, c) {
|
if !CheckAuthLevel(1, w, r, c) {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// TODO handle admin
|
||||||
|
if c.Mode == JSON {
|
||||||
|
rows, err := handle.Db.Query("select id, name from models where user_id=$1;", c.User.Id)
|
||||||
|
if err != nil {
|
||||||
|
return Error500(err)
|
||||||
|
}
|
||||||
|
defer rows.Close()
|
||||||
|
|
||||||
|
type Row struct {
|
||||||
|
Name string `json:"name"`
|
||||||
|
Id string `json:"id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
got, err := utils.GetDbMultitple[Row](c, "models where user_id=$1", c.User.Id);
|
||||||
|
if err != nil {
|
||||||
|
c.Logger.Warn("HERE 6")
|
||||||
|
return c.Error500(nil)
|
||||||
|
}
|
||||||
|
|
||||||
|
return c.SendJSON(got)
|
||||||
|
}
|
||||||
|
|
||||||
rows, err := handle.Db.Query("select id, name from models where user_id=$1;", c.User.Id)
|
rows, err := handle.Db.Query("select id, name from models where user_id=$1;", c.User.Id)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return Error500(err)
|
return Error500(err)
|
||||||
}
|
}
|
||||||
defer rows.Close()
|
defer rows.Close()
|
||||||
|
|
||||||
type row struct {
|
type row struct {
|
||||||
Name string
|
Name string
|
||||||
|
@ -346,6 +346,7 @@ type Context struct {
|
|||||||
Mode AnswerType
|
Mode AnswerType
|
||||||
Logger *log.Logger
|
Logger *log.Logger
|
||||||
Db *sql.DB
|
Db *sql.DB
|
||||||
|
Writer http.ResponseWriter
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c Context) ToJSON(r *http.Request, dat any) *Error {
|
func (c Context) ToJSON(r *http.Request, dat any) *Error {
|
||||||
@ -360,33 +361,40 @@ func (c Context) ToJSON(r *http.Request, dat any) *Error {
|
|||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c Context) SendJSON(w http.ResponseWriter, dat any) *Error {
|
func (c Context) SendJSON(dat any) *Error {
|
||||||
w.Header().Add("content-type", "application/json")
|
c.Writer.Header().Add("content-type", "application/json")
|
||||||
text, err := json.Marshal(dat)
|
text, err := json.Marshal(dat)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return c.Error500(err)
|
return c.Error500(err)
|
||||||
}
|
}
|
||||||
if _, err = w.Write(text); err != nil {
|
if _, err = c.Writer.Write(text); err != nil {
|
||||||
return c.Error500(err)
|
return c.Error500(err)
|
||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c Context) SendJSONStatus(w http.ResponseWriter, status int, dat any) *Error {
|
func (c Context) SendJSONStatus(status int, dat any) *Error {
|
||||||
w.Header().Add("content-type", "application/json")
|
c.Writer.Header().Add("content-type", "application/json")
|
||||||
w.WriteHeader(status)
|
c.Writer.WriteHeader(status)
|
||||||
text, err := json.Marshal(dat)
|
text, err := json.Marshal(dat)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return c.Error500(err)
|
return c.Error500(err)
|
||||||
}
|
}
|
||||||
if _, err = w.Write(text); err != nil {
|
if _, err = c.Writer.Write(text); err != nil {
|
||||||
return c.Error500(err)
|
return c.Error500(err)
|
||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c Context) JsonBadRequest(w http.ResponseWriter, dat any) *Error {
|
func (c Context) JsonBadRequest(dat any) *Error {
|
||||||
return c.SendJSONStatus(w, http.StatusBadRequest, dat)
|
return c.SendJSONStatus(http.StatusBadRequest, dat)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c Context) JsonErrorBadRequest(err error, dat any) *Error {
|
||||||
|
c.SetReportCaller(true)
|
||||||
|
c.Logger.Error("Error while processing request", "err", err, "dat", dat)
|
||||||
|
c.SetReportCaller(false)
|
||||||
|
return c.SendJSONStatus(http.StatusBadRequest, dat)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c Context) Error400(err error, message string, w http.ResponseWriter, path string, base string, data AnyMap) *Error {
|
func (c Context) Error400(err error, message string, w http.ResponseWriter, path string, base string, data AnyMap) *Error {
|
||||||
@ -460,7 +468,7 @@ func (c *Context) requireAuth(w http.ResponseWriter, r *http.Request) bool {
|
|||||||
|
|
||||||
var LogoffError = errors.New("Invalid token!")
|
var LogoffError = errors.New("Invalid token!")
|
||||||
|
|
||||||
func (x Handle) createContext(handler *Handle, mode AnswerType, r *http.Request) (*Context, error) {
|
func (x Handle) createContext(handler *Handle, mode AnswerType, r *http.Request, w http.ResponseWriter) (*Context, error) {
|
||||||
|
|
||||||
var token *string
|
var token *string
|
||||||
|
|
||||||
@ -486,12 +494,12 @@ func (x Handle) createContext(handler *Handle, mode AnswerType, r *http.Request)
|
|||||||
|
|
||||||
|
|
||||||
// TODO check that the token is still valid
|
// TODO check that the token is still valid
|
||||||
|
|
||||||
if token == nil {
|
if token == nil {
|
||||||
return &Context{
|
return &Context{
|
||||||
Mode: mode,
|
Mode: mode,
|
||||||
Logger: logger,
|
Logger: logger,
|
||||||
Db: handler.Db,
|
Db: handler.Db,
|
||||||
|
Writer: w,
|
||||||
}, nil
|
}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -500,7 +508,7 @@ func (x Handle) createContext(handler *Handle, mode AnswerType, r *http.Request)
|
|||||||
return nil, errors.Join(err, LogoffError)
|
return nil, errors.Join(err, LogoffError)
|
||||||
}
|
}
|
||||||
|
|
||||||
return &Context{token, user, mode, logger, handler.Db}, nil
|
return &Context{token, user, mode, logger, handler.Db, w}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// TODO check if I can use http.Redirect
|
// TODO check if I can use http.Redirect
|
||||||
@ -664,21 +672,28 @@ func NewHandler(db *sql.DB) *Handle {
|
|||||||
if r.Header.Get("Request-Type") == "htmlfull" {
|
if r.Header.Get("Request-Type") == "htmlfull" {
|
||||||
ans = HTMLFULL
|
ans = HTMLFULL
|
||||||
}
|
}
|
||||||
if r.Header.Get("content-type") == "application/json" {
|
if r.Header.Get("content-type") == "application/json" || r.Header.Get("response-type") == "application/json" {
|
||||||
ans = JSON
|
ans = JSON
|
||||||
}
|
}
|
||||||
|
|
||||||
//TODO JSON
|
if !strings.HasPrefix(r.URL.Path, "/api") {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
r.URL.Path = strings.Replace(r.URL.Path, "/api", "", 1)
|
||||||
|
|
||||||
|
w.Header().Add("Access-Control-Allow-Origin", "*")
|
||||||
|
w.Header().Add("Access-Control-Allow-Headers", "*")
|
||||||
|
w.Header().Add("Access-Control-Allow-Methods", "*")
|
||||||
|
|
||||||
//Login state
|
//Login state
|
||||||
context, err := x.createContext(x, ans, r)
|
context, err := x.createContext(x, ans, r, w)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
Logoff(ans, w, r)
|
Logoff(ans, w, r)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
w.Header().Add("Access-Control-Allow-Origin", "*")
|
// context.Logger.Info("Parsing", "path", r.URL.Path)
|
||||||
w.Header().Add("Access-Control-Allow-Headers", "*")
|
|
||||||
|
|
||||||
if r.Method == "GET" {
|
if r.Method == "GET" {
|
||||||
x.handleGets(w, r, context)
|
x.handleGets(w, r, context)
|
||||||
|
3
main.go
3
main.go
@ -31,7 +31,7 @@ func main() {
|
|||||||
panic(err)
|
panic(err)
|
||||||
}
|
}
|
||||||
defer db.Close()
|
defer db.Close()
|
||||||
log.Info("Starting server on :8000!")
|
log.Info("Starting server on :5002!")
|
||||||
|
|
||||||
//TODO check if file structure exists to save data
|
//TODO check if file structure exists to save data
|
||||||
handle := NewHandler(db)
|
handle := NewHandler(db)
|
||||||
@ -48,6 +48,7 @@ func main() {
|
|||||||
handle.StaticFiles("/js/", ".js", "text/javascript")
|
handle.StaticFiles("/js/", ".js", "text/javascript")
|
||||||
handle.ReadFiles("/imgs/", "views", ".png", "image/png;")
|
handle.ReadFiles("/imgs/", "views", ".png", "image/png;")
|
||||||
handle.ReadTypesFiles("/savedData/", ".", []string{".png", ".jpeg"}, []string{"image/png", "image/jpeg"})
|
handle.ReadTypesFiles("/savedData/", ".", []string{".png", ".jpeg"}, []string{"image/png", "image/jpeg"})
|
||||||
|
handle.ReadTypesFiles("/api/savedData/", ".", []string{".png", ".jpeg"}, []string{"image/png", "image/jpeg"})
|
||||||
|
|
||||||
handle.GetHTML("/", AnswerTemplate("index.html", nil, 0))
|
handle.GetHTML("/", AnswerTemplate("index.html", nil, 0))
|
||||||
|
|
||||||
|
29
nginx.dev.conf
Normal file
29
nginx.dev.conf
Normal file
@ -0,0 +1,29 @@
|
|||||||
|
|
||||||
|
events {
|
||||||
|
worker_connections 1024;
|
||||||
|
}
|
||||||
|
|
||||||
|
http {
|
||||||
|
|
||||||
|
map $http_upgrade $connection_upgrade {
|
||||||
|
default upgrade;
|
||||||
|
'' close;
|
||||||
|
}
|
||||||
|
|
||||||
|
server {
|
||||||
|
listen 8000;
|
||||||
|
|
||||||
|
location / {
|
||||||
|
proxy_http_version 1.1;
|
||||||
|
proxy_pass http://127.0.0.1:5001;
|
||||||
|
proxy_set_header X-Forwarded-For $proxy_add_x_forwarded_for;
|
||||||
|
proxy_set_header Upgrade $http_upgrade;
|
||||||
|
proxy_set_header Connection $connection_upgrade;
|
||||||
|
}
|
||||||
|
|
||||||
|
location /api {
|
||||||
|
|
||||||
|
proxy_pass http://127.0.0.1:5002;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
34
users.go
34
users.go
@ -94,7 +94,7 @@ func usersEndpints(db *sql.DB, handle *Handle) {
|
|||||||
// TODO Give this to the generateToken function
|
// TODO Give this to the generateToken function
|
||||||
token, login := generateToken(db, dat.Email, dat.Password)
|
token, login := generateToken(db, dat.Email, dat.Password)
|
||||||
if !login {
|
if !login {
|
||||||
return c.SendJSONStatus(w, http.StatusUnauthorized, "Email or password are incorrect")
|
return c.SendJSONStatus(http.StatusUnauthorized, "Email or password are incorrect")
|
||||||
}
|
}
|
||||||
|
|
||||||
user, err := dbtypes.UserFromToken(c.Db, token)
|
user, err := dbtypes.UserFromToken(c.Db, token)
|
||||||
@ -118,7 +118,7 @@ func usersEndpints(db *sql.DB, handle *Handle) {
|
|||||||
Email: user.Email,
|
Email: user.Email,
|
||||||
}
|
}
|
||||||
|
|
||||||
return c.SendJSON(w, userReturn)
|
return c.SendJSON(userReturn)
|
||||||
}
|
}
|
||||||
|
|
||||||
r.ParseForm()
|
r.ParseForm()
|
||||||
@ -170,7 +170,7 @@ func usersEndpints(db *sql.DB, handle *Handle) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
if len(dat.Username) == 0 || len(dat.Password) == 0 || len(dat.Email) == 0 {
|
if len(dat.Username) == 0 || len(dat.Password) == 0 || len(dat.Email) == 0 {
|
||||||
return c.SendJSONStatus(w, http.StatusBadRequest, "Please provide a valid json")
|
return c.SendJSONStatus(http.StatusBadRequest, "Please provide a valid json")
|
||||||
}
|
}
|
||||||
|
|
||||||
rows, err := db.Query("select username, email from users where username=$1 or email=$2;", dat.Username, dat.Email)
|
rows, err := db.Query("select username, email from users where username=$1 or email=$2;", dat.Username, dat.Email)
|
||||||
@ -187,16 +187,16 @@ func usersEndpints(db *sql.DB, handle *Handle) {
|
|||||||
return c.Error500(err)
|
return c.Error500(err)
|
||||||
}
|
}
|
||||||
if db_email == dat.Email {
|
if db_email == dat.Email {
|
||||||
return c.SendJSONStatus(w, http.StatusBadRequest, "Email already in use!")
|
return c.SendJSONStatus(http.StatusBadRequest, "Email already in use!")
|
||||||
}
|
}
|
||||||
if db_username == dat.Username {
|
if db_username == dat.Username {
|
||||||
return c.SendJSONStatus(w, http.StatusBadRequest, "Username already in use!")
|
return c.SendJSONStatus(http.StatusBadRequest, "Username already in use!")
|
||||||
}
|
}
|
||||||
panic("Unrechable")
|
panic("Unrechable")
|
||||||
}
|
}
|
||||||
|
|
||||||
if len([]byte(dat.Password)) > 68 {
|
if len([]byte(dat.Password)) > 68 {
|
||||||
return c.JsonBadRequest(w, "Password is to long!")
|
return c.JsonBadRequest("Password is to long!")
|
||||||
}
|
}
|
||||||
|
|
||||||
salt := generateSalt()
|
salt := generateSalt()
|
||||||
@ -213,7 +213,7 @@ func usersEndpints(db *sql.DB, handle *Handle) {
|
|||||||
// TODO Give this to the generateToken function
|
// TODO Give this to the generateToken function
|
||||||
token, login := generateToken(db, dat.Email, dat.Password)
|
token, login := generateToken(db, dat.Email, dat.Password)
|
||||||
if !login {
|
if !login {
|
||||||
return c.SendJSONStatus(w, 500, "Could not login after creatting account please try again later")
|
return c.SendJSONStatus(500, "Could not login after creatting account please try again later")
|
||||||
}
|
}
|
||||||
|
|
||||||
user, err := dbtypes.UserFromToken(c.Db, token)
|
user, err := dbtypes.UserFromToken(c.Db, token)
|
||||||
@ -237,7 +237,7 @@ func usersEndpints(db *sql.DB, handle *Handle) {
|
|||||||
Email: user.Email,
|
Email: user.Email,
|
||||||
}
|
}
|
||||||
|
|
||||||
return c.SendJSON(w, userReturn)
|
return c.SendJSON(userReturn)
|
||||||
}
|
}
|
||||||
|
|
||||||
r.ParseForm()
|
r.ParseForm()
|
||||||
@ -357,7 +357,7 @@ func usersEndpints(db *sql.DB, handle *Handle) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
if dat.Id != c.User.Id && c.User.UserType != int(dbtypes.User_Admin) {
|
if dat.Id != c.User.Id && c.User.UserType != int(dbtypes.User_Admin) {
|
||||||
return c.SendJSONStatus(w, 401, "You need to be an admin to update another users account")
|
return c.SendJSONStatus(403, "You need to be an admin to update another users account")
|
||||||
}
|
}
|
||||||
|
|
||||||
if dat.Id != c.User.Id {
|
if dat.Id != c.User.Id {
|
||||||
@ -367,7 +367,7 @@ func usersEndpints(db *sql.DB, handle *Handle) {
|
|||||||
|
|
||||||
err := utils.GetDBOnce(c, &data, "users where id=$1", dat.Id)
|
err := utils.GetDBOnce(c, &data, "users where id=$1", dat.Id)
|
||||||
if err == NotFoundError {
|
if err == NotFoundError {
|
||||||
return c.JsonBadRequest(w, "User does not exist")
|
return c.JsonBadRequest("User does not exist")
|
||||||
} else if err != nil {
|
} else if err != nil {
|
||||||
return c.Error500(err)
|
return c.Error500(err)
|
||||||
}
|
}
|
||||||
@ -384,9 +384,9 @@ func usersEndpints(db *sql.DB, handle *Handle) {
|
|||||||
|
|
||||||
if err != NotFoundError {
|
if err != NotFoundError {
|
||||||
if data.Id == dat.Id {
|
if data.Id == dat.Id {
|
||||||
return c.JsonBadRequest(w, "Email is the name as the previous one!")
|
return c.JsonBadRequest("Email is the name as the previous one!")
|
||||||
} else {
|
} else {
|
||||||
return c.JsonBadRequest(w, "Email already in use")
|
return c.JsonBadRequest("Email already in use")
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -414,7 +414,7 @@ func usersEndpints(db *sql.DB, handle *Handle) {
|
|||||||
UserType: user.User_Type,
|
UserType: user.User_Type,
|
||||||
}
|
}
|
||||||
|
|
||||||
return c.SendJSON(w, toReturnUser)
|
return c.SendJSON(toReturnUser)
|
||||||
})
|
})
|
||||||
|
|
||||||
handle.Post("/user/info/email", func(w http.ResponseWriter, r *http.Request, c *Context) *Error {
|
handle.Post("/user/info/email", func(w http.ResponseWriter, r *http.Request, c *Context) *Error {
|
||||||
@ -461,18 +461,18 @@ func usersEndpints(db *sql.DB, handle *Handle) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
if dat.Password == "" {
|
if dat.Password == "" {
|
||||||
return c.JsonBadRequest(w, "Password can not be empty")
|
return c.JsonBadRequest("Password can not be empty")
|
||||||
}
|
}
|
||||||
|
|
||||||
if dat.Password != dat.Password2 {
|
if dat.Password != dat.Password2 {
|
||||||
return c.JsonBadRequest(w, "New passwords did not match")
|
return c.JsonBadRequest("New passwords did not match")
|
||||||
}
|
}
|
||||||
|
|
||||||
c.Logger.Warn("test", "dat", dat)
|
c.Logger.Warn("test", "dat", dat)
|
||||||
|
|
||||||
_, login := generateToken(db, c.User.Email, dat.Old_Password)
|
_, login := generateToken(db, c.User.Email, dat.Old_Password)
|
||||||
if !login {
|
if !login {
|
||||||
return c.JsonBadRequest(w, "Password is incorrect");
|
return c.JsonBadRequest("Password is incorrect");
|
||||||
}
|
}
|
||||||
|
|
||||||
salt := generateSalt()
|
salt := generateSalt()
|
||||||
@ -486,7 +486,7 @@ func usersEndpints(db *sql.DB, handle *Handle) {
|
|||||||
return c.Error500(err)
|
return c.Error500(err)
|
||||||
}
|
}
|
||||||
|
|
||||||
return c.SendJSON(w, c.User.Id)
|
return c.SendJSON(c.User.Id)
|
||||||
}
|
}
|
||||||
|
|
||||||
r.ParseForm()
|
r.ParseForm()
|
||||||
|
57
webpage/src/lib/Tabs.svelte
Normal file
57
webpage/src/lib/Tabs.svelte
Normal file
@ -0,0 +1,57 @@
|
|||||||
|
<script lang="ts">
|
||||||
|
let {active} = $props<{active?: string}>();
|
||||||
|
|
||||||
|
function setActive(name: string) {
|
||||||
|
return () => active = name;
|
||||||
|
}
|
||||||
|
|
||||||
|
function isActive(name: string) {
|
||||||
|
return name == active;
|
||||||
|
}
|
||||||
|
</script>
|
||||||
|
|
||||||
|
<div class="tabs">
|
||||||
|
<div class="tab-buttons">
|
||||||
|
<slot name="buttons" {setActive} {isActive} />
|
||||||
|
</div>
|
||||||
|
<slot {isActive} />
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<style lang="scss">
|
||||||
|
.tabs {
|
||||||
|
border-radius: 5px;
|
||||||
|
overflow: hidden;
|
||||||
|
display: flex;
|
||||||
|
flex-wrap: wrap;
|
||||||
|
box-shadow: 0 2px 8px 1px #66666655;
|
||||||
|
gap: 0 5px;
|
||||||
|
|
||||||
|
.tab-buttons {
|
||||||
|
display: flex;
|
||||||
|
overflow-x: scroll;
|
||||||
|
|
||||||
|
:global(.tab) {
|
||||||
|
padding: 5px;
|
||||||
|
background: var(--light-grey);
|
||||||
|
border-radius: 5px 5px 0 0;
|
||||||
|
box-shadow: none;
|
||||||
|
font-size: 1.1rem;
|
||||||
|
}
|
||||||
|
|
||||||
|
:global(.tab.selected) {
|
||||||
|
box-shadow: inset 0 2px 8px 1px #66666655;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
:global(.content) {
|
||||||
|
display: none;
|
||||||
|
padding: 5px;
|
||||||
|
width: 100%;
|
||||||
|
}
|
||||||
|
|
||||||
|
:global(.content.selected) {
|
||||||
|
display: block;
|
||||||
|
box-shadow: 0 2px 2px 1px #66666655;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
</style>
|
@ -1,10 +1,12 @@
|
|||||||
|
import { goto } from '$app/navigation';
|
||||||
import { userStore } from 'routes/UserStore.svelte';
|
import { userStore } from 'routes/UserStore.svelte';
|
||||||
|
|
||||||
const API = "http://localhost:8000";
|
const API = "/api";
|
||||||
|
|
||||||
export async function get(url: string) {
|
export async function get(url: string) {
|
||||||
const headers = new Headers();
|
const headers = new Headers();
|
||||||
//headers.append('content-type', 'application/json');
|
//headers.append('content-type', 'application/json');
|
||||||
|
headers.append('response-type', 'application/json');
|
||||||
if (userStore.user) {
|
if (userStore.user) {
|
||||||
headers.append('token', userStore.user.token);
|
headers.append('token', userStore.user.token);
|
||||||
}
|
}
|
||||||
@ -40,3 +42,50 @@ export async function post(url: string, body: any) {
|
|||||||
|
|
||||||
return r.json();
|
return r.json();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
export async function rdelete(url: string, body: any) {
|
||||||
|
const headers = new Headers();
|
||||||
|
headers.append('content-type', 'application/json');
|
||||||
|
if (userStore.user) {
|
||||||
|
headers.append('token', userStore.user.token);
|
||||||
|
}
|
||||||
|
|
||||||
|
let r = await fetch(`${API}/${url}`, {
|
||||||
|
method: 'DELETE',
|
||||||
|
headers: headers,
|
||||||
|
body: JSON.stringify(body),
|
||||||
|
});
|
||||||
|
|
||||||
|
if (r.status !== 200) {
|
||||||
|
throw r;
|
||||||
|
}
|
||||||
|
|
||||||
|
return r.json();
|
||||||
|
}
|
||||||
|
|
||||||
|
export async function postFormData(url: string, body: FormData) {
|
||||||
|
const headers = new Headers();
|
||||||
|
//headers.append('content-type', 'multipart/form-data');
|
||||||
|
headers.append('response-type', 'application/json');
|
||||||
|
if (userStore.user) {
|
||||||
|
headers.append('token', userStore.user.token);
|
||||||
|
}
|
||||||
|
|
||||||
|
let r = await fetch(`${API}/${url}`, {
|
||||||
|
method: 'POST',
|
||||||
|
headers: headers,
|
||||||
|
body: body,
|
||||||
|
});
|
||||||
|
|
||||||
|
if (r.status == 401) {
|
||||||
|
userStore.user = undefined;
|
||||||
|
goto('/login');
|
||||||
|
throw new Error("Redirect");
|
||||||
|
}
|
||||||
|
|
||||||
|
if (r.status !== 200) {
|
||||||
|
throw r;
|
||||||
|
}
|
||||||
|
|
||||||
|
return r.json();
|
||||||
|
}
|
||||||
|
@ -1,8 +1,26 @@
|
|||||||
<script lang="ts">
|
<script lang="ts">
|
||||||
|
import MessageSimple from "src/lib/MessageSimple.svelte";
|
||||||
|
import { onMount } from "svelte";
|
||||||
|
import { get } from '$lib/requests.svelte'
|
||||||
|
|
||||||
let list = $state<{
|
let list = $state<{
|
||||||
name: string,
|
name: string,
|
||||||
id: string,
|
id: string,
|
||||||
}[]>([]);
|
}[]>([]);
|
||||||
|
|
||||||
|
let message: MessageSimple;
|
||||||
|
|
||||||
|
onMount(async () => {
|
||||||
|
try {
|
||||||
|
list = await get("models");
|
||||||
|
} catch (e) {
|
||||||
|
if (e instanceof Response) {
|
||||||
|
message.display(await e.json())
|
||||||
|
} else {
|
||||||
|
message.display("Could not request list of models");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
});
|
||||||
</script>
|
</script>
|
||||||
|
|
||||||
<svelte:head>
|
<svelte:head>
|
||||||
@ -12,6 +30,7 @@
|
|||||||
</svelte:head>
|
</svelte:head>
|
||||||
|
|
||||||
<main>
|
<main>
|
||||||
|
<MessageSimple bind:this={message} />
|
||||||
{#if list.length > 0}
|
{#if list.length > 0}
|
||||||
<div class="list-header">
|
<div class="list-header">
|
||||||
<h2>My Models</h2>
|
<h2>My Models</h2>
|
||||||
@ -57,3 +76,29 @@
|
|||||||
</div>
|
</div>
|
||||||
{/if}
|
{/if}
|
||||||
</main>
|
</main>
|
||||||
|
|
||||||
|
<style lang="scss">
|
||||||
|
main {
|
||||||
|
padding: 20px 15vw;
|
||||||
|
}
|
||||||
|
.list-header {
|
||||||
|
display: flex;
|
||||||
|
padding-bottom: 10px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.list-header h2 {
|
||||||
|
margin: 0;
|
||||||
|
padding: 10px 5px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.list-header .expand {
|
||||||
|
flex-grow: 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
.list-header .button,
|
||||||
|
.list-header button {
|
||||||
|
padding: 10px 10px;
|
||||||
|
height: calc(100% - 20px);
|
||||||
|
margin-top: 5px;
|
||||||
|
}
|
||||||
|
</style>
|
||||||
|
@ -1,6 +1,8 @@
|
|||||||
<script lang="ts">
|
<script lang="ts">
|
||||||
import FileUpload from "src/lib/FileUpload.svelte";
|
import FileUpload from "src/lib/FileUpload.svelte";
|
||||||
import MessageSimple from "src/lib/MessageSimple.svelte";
|
import MessageSimple from "src/lib/MessageSimple.svelte";
|
||||||
|
import { postFormData } from "src/lib/requests.svelte";
|
||||||
|
import {goto} from '$app/navigation';
|
||||||
|
|
||||||
import "src/styles/forms.css";
|
import "src/styles/forms.css";
|
||||||
|
|
||||||
@ -18,9 +20,28 @@
|
|||||||
file: undefined,
|
file: undefined,
|
||||||
});
|
});
|
||||||
|
|
||||||
function onSubmit() {
|
async function onSubmit() {
|
||||||
message.display("");
|
message.display("");
|
||||||
buttonClicked = new Promise<void>(() => {});
|
buttonClicked = new Promise<void>(() => {});
|
||||||
|
|
||||||
|
if (!data.file || !data.name) return;
|
||||||
|
|
||||||
|
let formData = new FormData();
|
||||||
|
formData.append("name", data.name);
|
||||||
|
formData.append("file", data.file, 'base-image.png')
|
||||||
|
|
||||||
|
try {
|
||||||
|
let id = await postFormData('models/add', formData);
|
||||||
|
goto(`/models/edit?id=${id}`)
|
||||||
|
} catch (e) {
|
||||||
|
if (e instanceof Response) {
|
||||||
|
message.display(await e.json())
|
||||||
|
} else {
|
||||||
|
message.display("Was not able to create model")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
buttonClicked = Promise.resolve();
|
||||||
}
|
}
|
||||||
</script>
|
</script>
|
||||||
|
|
||||||
|
525
webpage/src/routes/models/edit/+page.svelte
Normal file
525
webpage/src/routes/models/edit/+page.svelte
Normal file
@ -0,0 +1,525 @@
|
|||||||
|
<script lang="ts" context="module">
|
||||||
|
export type Model = {
|
||||||
|
id: string;
|
||||||
|
name: string;
|
||||||
|
color_mode: string;
|
||||||
|
width: number;
|
||||||
|
height: number;
|
||||||
|
status: number;
|
||||||
|
}
|
||||||
|
|
||||||
|
export type Layer = {
|
||||||
|
layer_type: number;
|
||||||
|
shape: string;
|
||||||
|
}
|
||||||
|
|
||||||
|
export type Definitions = {
|
||||||
|
epoch: number;
|
||||||
|
epoch_progress: number;
|
||||||
|
status: number;
|
||||||
|
accuracy: number;
|
||||||
|
layers?: Layer[];
|
||||||
|
}
|
||||||
|
</script>
|
||||||
|
|
||||||
|
<script lang="ts">
|
||||||
|
import { onMount } from "svelte";
|
||||||
|
|
||||||
|
import BaseModelInfo from "./BaseModelInfo.svelte";
|
||||||
|
import DeleteModel from "./DeleteModel.svelte";
|
||||||
|
import { goto } from "$app/navigation";
|
||||||
|
import { get } from "src/lib/requests.svelte";
|
||||||
|
|
||||||
|
import 'src/styles/forms.css'
|
||||||
|
import ModelData from "./ModelData.svelte";
|
||||||
|
import DeleteZip from "./DeleteZip.svelte";
|
||||||
|
|
||||||
|
let model: Promise<Model> = $state(new Promise(() => {}));
|
||||||
|
let definitions: Promise<Definitions[]> = $state(new Promise(() => {}));
|
||||||
|
|
||||||
|
let id: string | undefined = $state()
|
||||||
|
|
||||||
|
async function getModel() {
|
||||||
|
try {
|
||||||
|
let temp_model: Model = await get(`models/edit?id=${id}`);
|
||||||
|
|
||||||
|
if (temp_model.status == 3) {
|
||||||
|
setTimeout(getModel, 2000);
|
||||||
|
}
|
||||||
|
|
||||||
|
model = Promise.resolve(temp_model);
|
||||||
|
} catch (e) {
|
||||||
|
if (e instanceof Response) {
|
||||||
|
model = Promise.reject(await e.json())
|
||||||
|
} else {
|
||||||
|
model = Promise.reject("Could not load model");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
onMount(() => {
|
||||||
|
let url = new URLSearchParams(window.location.search);
|
||||||
|
const _id = url.get('id');
|
||||||
|
if (!_id) {
|
||||||
|
goto('/models')
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
id = _id;
|
||||||
|
getModel();
|
||||||
|
});
|
||||||
|
|
||||||
|
async function resetModel() {
|
||||||
|
throw new Error("TODO");
|
||||||
|
}
|
||||||
|
|
||||||
|
// Auto reload after 2s when model.status 3,4
|
||||||
|
</script>
|
||||||
|
|
||||||
|
<svelte:head>
|
||||||
|
{#await model}
|
||||||
|
<title>
|
||||||
|
Model
|
||||||
|
</title>
|
||||||
|
{:then m}
|
||||||
|
{#if m}
|
||||||
|
<title>
|
||||||
|
Model: {m.name}
|
||||||
|
</title>
|
||||||
|
{:else}
|
||||||
|
<title>
|
||||||
|
Model
|
||||||
|
</title>
|
||||||
|
{/if}
|
||||||
|
{/await}
|
||||||
|
</svelte:head>
|
||||||
|
|
||||||
|
<!-- {{/* Is called from a diffrent endpoint so that it does not matter where this is from :) which means that . can mean what ever I want */}}
|
||||||
|
{{ define "data-model-create-class-table-table" }}
|
||||||
|
<div>
|
||||||
|
<table>
|
||||||
|
<thead>
|
||||||
|
<tr>
|
||||||
|
<th>
|
||||||
|
File Path
|
||||||
|
</th>
|
||||||
|
<th>
|
||||||
|
Mode
|
||||||
|
</th>
|
||||||
|
<th>
|
||||||
|
<!-- Img -- >
|
||||||
|
</th>
|
||||||
|
<th>
|
||||||
|
<!-- Status -- >
|
||||||
|
</th>
|
||||||
|
</tr>
|
||||||
|
</thead>
|
||||||
|
<tbody>
|
||||||
|
{{range .List}}
|
||||||
|
<tr>
|
||||||
|
<td>
|
||||||
|
{{ if eq .FilePath "id://" }}
|
||||||
|
Managed
|
||||||
|
{{ else }}
|
||||||
|
{{.FilePath}}
|
||||||
|
{{ end }}
|
||||||
|
</td>
|
||||||
|
<td>
|
||||||
|
{{ if (eq .Mode 2) }}
|
||||||
|
Testing
|
||||||
|
{{ else }}
|
||||||
|
Training
|
||||||
|
{{ end }}
|
||||||
|
</td>
|
||||||
|
<td class="text-center">
|
||||||
|
{{ if startsWith .FilePath "id://" }}
|
||||||
|
<img src="/savedData/{{ $.Model.Id }}/data/{{ .Id }}.{{ $.Model.Format }}" height="30px" width="30px" style="object-fit: contain;" />
|
||||||
|
{{ else }}
|
||||||
|
TODO
|
||||||
|
img {{ .FilePath }}
|
||||||
|
{{ end }}
|
||||||
|
</td>
|
||||||
|
<td class="text-center">
|
||||||
|
{{ if eq .Status 1 }}
|
||||||
|
<span class="bi bi-check-circle-fill" style="color: green"></span>
|
||||||
|
{{ else }}
|
||||||
|
<span class="bi bi-exclamation-triangle-fill" style="color: red"></span>
|
||||||
|
{{ end }}
|
||||||
|
</td>
|
||||||
|
</tr>
|
||||||
|
{{end}}
|
||||||
|
</tbody>
|
||||||
|
</table>
|
||||||
|
<div class="flex justify-center align-center">
|
||||||
|
<div class="grow-1 flex justify-end align-center ">
|
||||||
|
{{ if gt .Page 0 }}
|
||||||
|
<button
|
||||||
|
hx-get="/models/data/list?id={{ .Id }}&page={{ add .Page -1 }}"
|
||||||
|
hx-target=".content[data-tab='{{ .Name }}']"
|
||||||
|
hx-swap="innerHTML"
|
||||||
|
hx-headers='{"REQUEST-TYPE": "html"}'
|
||||||
|
data-tab="{{ .Name }}">
|
||||||
|
Prev
|
||||||
|
</button>
|
||||||
|
{{ end }}
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div style="padding: 10px;">
|
||||||
|
{{ .Page }}
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div class="grow-1 flex justify-start align-center">
|
||||||
|
{{ if .ShowNext }}
|
||||||
|
<button
|
||||||
|
hx-get="/models/data/list?id={{ .Id }}&page={{ add .Page 1 }}"
|
||||||
|
hx-target=".content[data-tab='{{ .Name }}']"
|
||||||
|
hx-swap="innerHTML"
|
||||||
|
hx-headers='{"REQUEST-TYPE": "html"}'
|
||||||
|
data-tab="{{ .Name }}">
|
||||||
|
Next
|
||||||
|
</button>
|
||||||
|
{{ end }}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
{{ end }}
|
||||||
|
|
||||||
|
-->
|
||||||
|
|
||||||
|
<!--
|
||||||
|
{{ define "data-model-create-class-table" }}
|
||||||
|
{{ if eq (len .Classes) 0 }}
|
||||||
|
TODO CREATE TABLE
|
||||||
|
{{else}}
|
||||||
|
<div class="tabs-header">
|
||||||
|
{{/* Handle the case where there are to many buttons */}}
|
||||||
|
<div class="header">
|
||||||
|
{{ range .Classes }}
|
||||||
|
{{/* TODO Auto Load 1st */}}
|
||||||
|
<button
|
||||||
|
hx-get="/models/data/list?id={{ .Id }}"
|
||||||
|
hx-target=".content[data-tab='{{ .Name }}']"
|
||||||
|
hx-swap="innerHTML"
|
||||||
|
hx-headers='{"REQUEST-TYPE": "html"}'
|
||||||
|
hx-trigger="click"
|
||||||
|
class="tab"
|
||||||
|
data-tab="{{ .Name }}">
|
||||||
|
{{ .Name }}
|
||||||
|
</button>
|
||||||
|
{{ end }}
|
||||||
|
</div>
|
||||||
|
{{ range $i, $a := .Classes }}
|
||||||
|
{{ if eq $i 0}}
|
||||||
|
<div
|
||||||
|
hx-get="/models/data/list?id={{ .Id }}"
|
||||||
|
hx-target=".content[data-tab='{{ $a.Name }}']"
|
||||||
|
hx-swap="innerHTML"
|
||||||
|
hx-headers='{"REQUEST-TYPE": "html"}'
|
||||||
|
hx-trigger="load"
|
||||||
|
class="content"
|
||||||
|
data-tab="{{ $a.Name }}">
|
||||||
|
</div>
|
||||||
|
{{ else }}
|
||||||
|
<div
|
||||||
|
class="content"
|
||||||
|
data-tab="{{ $a.Name }}" >
|
||||||
|
</div>
|
||||||
|
{{ end }}
|
||||||
|
{{ end }}
|
||||||
|
</div>
|
||||||
|
{{end}}
|
||||||
|
{{ end }}
|
||||||
|
-->
|
||||||
|
|
||||||
|
<!--
|
||||||
|
{{ define "train-model-card" }}
|
||||||
|
<form hx-post="/models/train" hx-headers='{"REQUEST-TYPE": "html"}' hx-swap="outerHTML" {{ if .Error }} class="submitted" {{end}} >
|
||||||
|
{{ if .HasData }}
|
||||||
|
{{ if .NumberOfInvalidImages }}
|
||||||
|
{{ if gt .NumberOfInvalidImages 0 }}
|
||||||
|
<p class="danger">
|
||||||
|
There are images {{ .NumberOfInvalidImages }} that were loaded that do not have the correct format. These images will be delete when the model trains.
|
||||||
|
</p>
|
||||||
|
<input type="hidden" value="{{ .NumberOfInvalidImages }}" name="id" />
|
||||||
|
{{ end }}
|
||||||
|
{{ end }}
|
||||||
|
{{ if .ErrorMessage }}
|
||||||
|
<p class="danger">
|
||||||
|
{{ .ErrorMessage }}
|
||||||
|
</p>
|
||||||
|
{{ end }}
|
||||||
|
{{/* TODO expading mode */}}
|
||||||
|
<input type="hidden" value="{{ .Model.Id }}" name="id" />
|
||||||
|
<fieldset>
|
||||||
|
<legend>
|
||||||
|
Model Type
|
||||||
|
</legend>
|
||||||
|
<div class="input-radial">
|
||||||
|
<input id="model_type_simple" value="simple" name="model_type" type="radio" checked />
|
||||||
|
<label for="model_type_simple">Simple</label><br/>
|
||||||
|
<input id="model_type_expandable" value="expandable" name="model_type" type="radio" />
|
||||||
|
<label for="model_type_expandable">Expandable</label>
|
||||||
|
</div>
|
||||||
|
</fieldset>
|
||||||
|
{{/* TODO allow more models to be created */}}
|
||||||
|
<fieldset>
|
||||||
|
<label for="number_of_models">Number of Models</label>
|
||||||
|
<input id="number_of_models" type="number" name="number_of_models" value="1" />
|
||||||
|
</fieldset>
|
||||||
|
{{/* TODO to Change the acc */}}
|
||||||
|
<fieldset>
|
||||||
|
<label for="accuracy">Target accuracy</label>
|
||||||
|
<input id="accuracy" type="number" name="accuracy" value="95" />
|
||||||
|
</fieldset>
|
||||||
|
{{/* TODO allow to chose the base of the model */}}
|
||||||
|
{{/* TODO allow to change the shape of the model */}}
|
||||||
|
<button>
|
||||||
|
Train
|
||||||
|
</button>
|
||||||
|
{{ else }}
|
||||||
|
<h2>
|
||||||
|
To train the model please provide data to the model first
|
||||||
|
</h2>
|
||||||
|
{{ end }}
|
||||||
|
</form>
|
||||||
|
{{ end }}
|
||||||
|
|
||||||
|
-->
|
||||||
|
|
||||||
|
<!--
|
||||||
|
{{ define "run-model-card" }}
|
||||||
|
<form hx-headers='{"REQUEST-TYPE": "html"}' enctype="multipart/form-data" hx-post="/models/run" hx-swap="outerHTML">
|
||||||
|
<input type="hidden" name="id" value={{.Model.Id}} />
|
||||||
|
<fieldset class="file-upload" >
|
||||||
|
<label for="file">Image</label>
|
||||||
|
<div class="form-msg">
|
||||||
|
Run image through them model and get the result
|
||||||
|
</div>
|
||||||
|
<div class="icon-holder">
|
||||||
|
<button class="icon">
|
||||||
|
<img replace="icon" src="/imgs/upload-icon.png" />
|
||||||
|
<span replace="File Selected">
|
||||||
|
Image File
|
||||||
|
</span>
|
||||||
|
</button>
|
||||||
|
{{ if .ImageError }}
|
||||||
|
<span class="form-msg error">
|
||||||
|
The provided image was not valid for this model
|
||||||
|
</span>
|
||||||
|
{{ end }}
|
||||||
|
<input id="file" name="file" type="file" required accept="image/png" />
|
||||||
|
</div>
|
||||||
|
</fieldset>
|
||||||
|
<button>
|
||||||
|
Run
|
||||||
|
</button>
|
||||||
|
{{ if .NotFound }}
|
||||||
|
<div class="result">
|
||||||
|
<h1>
|
||||||
|
The class was not found
|
||||||
|
</h1>
|
||||||
|
</div>
|
||||||
|
{{ else if .Result }}
|
||||||
|
<div>
|
||||||
|
<h1>
|
||||||
|
Result
|
||||||
|
</h1>
|
||||||
|
The image was classified as {{.Result}}
|
||||||
|
</div>
|
||||||
|
{{ end }}
|
||||||
|
</form>
|
||||||
|
{{ end }}
|
||||||
|
|
||||||
|
-->
|
||||||
|
|
||||||
|
<main>
|
||||||
|
{#await model}
|
||||||
|
Loading
|
||||||
|
{:then m}
|
||||||
|
{#if m.status == 1}
|
||||||
|
<div>
|
||||||
|
<h1 class="text-center">
|
||||||
|
{ m.name }
|
||||||
|
</h1>
|
||||||
|
<!-- TODO add cool animation -->
|
||||||
|
<h2 class="text-center">
|
||||||
|
Preparing the model
|
||||||
|
</h2>
|
||||||
|
</div>
|
||||||
|
{:else if m.status == -1}
|
||||||
|
<div>
|
||||||
|
<h1 class="text-center">
|
||||||
|
{m.name}
|
||||||
|
</h1>
|
||||||
|
<!-- TODO improve message -->
|
||||||
|
<h2 class="text-center">
|
||||||
|
Failed to prepare model
|
||||||
|
</h2>
|
||||||
|
|
||||||
|
<div>
|
||||||
|
TODO button delete
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<!--form hx-delete="/models/delete">
|
||||||
|
<input type="hidden" name="id" value="{{ .Model.Id }}" />
|
||||||
|
<button class="danger">
|
||||||
|
Delete
|
||||||
|
</button>
|
||||||
|
</form-->
|
||||||
|
</div>
|
||||||
|
<!-- PRE TRAINING STATUS -->
|
||||||
|
{:else if m.status == 2 }
|
||||||
|
<BaseModelInfo model={m} />
|
||||||
|
<ModelData model={m} on:reload={getModel} />
|
||||||
|
<!-- {{ template "train-model-card" . }} -->
|
||||||
|
<DeleteModel model={m} />
|
||||||
|
{:else if m.status == -2 }
|
||||||
|
<BaseModelInfo model={m} />
|
||||||
|
<DeleteZip model={m} on:reload={getModel} />
|
||||||
|
<DeleteModel model={m} />
|
||||||
|
{:else if m.status == 3 }
|
||||||
|
<BaseModelInfo model={m} />
|
||||||
|
<div class="card">
|
||||||
|
<!-- TODO improve this -->
|
||||||
|
Processing zip file...
|
||||||
|
</div>
|
||||||
|
{:else if m.status == -3 || m.status == -4}
|
||||||
|
<BaseModelInfo model={m} />
|
||||||
|
<form on:submit={resetModel}>
|
||||||
|
Failed Prepare for training.<br/>
|
||||||
|
<div class="spacer" ></div>
|
||||||
|
<button class="danger">
|
||||||
|
Try Again
|
||||||
|
<button>
|
||||||
|
</form>
|
||||||
|
<DeleteModel model={m} />
|
||||||
|
{:else if m.status == 4}
|
||||||
|
<BaseModelInfo model={m} />
|
||||||
|
<!-- TODO request /models/edit?id={m.id} -->
|
||||||
|
<div class="card">
|
||||||
|
<!-- TODO improve this -->
|
||||||
|
Training the model...<br/>
|
||||||
|
<!-- TODO Add progress status on definitions -->
|
||||||
|
{#await definitions}
|
||||||
|
Loading
|
||||||
|
{:then defs}
|
||||||
|
<table>
|
||||||
|
<thead>
|
||||||
|
<tr>
|
||||||
|
<th>
|
||||||
|
Done Progress
|
||||||
|
</th>
|
||||||
|
<th>
|
||||||
|
Training Round Progress
|
||||||
|
</th>
|
||||||
|
<th>
|
||||||
|
Accuracy
|
||||||
|
</th>
|
||||||
|
<th>
|
||||||
|
Status
|
||||||
|
</th>
|
||||||
|
</tr>
|
||||||
|
</thead>
|
||||||
|
<tbody>
|
||||||
|
{#each defs as def}
|
||||||
|
<tr>
|
||||||
|
<td>
|
||||||
|
{def.epoch}
|
||||||
|
</td>
|
||||||
|
<td>
|
||||||
|
{def.epoch_progress}/20
|
||||||
|
</td>
|
||||||
|
<td>
|
||||||
|
{def.accuracy}%
|
||||||
|
</td>
|
||||||
|
<td style="text-align: center;">
|
||||||
|
{#if def.status == 2}
|
||||||
|
<span class="bi bi-book" style="color: green;"></span>
|
||||||
|
{:else if [3,6,-3].includes(def.status) }
|
||||||
|
<span class="bi bi-book-half" style="color: {{
|
||||||
|
'3': 'green',
|
||||||
|
'-3': 'red',
|
||||||
|
'6': 'orange',
|
||||||
|
}[String(def.status)]};"></span>
|
||||||
|
{:else}
|
||||||
|
{def.status}
|
||||||
|
{/if}
|
||||||
|
</td>
|
||||||
|
</tr>
|
||||||
|
{#if def.status == 3 && def.layers}
|
||||||
|
<tr>
|
||||||
|
<td colspan="4">
|
||||||
|
<svg viewBox="0 200 1000 600">
|
||||||
|
{#each def.layers as layer, i}
|
||||||
|
{@const sep_mod = def.layers.length > 8 ? Math.max(10, 100 - (def.layers.length - 8) * 10) : 100}
|
||||||
|
{#if layer.layer_type == 1}
|
||||||
|
<polygon
|
||||||
|
points="50,450 200,250 200,550 50,750"
|
||||||
|
stroke="black"
|
||||||
|
stroke-width="2"
|
||||||
|
fill="green"></polygon>
|
||||||
|
{:else if layer.layer_type == 4}
|
||||||
|
<polygon
|
||||||
|
points="{50 + (i * sep_mod)},450 {200 + (i * sep_mod)},250 {200 + (i * sep_mod)},550 {50 + (i * sep_mod)},750"
|
||||||
|
stroke="black"
|
||||||
|
stroke-width="2"
|
||||||
|
fill="orange">
|
||||||
|
</polygon>
|
||||||
|
{:else if layer.layer_type == 3}
|
||||||
|
<polygon
|
||||||
|
points="{50 + (i * sep_mod)},450 {200 + (i * sep_mod)},250 {200 + (i * sep_mod)}},550 {50 + (i * sep_mod)},750"
|
||||||
|
stroke="black"
|
||||||
|
stroke-width="2"
|
||||||
|
fill="red">
|
||||||
|
</polygon>
|
||||||
|
{:else if layer.layer_type == 3}
|
||||||
|
<polygon
|
||||||
|
points="{50 + (i * sep_mod)},550 {200 + (i * sep_mod)},350 {200 + (i * sep_mod)},450 {50 + (i * sep_mod)},650"
|
||||||
|
stroke="black"
|
||||||
|
stroke-width="2"
|
||||||
|
fill="blue">
|
||||||
|
</polygon>
|
||||||
|
{:else}
|
||||||
|
<div>
|
||||||
|
{layer.layer_type}
|
||||||
|
{layer.shape}
|
||||||
|
</div>
|
||||||
|
{/if}
|
||||||
|
{/each}
|
||||||
|
</svg>
|
||||||
|
</td>
|
||||||
|
</tr>
|
||||||
|
{/if}
|
||||||
|
{/each}
|
||||||
|
</tbody>
|
||||||
|
</table>
|
||||||
|
{/await}
|
||||||
|
{{/* TODO Add ability to stop training */}}
|
||||||
|
</div>
|
||||||
|
{:else if m.status == 5}
|
||||||
|
<BaseModelInfo model={m} />
|
||||||
|
TODO run model
|
||||||
|
<!--
|
||||||
|
<form hx-delete="/models/train/reset" hx-headers='{"REQUEST-TYPE": "html"}' hx-swap="outerHTML">
|
||||||
|
Failed Prepare for training.<br/>
|
||||||
|
<div class="spacer" ></div>
|
||||||
|
<input type="hidden" name="id" value="{{ .Model.Id }}" />
|
||||||
|
<button class="danger">
|
||||||
|
Try Again
|
||||||
|
</button>
|
||||||
|
</form>
|
||||||
|
-->
|
||||||
|
<DeleteModel model={m} />
|
||||||
|
{:else}
|
||||||
|
<h1>
|
||||||
|
Unknown Status of the model.
|
||||||
|
</h1>
|
||||||
|
{/if}
|
||||||
|
{/await}
|
||||||
|
</main>
|
||||||
|
|
||||||
|
<style lang="scss">
|
||||||
|
main {
|
||||||
|
padding: 20px 15vw;
|
||||||
|
}
|
||||||
|
</style>
|
43
webpage/src/routes/models/edit/BaseModelInfo.svelte
Normal file
43
webpage/src/routes/models/edit/BaseModelInfo.svelte
Normal file
@ -0,0 +1,43 @@
|
|||||||
|
<script lang="ts">
|
||||||
|
import type { Model } from './+page.svelte';
|
||||||
|
let { model } = $props<{ model: Model }>();
|
||||||
|
</script>
|
||||||
|
|
||||||
|
<div class="card model-card">
|
||||||
|
<h1>
|
||||||
|
{model.name}
|
||||||
|
</h1>
|
||||||
|
<div class="second-line">
|
||||||
|
<img src="/api/savedData/{model.id}/baseimage.png" alt="" />
|
||||||
|
<div class="info">
|
||||||
|
<div>
|
||||||
|
<span class="bold bigger">Image Type:</span>
|
||||||
|
{model.color_mode}
|
||||||
|
</div>
|
||||||
|
<div>
|
||||||
|
<span class="bold bigger">Image Size:</span>
|
||||||
|
{model.width}x{model.height}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<style>
|
||||||
|
.model-card {
|
||||||
|
h1 {
|
||||||
|
margin: 0;
|
||||||
|
padding-bottom: 10px;
|
||||||
|
}
|
||||||
|
|
||||||
|
img {
|
||||||
|
width: 25%;
|
||||||
|
height: 100%;
|
||||||
|
object-fit: contain;
|
||||||
|
}
|
||||||
|
|
||||||
|
.second-line {
|
||||||
|
display: flex;
|
||||||
|
gap: 20px;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
</style>
|
31
webpage/src/routes/models/edit/DeleteModel.svelte
Normal file
31
webpage/src/routes/models/edit/DeleteModel.svelte
Normal file
@ -0,0 +1,31 @@
|
|||||||
|
<script lang="ts">
|
||||||
|
import type { Model } from './+page.svelte';
|
||||||
|
|
||||||
|
let {model}: { model: Model } = $props();
|
||||||
|
let name: string = $state("");
|
||||||
|
let submmited: boolean = $state(false);
|
||||||
|
let nameDoesNotMatch: string = $state("");
|
||||||
|
|
||||||
|
function deleteModel() {
|
||||||
|
submmited = true;
|
||||||
|
nameDoesNotMatch = "";
|
||||||
|
console.error("TODO")
|
||||||
|
}
|
||||||
|
</script>
|
||||||
|
|
||||||
|
<form on:submit|preventDefault={deleteModel} class:submmited>
|
||||||
|
<fieldset>
|
||||||
|
<label for="name">
|
||||||
|
To delete this model please type "{model.name}":
|
||||||
|
</label>
|
||||||
|
<input name="name" id="name" required bind:value={name} />
|
||||||
|
{#if nameDoesNotMatch }
|
||||||
|
<span class="form-msg red">
|
||||||
|
Name does not match "{model.name}"
|
||||||
|
</span>
|
||||||
|
{/if}
|
||||||
|
</fieldset>
|
||||||
|
<button class="danger">
|
||||||
|
Delete
|
||||||
|
</button>
|
||||||
|
</form>
|
39
webpage/src/routes/models/edit/DeleteZip.svelte
Normal file
39
webpage/src/routes/models/edit/DeleteZip.svelte
Normal file
@ -0,0 +1,39 @@
|
|||||||
|
<script lang="ts">
|
||||||
|
import { rdelete } from "src/lib/requests.svelte";
|
||||||
|
import type { Model } from "./+page.svelte";
|
||||||
|
import MessageSimple from "src/lib/MessageSimple.svelte";
|
||||||
|
import { createEventDispatcher } from "svelte";
|
||||||
|
|
||||||
|
let message: MessageSimple;
|
||||||
|
|
||||||
|
let { model } = $props<{model: Model}>();
|
||||||
|
|
||||||
|
const dispatch = createEventDispatcher<{reload: void}>();
|
||||||
|
|
||||||
|
async function deleteZip() {
|
||||||
|
message.display("");
|
||||||
|
|
||||||
|
try {
|
||||||
|
await rdelete("models/data/delete-zip-file", { id: model.id });
|
||||||
|
dispatch('reload');
|
||||||
|
} catch (e) {
|
||||||
|
if (e instanceof Response) {
|
||||||
|
message.display(await e.json());
|
||||||
|
} else {
|
||||||
|
message.display("Could not delete the zip file");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
</script>
|
||||||
|
|
||||||
|
|
||||||
|
<form on:submit|preventDefault={deleteZip}>
|
||||||
|
Failed to proccess the zip file.<br/>
|
||||||
|
Delete file and proccess again.<br/>
|
||||||
|
<br/>
|
||||||
|
<div class="spacer" ></div>
|
||||||
|
<MessageSimple bind:this={message} />
|
||||||
|
<button class="danger">
|
||||||
|
Delete Zip File
|
||||||
|
</button>
|
||||||
|
</form>
|
187
webpage/src/routes/models/edit/ModelData.svelte
Normal file
187
webpage/src/routes/models/edit/ModelData.svelte
Normal file
@ -0,0 +1,187 @@
|
|||||||
|
<script lang="ts" context="module">
|
||||||
|
export type Class = {
|
||||||
|
name: string;
|
||||||
|
id: string;
|
||||||
|
}
|
||||||
|
</script>
|
||||||
|
<script lang="ts">
|
||||||
|
import FileUpload from "src/lib/FileUpload.svelte";
|
||||||
|
import Tabs from "src/lib/Tabs.svelte";
|
||||||
|
import type { Model } from "./+page.svelte";
|
||||||
|
import { postFormData, get } from "src/lib/requests.svelte";
|
||||||
|
import MessageSimple from "src/lib/MessageSimple.svelte";
|
||||||
|
import { createEventDispatcher } from "svelte";
|
||||||
|
import ModelTable from "./ModelTable.svelte";
|
||||||
|
|
||||||
|
let { model } = $props<{model: Model}>();
|
||||||
|
|
||||||
|
let classes: Class[] = $state([]);
|
||||||
|
|
||||||
|
let file: File | undefined = $state();
|
||||||
|
|
||||||
|
const dispatch = createEventDispatcher<{
|
||||||
|
reload: void
|
||||||
|
}>();
|
||||||
|
|
||||||
|
let uploading: Promise<void> = $state(Promise.resolve())
|
||||||
|
let numberOfInvalidImages = $state(0);
|
||||||
|
|
||||||
|
let uploadImage: MessageSimple;
|
||||||
|
|
||||||
|
async function uploadZip() {
|
||||||
|
if (!file) return;
|
||||||
|
|
||||||
|
uploading = new Promise(() => {});
|
||||||
|
|
||||||
|
let form = new FormData();
|
||||||
|
form.append('id', model.id);
|
||||||
|
form.append('file', file, 'upload.zip');
|
||||||
|
|
||||||
|
try {
|
||||||
|
await postFormData('models/data/upload', form);
|
||||||
|
dispatch('reload');
|
||||||
|
} catch (e) {
|
||||||
|
if (e instanceof Response) {
|
||||||
|
uploadImage.display(await e.json());
|
||||||
|
} else {
|
||||||
|
uploadImage.display('');
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
uploading = Promise.resolve();
|
||||||
|
}
|
||||||
|
|
||||||
|
$effect(() => {
|
||||||
|
getData();
|
||||||
|
});
|
||||||
|
|
||||||
|
async function getData() {
|
||||||
|
if (!model) return;
|
||||||
|
try {
|
||||||
|
let data = await get(`models/edit/classes?id=${model.id}`);
|
||||||
|
classes = data.classes
|
||||||
|
numberOfInvalidImages = data.number_of_invalid_images;
|
||||||
|
} catch {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
</script>
|
||||||
|
|
||||||
|
<div class="card">
|
||||||
|
<h3>
|
||||||
|
Training data
|
||||||
|
</h3>
|
||||||
|
{#if classes.length == 0}
|
||||||
|
<p>
|
||||||
|
You need to upload data so the model can train.
|
||||||
|
</p>
|
||||||
|
<Tabs active="upload" let:isActive>
|
||||||
|
<div slot="buttons" let:setActive let:isActive>
|
||||||
|
<button class="tab" class:selected={isActive("upload")} on:click={setActive("upload")}>
|
||||||
|
Upload
|
||||||
|
</button>
|
||||||
|
<button class="tab" class:selected={isActive("create-class")} on:click={setActive("create-class")}>
|
||||||
|
Create Class
|
||||||
|
</button>
|
||||||
|
<button class="tab" class:selected={isActive("api")} on:click={setActive("api")}>
|
||||||
|
Api
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
<div class="content" class:selected={isActive("upload")}>
|
||||||
|
<form on:submit|preventDefault={uploadZip}>
|
||||||
|
<fieldset class="file-upload" >
|
||||||
|
<label for="file">Data file</label>
|
||||||
|
<div class="form-msg">
|
||||||
|
Please provide a file that has the training and testing data<br/>
|
||||||
|
The file must have 2 folders one with testing images and one with training images. <br/>
|
||||||
|
Each of the folders will contain the classes of the model. The folders must be the same in testing and training.
|
||||||
|
The class folders must have the images for the classes.
|
||||||
|
<pre>
|
||||||
|
training\
|
||||||
|
class1\
|
||||||
|
img1.png
|
||||||
|
img2.png
|
||||||
|
img2.png
|
||||||
|
...
|
||||||
|
class2\
|
||||||
|
img1.png
|
||||||
|
img2.png
|
||||||
|
img2.png
|
||||||
|
...
|
||||||
|
...
|
||||||
|
testing\
|
||||||
|
class1\
|
||||||
|
img1.png
|
||||||
|
img2.png
|
||||||
|
img2.png
|
||||||
|
...
|
||||||
|
class2\
|
||||||
|
img1.png
|
||||||
|
img2.png
|
||||||
|
img2.png
|
||||||
|
...
|
||||||
|
...
|
||||||
|
</pre>
|
||||||
|
</div>
|
||||||
|
<FileUpload replace_slot bind:file={file} accept="application/zip" >
|
||||||
|
<img src="/imgs/upload-icon.png" alt="" />
|
||||||
|
<span>
|
||||||
|
Upload Zip File
|
||||||
|
</span>
|
||||||
|
<div slot="replaced">
|
||||||
|
<img src="/imgs/upload-icon.png" alt="" />
|
||||||
|
<span>
|
||||||
|
File selected
|
||||||
|
</span>
|
||||||
|
</div>
|
||||||
|
</FileUpload>
|
||||||
|
</fieldset>
|
||||||
|
<MessageSimple bind:this={uploadImage} />
|
||||||
|
{#await uploading}
|
||||||
|
<button disabled>
|
||||||
|
Uploading
|
||||||
|
</button>
|
||||||
|
{:then}
|
||||||
|
<button>
|
||||||
|
Add
|
||||||
|
</button>
|
||||||
|
{/await}
|
||||||
|
</form>
|
||||||
|
</div>
|
||||||
|
<div class="content" class:selected={isActive("create-class")}>
|
||||||
|
<ModelTable classes={classes} />
|
||||||
|
</div>
|
||||||
|
<div class="content" class:selected={isActive("api")}>
|
||||||
|
TODO
|
||||||
|
</div>
|
||||||
|
</Tabs>
|
||||||
|
<div class="tabs">
|
||||||
|
</div>
|
||||||
|
{:else}
|
||||||
|
<p>
|
||||||
|
You need to upload data so the model can train.
|
||||||
|
</p>
|
||||||
|
{#if numberOfInvalidImages > 0}
|
||||||
|
<p class="danger">
|
||||||
|
There are images {numberOfInvalidImages} that were loaded that do not have the correct format. These images will be delete when the model trains.
|
||||||
|
</p>
|
||||||
|
{/if}
|
||||||
|
<Tabs active="create-class" let:isActive>
|
||||||
|
<div slot="buttons" let:setActive let:isActive>
|
||||||
|
<button class="tab" class:selected={isActive("create-class")} on:click={setActive("create-class")}>
|
||||||
|
Create Class
|
||||||
|
</button>
|
||||||
|
<button class="tab" class:selected={isActive("api")} on:click={setActive("api")}>
|
||||||
|
Api
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
<div class="content" class:selected={isActive("create-class")}>
|
||||||
|
<ModelTable classes={classes} />
|
||||||
|
</div>
|
||||||
|
<div class="content" class:selected={isActive("api")}>
|
||||||
|
TODO
|
||||||
|
</div>
|
||||||
|
</Tabs>
|
||||||
|
{/if}
|
||||||
|
</div>
|
63
webpage/src/routes/models/edit/ModelTable.svelte
Normal file
63
webpage/src/routes/models/edit/ModelTable.svelte
Normal file
@ -0,0 +1,63 @@
|
|||||||
|
<script lang="ts">
|
||||||
|
import Tabs from 'src/lib/Tabs.svelte';
|
||||||
|
import type { Class } from './ModelData.svelte';
|
||||||
|
import { get } from 'src/lib/requests.svelte';
|
||||||
|
|
||||||
|
let selected_class: Class | undefined = $state();
|
||||||
|
|
||||||
|
let { classes } = $props<{ classes: Class[] }>();
|
||||||
|
|
||||||
|
function setActiveClass(c: Class, tb_fn: (name: string) => (() => void)) {
|
||||||
|
selected_class = c;
|
||||||
|
console.log("test", c, classes, c.name)
|
||||||
|
tb_fn(c.name)();
|
||||||
|
}
|
||||||
|
|
||||||
|
$effect(() => {
|
||||||
|
selected_class = classes[0];
|
||||||
|
});
|
||||||
|
|
||||||
|
async function getList() {
|
||||||
|
console.log(selected_class);
|
||||||
|
|
||||||
|
try {
|
||||||
|
|
||||||
|
let url = new URLSearchParams();
|
||||||
|
url.append('id', selected_class?.id ?? '');
|
||||||
|
|
||||||
|
let res = await get('models/data/list?' + url.toString());
|
||||||
|
console.log(res);
|
||||||
|
} catch (e) {
|
||||||
|
console.error("TODO notify user", e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
$effect(() => {
|
||||||
|
if (selected_class) {
|
||||||
|
getList();
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
</script>
|
||||||
|
|
||||||
|
{#if classes.length == 0}
|
||||||
|
TODO CREATE TABLE
|
||||||
|
{:else}
|
||||||
|
<Tabs active={classes[0]?.name} let:isActive>
|
||||||
|
<div slot="buttons" let:setActive let:isActive>
|
||||||
|
<!-- TODO Auto Load 1st -->
|
||||||
|
{#each classes as item}
|
||||||
|
<button
|
||||||
|
on:click={() => setActiveClass(item, setActive)}
|
||||||
|
class="tab"
|
||||||
|
class:selected={isActive(item.name)}
|
||||||
|
>
|
||||||
|
{item.name}
|
||||||
|
</button>
|
||||||
|
{/each}
|
||||||
|
</div>
|
||||||
|
</Tabs>
|
||||||
|
{/if}
|
||||||
|
|
||||||
|
<style lang="scss">
|
||||||
|
</style>
|
@ -90,3 +90,14 @@ a.button {
|
|||||||
.danger {
|
.danger {
|
||||||
color: red;
|
color: red;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
.card {
|
||||||
|
box-shadow: 0 2px 5px 1px #66666655;
|
||||||
|
padding: 20px;
|
||||||
|
border-radius: 10px;
|
||||||
|
margin: 20px 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
.card h3 {
|
||||||
|
margin-top: 0;
|
||||||
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user