Add generative background replace action

cloudinary · Aug 4, 2024 · c31de57 · c31de57
1 parent c530e61
commit c31de57
Show file tree

Hide file tree

Showing 3 changed files with 25 additions and 0 deletions.
diff --git a/url_gen/lib/transformation/effect/effect.dart b/url_gen/lib/transformation/effect/effect.dart
@@ -306,6 +306,10 @@ class Effect extends Action {
     return GenerativeRestore();
   }
 
+  static GenerativeBackgroundReplace generativeBackgroundReplace({String? prompt}) {
+    return GenerativeBackgroundReplace(prompt);
+  }
+
   static BlurFaces blurFaces({dynamic strength}) {
     return BlurFaces(strength: strength);
   }

diff --git a/url_gen/lib/transformation/effect/effect_actions.dart b/url_gen/lib/transformation/effect/effect_actions.dart
@@ -1073,6 +1073,22 @@ class GenerativeRestore extends Effect {
   }
 }
 
+class GenerativeBackgroundReplace extends Effect {
+  String? _prompt;
+
+  GenerativeBackgroundReplace(this._prompt);
+
+  GenerativeBackgroundReplace prompt(String prompt) {
+    _prompt = prompt;
+    return this;
+  }
+
+  @override
+  String toString() {
+    return super.toString().joinWithValues(['gen_background_replace'], actionSeparator: paramKeyValueSeparator).joinWithValues([_prompt != null ? 'prompt_$_prompt' : null]);
+  }
+}
+
 class BlurRegion extends Effect {
 
   dynamic _strength;

diff --git a/url_gen/test/transformation/effect_test.dart b/url_gen/test/transformation/effect_test.dart
@@ -345,6 +345,11 @@ void main() {
     cldAssert('e_gen_restore', Effect.generativeRestore());
   });
 
+  test('Test successful generative background replace effect', () {
+    cldAssert('e_gen_background_replace', Effect.generativeBackgroundReplace());
+    cldAssert('e_gen_background_replace:prompt_dog', Effect.generativeBackgroundReplace(prompt: 'dog'));
+  });
+
   test('Test successful upscale effect', () {
     cldAssert('e_upscale', Effect.upscale());
   });