Added explore mode option

igolaizola · Jul 6, 2024 · 4ac03f0 · 4ac03f0
1 parent 2739fd1
commit 4ac03f0
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 2 deletions.
diff --git a/pkg/cli/cli.go b/pkg/cli/cli.go
@@ -82,6 +82,7 @@ func newGenerateCommand() *ffcli.Command {
 	fs.BoolVar(&cfg.Watermark, "watermark", false, "add watermark (optional)")
 	fs.IntVar(&cfg.Width, "width", 0, "output video width (optional)")
 	fs.IntVar(&cfg.Height, "height", 0, "output video height (optional)")
+	fs.BoolVar(&cfg.Explore, "explore", false, "explore mode (optional)")
 
 	return &ffcli.Command{
 		Name:       cmd,
@@ -115,6 +116,7 @@ func newExtendCommand() *ffcli.Command {
 	fs.BoolVar(&cfg.Interpolate, "interpolate", true, "interpolate frames (optional)")
 	fs.BoolVar(&cfg.Upscale, "upscale", false, "upscale frames (optional)")
 	fs.BoolVar(&cfg.Watermark, "watermark", false, "add watermark (optional)")
+	fs.BoolVar(&cfg.Explore, "explore", false, "explore mode (optional)")
 
 	return &ffcli.Command{
 		Name:       cmd,

diff --git a/pkg/cmd/extend/extend.go b/pkg/cmd/extend/extend.go
@@ -27,6 +27,7 @@ type Config struct {
 	Interpolate bool
 	Upscale     bool
 	Watermark   bool
+	Explore     bool
 }
 
 // Run generates a video from an image and a text prompt.
@@ -94,6 +95,7 @@ func Run(ctx context.Context, cfg *Config) error {
 			Upscale:     cfg.Upscale,
 			Watermark:   cfg.Watermark,
 			Extend:      false,
+			ExploreMode: cfg.Explore,
 		})
 		if err != nil {
 			return fmt.Errorf("vidai: couldn't generate video: %w", err)

diff --git a/pkg/cmd/generate/generate.go b/pkg/cmd/generate/generate.go
@@ -28,6 +28,7 @@ type Config struct {
 	Watermark   bool
 	Width       int
 	Height      int
+	Explore     bool
 }
 
 // Run generates a video from an image and a text prompt.
@@ -71,6 +72,7 @@ func Run(ctx context.Context, cfg *Config) error {
 		Extend:      false,
 		Width:       cfg.Width,
 		Height:      cfg.Height,
+		ExploreMode: cfg.Explore,
 	})
 	if err != nil {
 		return fmt.Errorf("vidai: couldn't generate video: %w", err)

diff --git a/pkg/runway/runway.go b/pkg/runway/runway.go
@@ -254,6 +254,7 @@ type GenerateRequest struct {
 	Extend      bool
 	Width       int
 	Height      int
+	ExploreMode bool
 }
 
 func (c *Client) Generate(ctx context.Context, cfg *GenerateRequest) (*Generation, error) {
@@ -316,7 +317,7 @@ func (c *Client) Generate(ctx context.Context, cfg *GenerateRequest) (*Generatio
 				},
 				Name:           name,
 				AssetGroupName: "Generative Video",
-				ExploreMode:    false,
+				ExploreMode:    cfg.ExploreMode,
 			},
 			AsTeamID: c.teamID,
 		}
@@ -333,7 +334,7 @@ func (c *Client) Generate(ctx context.Context, cfg *GenerateRequest) (*Generatio
 				Seconds:        10,
 				TextPrompt:     cfg.Prompt,
 				Seed:           seed,
-				ExploreMode:    false,
+				ExploreMode:    cfg.ExploreMode,
 				Watermark:      cfg.Watermark,
 				EnhancePrompt:  true,
 				Width:          width,