init

domain/models/video_generation.go

@@ -0,0 +1,79 @@
+package models
+
+import (
+	"time"
+
+	"gorm.io/gorm"
+)
+
+type VideoGeneration struct {
+	ID        uint           `gorm:"primarykey" json:"id"`
+	CreatedAt time.Time      `json:"created_at"`
+	UpdatedAt time.Time      `json:"updated_at"`
+	DeletedAt gorm.DeletedAt `gorm:"index" json:"-"`
+
+	StoryboardID *uint       `gorm:"index" json:"storyboard_id,omitempty"`
+	Storyboard   *Storyboard `gorm:"foreignKey:StoryboardID" json:"storyboard,omitempty"`
+
+	DramaID uint  `gorm:"not null;index" json:"drama_id"`
+	Drama   Drama `gorm:"foreignKey:DramaID" json:"drama,omitempty"`
+
+	Provider string `gorm:"type:varchar(50);not null;index" json:"provider"`
+	Prompt   string `gorm:"type:text;not null" json:"prompt"`
+	Model    string `gorm:"type:varchar(100)" json:"model,omitempty"`
+
+	ImageGenID *uint           `gorm:"index" json:"image_gen_id,omitempty"`
+	ImageGen   ImageGeneration `gorm:"foreignKey:ImageGenID" json:"image_gen,omitempty"`
+
+	// 参考图模式：single(单图), first_last(首尾帧), multiple(多图), none(无)
+	ReferenceMode *string `gorm:"type:varchar(20)" json:"reference_mode,omitempty"`
+
+	ImageURL           *string `gorm:"type:varchar(1000)" json:"image_url,omitempty"`
+	FirstFrameURL      *string `gorm:"type:varchar(1000)" json:"first_frame_url,omitempty"`
+	LastFrameURL       *string `gorm:"type:varchar(1000)" json:"last_frame_url,omitempty"`
+	ReferenceImageURLs *string `gorm:"type:text" json:"reference_image_urls,omitempty"` // JSON数组存储多张参考图
+
+	Duration     *int    `json:"duration,omitempty"`
+	FPS          *int    `json:"fps,omitempty"`
+	Resolution   *string `gorm:"type:varchar(50)" json:"resolution,omitempty"`
+	AspectRatio  *string `gorm:"type:varchar(20)" json:"aspect_ratio,omitempty"`
+	Style        *string `gorm:"type:varchar(100)" json:"style,omitempty"`
+	MotionLevel  *int    `json:"motion_level,omitempty"`
+	CameraMotion *string `gorm:"type:varchar(100)" json:"camera_motion,omitempty"`
+	Seed         *int64  `json:"seed,omitempty"`
+
+	VideoURL  *string `gorm:"type:varchar(1000)" json:"video_url,omitempty"`
+	MinioURL  *string `gorm:"type:varchar(1000)" json:"minio_url,omitempty"`
+	LocalPath *string `gorm:"type:varchar(500)" json:"local_path,omitempty"`
+
+	Status VideoStatus `gorm:"type:varchar(20);not null;default:'pending';index" json:"status"`
+	TaskID *string     `gorm:"type:varchar(200);index" json:"task_id,omitempty"`
+
+	ErrorMsg    *string    `gorm:"type:text" json:"error_msg,omitempty"`
+	CompletedAt *time.Time `json:"completed_at,omitempty"`
+
+	Width  *int `json:"width,omitempty"`
+	Height *int `json:"height,omitempty"`
+}
+
+type VideoStatus string
+
+const (
+	VideoStatusPending    VideoStatus = "pending"
+	VideoStatusProcessing VideoStatus = "processing"
+	VideoStatusCompleted  VideoStatus = "completed"
+	VideoStatusFailed     VideoStatus = "failed"
+)
+
+type VideoProvider string
+
+const (
+	VideoProviderRunway VideoProvider = "runway"
+	VideoProviderPika   VideoProvider = "pika"
+	VideoProviderDoubao VideoProvider = "doubao"
+	VideoProviderOpenAI VideoProvider = "openai"
+)
+
+func (VideoGeneration) TableName() string {
+	return "video_generations"
+}