Add support for upscale effect

cloudinary · May 27, 2024 · de8607c · de8607c
1 parent a88d355
commit de8607c
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 2 deletions.
diff --git a/src/Transformation/Effect/Generative/GenerativeEffect.php b/src/Transformation/Effect/Generative/GenerativeEffect.php
@@ -17,8 +17,9 @@ abstract class GenerativeEffect
 {
     const GENERATIVE_RECOLOR = 'gen_recolor';
     const GENERATIVE_REMOVE  = 'gen_remove';
-    const GENERATIVE_REPLACE  = 'gen_replace';
-    const GENERATIVE_RESTORE  = 'gen_restore';
+    const GENERATIVE_REPLACE = 'gen_replace';
+    const GENERATIVE_RESTORE = 'gen_restore';
+    const UPSCALE            = 'upscale';
 
     use GenerativeEffectTrait;
 }
diff --git a/src/Transformation/Effect/Generative/GenerativeEffectTrait.php b/src/Transformation/Effect/Generative/GenerativeEffectTrait.php
@@ -74,4 +74,15 @@ public static function generativeReplace($fromPrompt, $toPrompt, $preserveGeomet
     {
         return new GenerativeReplace($fromPrompt, $toPrompt, $preserveGeometry, $detectMultiple);
     }
+
+    /**
+     * Uses AI-based prediction to add fine detail while upscaling small images.
+     * This 'super-resolution' feature scales each dimension by four, multiplying the total number of pixels by 16.
+     *
+     * @return GenerativeEffectAction
+     */
+    public static function upscale()
+    {
+        return new GenerativeEffectAction(GenerativeEffect::UPSCALE);
+    }
 }
diff --git a/tests/Unit/Transformation/Image/GenerativeEffectTest.php b/tests/Unit/Transformation/Image/GenerativeEffectTest.php
@@ -94,4 +94,12 @@ public function testGenerativeReplace()
             (string)Effect::generativeReplace('balloon', 'airplane')->preserveGeometry(false)->detectMultiple()
         );
     }
+
+    public function testGenerativeUpscale()
+    {
+        self::assertEquals(
+            'e_upscale',
+            (string)Effect::upscale()
+        );
+    }
 }