@umituz/react-native-ai-fal-provider 1.0.35 → 1.0.37

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@umituz/react-native-ai-fal-provider",
-  "version": "1.0.35",
+  "version": "1.0.37",
   "description": "FAL AI provider for React Native - implements IAIProvider interface for unified AI generation",
   "main": "./src/index.ts",
   "types": "./src/index.ts",

package/src/infrastructure/services/fal-provider.ts

@@ -21,6 +21,17 @@ import type { FalQueueStatus } from "../../domain/entities/fal.types";
 import { DEFAULT_FAL_CONFIG, FAL_CAPABILITIES } from "./fal-provider.constants";
 import { mapFalStatusToJobStatus } from "./fal-status-mapper";
 import { FAL_IMAGE_FEATURE_MODELS, FAL_VIDEO_FEATURE_MODELS } from "../../domain/constants/feature-models.constants";
+import {
+  buildSingleImageInput,
+  buildUpscaleInput,
+  buildPhotoRestoreInput,
+  buildFaceSwapInput,
+  buildRemoveBackgroundInput,
+  buildRemoveObjectInput,
+  buildReplaceBackgroundInput,
+  buildImageToImageInput,
+  buildVideoFromImageInput,
+} from "../utils/input-builders.util";
 
 declare const __DEV__: boolean;
 
@@ -176,16 +187,58 @@ export class FalProvider implements IAIProvider {
     return FAL_IMAGE_FEATURE_MODELS[feature];
   }
 
-  buildImageFeatureInput(_feature: ImageFeatureType, _data: ImageFeatureInputData): Record<string, unknown> {
-    throw new Error("Use input builders directly: buildUpscaleInput, buildFaceSwapInput, etc.");
+  buildImageFeatureInput(feature: ImageFeatureType, data: ImageFeatureInputData): Record<string, unknown> {
+    const { imageBase64, targetImageBase64, prompt, options } = data;
+
+    switch (feature) {
+      case "upscale":
+      case "hd-touch-up":
+        return buildUpscaleInput(imageBase64, options);
+
+      case "photo-restore":
+        return buildPhotoRestoreInput(imageBase64, options);
+
+      case "face-swap":
+        if (!targetImageBase64) throw new Error("Face swap requires target image");
+        return buildFaceSwapInput(imageBase64, targetImageBase64, options);
+
+      case "remove-background":
+        return buildRemoveBackgroundInput(imageBase64, options);
+
+      case "remove-object":
+        return buildRemoveObjectInput(imageBase64, options);
+
+      case "replace-background":
+        if (!prompt) throw new Error("Replace background requires prompt");
+        return buildReplaceBackgroundInput(imageBase64, { prompt, ...options });
+
+      case "anime-selfie":
+        return buildImageToImageInput(imageBase64, {
+          prompt: prompt || (options?.prompt as string) || "",
+          negativePrompt: (options?.negativePrompt as string) || "",
+          strength: options?.strength as number,
+          num_inference_steps: options?.num_inference_steps as number,
+          guidance_scale: options?.guidance_scale as number,
+        });
+
+      default:
+        return buildSingleImageInput(imageBase64, options);
+    }
   }
 
   getVideoFeatureModel(feature: VideoFeatureType): string {
     return FAL_VIDEO_FEATURE_MODELS[feature];
   }
 
-  buildVideoFeatureInput(_feature: VideoFeatureType, _data: VideoFeatureInputData): Record<string, unknown> {
-    throw new Error("Use buildVideoFromImageInput directly");
+  buildVideoFeatureInput(_feature: VideoFeatureType, data: VideoFeatureInputData): Record<string, unknown> {
+    const { sourceImageBase64, targetImageBase64, prompt, options } = data;
+
+    return buildVideoFromImageInput(sourceImageBase64, {
+      motion_prompt: prompt,
+      target_image: targetImageBase64,
+      duration: options?.duration as number,
+      ...options,
+    });
   }
 }