create project

2026-01-16 17:30:40 +08:00
commit effac6b017
157 changed files with 45997 additions and 0 deletions
--- a/domain/models/image_generation.go
+++ b/domain/models/image_generation.go
@@ -0,0 +1,75 @@
+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" // 分镜图片
+)