80 lines
3.0 KiB
Go
80 lines
3.0 KiB
Go
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"
|
||
}
|