76 lines
3.4 KiB
Go
76 lines
3.4 KiB
Go
package models
|
|
|
|
import (
|
|
"time"
|
|
|
|
"gorm.io/datatypes"
|
|
)
|
|
|
|
type ImageGeneration struct {
|
|
ID uint `gorm:"primarykey" json:"id"`
|
|
StoryboardID *uint `gorm:"index" json:"storyboard_id,omitempty"`
|
|
DramaID uint `gorm:"not null;index" json:"drama_id"`
|
|
SceneID *uint `gorm:"index" json:"scene_id,omitempty"`
|
|
CharacterID *uint `gorm:"index" json:"character_id,omitempty"`
|
|
ImageType string `gorm:"size:20;index;default:'storyboard'" json:"image_type"`
|
|
FrameType *string `gorm:"size:20" json:"frame_type,omitempty"`
|
|
Provider string `gorm:"size:50;not null" json:"provider"`
|
|
Prompt string `gorm:"type:text;not null" json:"prompt"`
|
|
NegPrompt *string `gorm:"column:negative_prompt;type:text" json:"negative_prompt,omitempty"`
|
|
Model string `gorm:"size:100" json:"model"`
|
|
Size string `gorm:"size:20" json:"size"`
|
|
Quality string `gorm:"size:20" json:"quality"`
|
|
Style *string `gorm:"size:50" json:"style,omitempty"`
|
|
Steps *int `json:"steps,omitempty"`
|
|
CfgScale *float64 `json:"cfg_scale,omitempty"`
|
|
Seed *int64 `json:"seed,omitempty"`
|
|
ImageURL *string `gorm:"type:text" json:"image_url,omitempty"`
|
|
MinioURL *string `gorm:"type:text" json:"minio_url,omitempty"`
|
|
LocalPath *string `gorm:"type:text" json:"local_path,omitempty"`
|
|
Status ImageGenerationStatus `gorm:"size:20;not null;default:'pending'" json:"status"`
|
|
TaskID *string `gorm:"size:200" json:"task_id,omitempty"`
|
|
ErrorMsg *string `gorm:"type:text" json:"error_msg,omitempty"`
|
|
Width *int `json:"width,omitempty"`
|
|
Height *int `json:"height,omitempty"`
|
|
ReferenceImages datatypes.JSON `gorm:"type:json" json:"reference_images,omitempty"`
|
|
CreatedAt time.Time `json:"created_at"`
|
|
UpdatedAt time.Time `json:"updated_at"`
|
|
CompletedAt *time.Time `json:"completed_at,omitempty"`
|
|
|
|
Storyboard *Storyboard `gorm:"foreignKey:StoryboardID" json:"storyboard,omitempty"`
|
|
Drama Drama `gorm:"foreignKey:DramaID" json:"drama,omitempty"`
|
|
Scene *Scene `gorm:"foreignKey:SceneID" json:"scene,omitempty"`
|
|
Character *Character `gorm:"foreignKey:CharacterID" json:"character,omitempty"`
|
|
}
|
|
|
|
func (ImageGeneration) TableName() string {
|
|
return "image_generations"
|
|
}
|
|
|
|
type ImageGenerationStatus string
|
|
|
|
const (
|
|
ImageStatusPending ImageGenerationStatus = "pending"
|
|
ImageStatusProcessing ImageGenerationStatus = "processing"
|
|
ImageStatusCompleted ImageGenerationStatus = "completed"
|
|
ImageStatusFailed ImageGenerationStatus = "failed"
|
|
)
|
|
|
|
type ImageProvider string
|
|
|
|
const (
|
|
ProviderOpenAI ImageProvider = "openai"
|
|
ProviderMidjourney ImageProvider = "midjourney"
|
|
ProviderStableDiffusion ImageProvider = "stable_diffusion"
|
|
ProviderDALLE ImageProvider = "dalle"
|
|
)
|
|
|
|
// ImageType 图片类型
|
|
type ImageType string
|
|
|
|
const (
|
|
ImageTypeCharacter ImageType = "character" // 角色图片
|
|
ImageTypeScene ImageType = "scene" // 场景图片
|
|
ImageTypeStoryboard ImageType = "storyboard" // 分镜图片
|
|
)
|