@umituz/react-native-ai-generation-content 1.17.0 → 1.17.1

package/src/features/remove-object/presentation/hooks/useRemoveObjectFeature.ts

@@ -0,0 +1,168 @@
+/**
+ * useRemoveObjectFeature Hook
+ * Manages remove object feature state and actions
+ */
+
+import { useState, useCallback } from "react";
+import { executeRemoveObject } from "../../infrastructure/services";
+import type {
+  RemoveObjectFeatureState,
+  RemoveObjectFeatureConfig,
+  RemoveObjectResult,
+} from "../../domain/types";
+
+declare const __DEV__: boolean;
+
+export interface UseRemoveObjectFeatureProps {
+  config: RemoveObjectFeatureConfig;
+  userId: string;
+  onSelectImage: () => Promise<string | null>;
+  onSelectMask?: () => Promise<string | null>;
+  onSaveImage: (imageUrl: string) => Promise<void>;
+}
+
+export interface UseRemoveObjectFeatureReturn extends RemoveObjectFeatureState {
+  selectImage: () => Promise<void>;
+  selectMask: () => Promise<void>;
+  setPrompt: (prompt: string) => void;
+  process: () => Promise<void>;
+  save: () => Promise<void>;
+  reset: () => void;
+}
+
+const initialState: RemoveObjectFeatureState = {
+  imageUri: null,
+  maskUri: null,
+  prompt: "",
+  processedUrl: null,
+  isProcessing: false,
+  progress: 0,
+  error: null,
+};
+
+export function useRemoveObjectFeature(
+  props: UseRemoveObjectFeatureProps,
+): UseRemoveObjectFeatureReturn {
+  const { config, userId, onSelectImage, onSelectMask, onSaveImage } = props;
+  const [state, setState] = useState<RemoveObjectFeatureState>(initialState);
+
+  const selectImage = useCallback(async () => {
+    try {
+      const uri = await onSelectImage();
+      if (uri) {
+        setState((prev) => ({ ...prev, imageUri: uri, error: null }));
+        config.onImageSelect?.(uri);
+      }
+    } catch (error) {
+      const message = error instanceof Error ? error.message : String(error);
+      setState((prev) => ({ ...prev, error: message }));
+    }
+  }, [onSelectImage, config]);
+
+  const selectMask = useCallback(async () => {
+    if (!onSelectMask) return;
+
+    try {
+      const uri = await onSelectMask();
+      if (uri) {
+        setState((prev) => ({ ...prev, maskUri: uri, error: null }));
+        config.onMaskSelect?.(uri);
+      }
+    } catch (error) {
+      const message = error instanceof Error ? error.message : String(error);
+      setState((prev) => ({ ...prev, error: message }));
+    }
+  }, [onSelectMask, config]);
+
+  const setPrompt = useCallback((prompt: string) => {
+    setState((prev) => ({ ...prev, prompt }));
+  }, []);
+
+  const handleProgress = useCallback((progress: number) => {
+    setState((prev) => ({ ...prev, progress }));
+  }, []);
+
+  const process = useCallback(async () => {
+    if (!state.imageUri) return;
+
+    setState((prev) => ({
+      ...prev,
+      isProcessing: true,
+      progress: 0,
+      error: null,
+    }));
+
+    config.onProcessingStart?.();
+
+    if (__DEV__) {
+      // eslint-disable-next-line no-console
+      console.log("[useRemoveObjectFeature] Starting object removal process");
+    }
+
+    const imageBase64 = await config.prepareImage(state.imageUri);
+    const maskBase64 = state.maskUri
+      ? await config.prepareImage(state.maskUri)
+      : undefined;
+
+    const result: RemoveObjectResult = await executeRemoveObject(
+      {
+        imageUri: state.imageUri,
+        imageBase64,
+        maskBase64,
+        prompt: state.prompt || undefined,
+        userId,
+      },
+      {
+        model: config.model,
+        buildInput: config.buildInput,
+        extractResult: config.extractResult,
+        onProgress: handleProgress,
+      },
+    );
+
+    if (result.success && result.imageUrl) {
+      const url = result.imageUrl;
+      setState((prev) => ({
+        ...prev,
+        isProcessing: false,
+        processedUrl: url,
+        progress: 100,
+      }));
+      config.onProcessingComplete?.(result);
+    } else {
+      const errorMessage = result.error || "Processing failed";
+      setState((prev) => ({
+        ...prev,
+        isProcessing: false,
+        error: errorMessage,
+        progress: 0,
+      }));
+      config.onError?.(errorMessage);
+    }
+  }, [state.imageUri, state.maskUri, state.prompt, userId, config, handleProgress]);
+
+  const save = useCallback(async () => {
+    if (!state.processedUrl) return;
+
+    try {
+      await onSaveImage(state.processedUrl);
+    } catch (error) {
+      const message = error instanceof Error ? error.message : String(error);
+      setState((prev) => ({ ...prev, error: message }));
+    }
+  }, [state.processedUrl, onSaveImage]);
+
+  const reset = useCallback(() => {
+    setState(initialState);
+  }, []);
+
+  return {
+    ...state,
+    selectImage,
+    selectMask,
+    setPrompt,
+    process,
+    save,
+    reset,
+  };
+}

package/src/features/remove-object/presentation/index.ts

@@ -0,0 +1,5 @@
+/**
+ * Remove Object Presentation Index
+ */
+
+export * from "./hooks";

package/src/features/upscaling/domain/types/index.ts

@@ -1,2 +1 @@
 export * from "./upscale.types";
-export * from "./provider.types";

package/src/features/upscaling/domain/types/upscale.types.ts

@@ -12,6 +12,7 @@ export interface UpscaleOptions {
 
 export interface UpscaleRequest {
   imageUri: string;
+  imageBase64?: string;
   userId: string;
   options?: UpscaleOptions;
 }
@@ -48,10 +49,23 @@ export interface UpscaleTranslations {
   compareHint?: string;
 }
 
+export type UpscaleInputBuilder = (
+  base64: string,
+  options?: UpscaleOptions,
+) => Record<string, unknown>;
+
+export type UpscaleResultExtractor = (
+  result: unknown,
+) => string | undefined;
+
 export interface UpscaleFeatureConfig {
   providerId?: string;
   defaultScaleFactor?: UpscaleScaleFactor;
   creditCost?: number;
+  model: string;
+  buildInput: UpscaleInputBuilder;
+  extractResult?: UpscaleResultExtractor;
+  prepareImage: (imageUri: string) => Promise<string>;
   onImageSelect?: (uri: string) => void;
   onProcessingStart?: () => void;
   onProcessingComplete?: (result: UpscaleResult) => void;

package/src/features/upscaling/index.ts

@@ -12,17 +12,12 @@ export type {
   UpscaleFeatureState,
   UpscaleTranslations,
   UpscaleFeatureConfig,
-  IUpscaleProvider,
-  UpscaleProviderEntry,
+  UpscaleInputBuilder,
+  UpscaleResultExtractor,
 } from "./domain";
 
 // Infrastructure Services
-export {
-  upscaleProviderRegistry,
-  executeUpscale,
-  getAvailableProvider,
-  hasUpscaleProvider,
-} from "./infrastructure";
+export { executeUpscale, hasUpscaleSupport } from "./infrastructure";
 export type { ExecuteUpscaleOptions } from "./infrastructure";
 
 // Presentation Hooks
@@ -38,6 +33,3 @@ export type {
   UpscaleFeatureProps,
   UpscaleResultViewProps,
 } from "./presentation";
-
-// ComparisonSlider is available from ./presentation/components if needed directly
-// but not re-exported to avoid conflict with background feature's ComparisonSlider

package/src/features/upscaling/infrastructure/services/index.ts

@@ -1,7 +1,2 @@
-export { upscaleProviderRegistry } from "./upscale-provider-registry";
-export {
-  executeUpscale,
-  getAvailableProvider,
-  hasUpscaleProvider,
-} from "./upscale-executor";
+export { executeUpscale, hasUpscaleSupport } from "./upscale-executor";
 export type { ExecuteUpscaleOptions } from "./upscale-executor";

package/src/features/upscaling/infrastructure/services/upscale-executor.ts

@@ -1,64 +1,98 @@
 /**
  * Upscale Executor
- * Executes upscale operations using registered providers
+ * Provider-agnostic upscale execution using active AI provider
+ * Model and input format are provided via options from app level
+ *
+ * Note: imageBase64 must be provided in the request. Image URI to base64
+ * conversion should be handled at the app level before calling this executor.
  */
 
-import { upscaleProviderRegistry } from "./upscale-provider-registry";
-import type { UpscaleRequest, UpscaleResult } from "../../domain/types";
+import { providerRegistry } from "../../../../infrastructure/services";
+import { cleanBase64 } from "../../../../infrastructure/utils";
+import type {
+  UpscaleRequest,
+  UpscaleResult,
+  UpscaleInputBuilder,
+  UpscaleResultExtractor,
+} from "../../domain/types";
 
 declare const __DEV__: boolean;
 
 export interface ExecuteUpscaleOptions {
-  providerId?: string;
+  model: string;
+  buildInput: UpscaleInputBuilder;
+  extractResult?: UpscaleResultExtractor;
   onProgress?: (progress: number) => void;
 }
 
+function defaultExtractResult(result: unknown): string | undefined {
+  if (typeof result !== "object" || result === null) return undefined;
+
+  const r = result as Record<string, unknown>;
+
+  if (typeof r.image === "string") return r.image;
+  if (Array.isArray(r.images) && r.images[0]?.url) return r.images[0].url;
+
+  return undefined;
+}
+
 export async function executeUpscale(
   request: UpscaleRequest,
-  options?: ExecuteUpscaleOptions,
+  options: ExecuteUpscaleOptions,
 ): Promise<UpscaleResult> {
-  const provider = upscaleProviderRegistry.get(options?.providerId);
+  const provider = providerRegistry.getActiveProvider();
 
   if (!provider) {
-    return {
-      success: false,
-      error: "No upscale provider available",
-    };
+    return { success: false, error: "No AI provider configured" };
   }
 
-  if (!provider.isAvailable()) {
-    return {
-      success: false,
-      error: `Provider ${provider.providerId} is not available`,
-    };
+  if (!provider.isInitialized()) {
+    return { success: false, error: "AI provider not initialized" };
   }
 
+  if (!request.imageBase64) {
+    return { success: false, error: "Image base64 is required" };
+  }
+
+  const { model, buildInput, extractResult, onProgress } = options;
+
   if (__DEV__) {
     // eslint-disable-next-line no-console
-    console.log(`[UpscaleExecutor] Using provider: ${provider.providerId}`);
+    console.log(`[Upscale] Provider: ${provider.providerId}, Model: ${model}`);
   }
 
   try {
-    return await provider.upscale(request, options?.onProgress);
+    onProgress?.(10);
+
+    const base64 = cleanBase64(request.imageBase64);
+    onProgress?.(30);
+
+    const input = buildInput(base64, request.options);
+    onProgress?.(40);
+
+    const result = await provider.run(model, input);
+    onProgress?.(90);
+
+    const extractor = extractResult || defaultExtractResult;
+    const imageUrl = extractor(result);
+    onProgress?.(100);
+
+    if (!imageUrl) {
+      return { success: false, error: "No image in response" };
+    }
+
+    return { success: true, imageUrl };
   } catch (error) {
     const message = error instanceof Error ? error.message : String(error);
-
     if (__DEV__) {
       // eslint-disable-next-line no-console
-      console.error("[UpscaleExecutor] Error:", message);
+      console.error("[Upscale] Error:", message);
     }
-
-    return {
-      success: false,
-      error: message,
-    };
+    return { success: false, error: message };
   }
 }
 
-export function getAvailableProvider(providerId?: string) {
-  return upscaleProviderRegistry.get(providerId);
-}
-
-export function hasUpscaleProvider(providerId: string): boolean {
-  return upscaleProviderRegistry.hasProvider(providerId);
+export function hasUpscaleSupport(): boolean {
+  const provider = providerRegistry.getActiveProvider();
+  return provider !== null && provider.isInitialized();
 }

package/src/features/upscaling/presentation/hooks/useUpscaleFeature.ts

@@ -14,7 +14,7 @@ import type {
 declare const __DEV__: boolean;
 
 export interface UseUpscaleFeatureProps {
-  config?: UpscaleFeatureConfig;
+  config: UpscaleFeatureConfig;
   userId: string;
   onSelectImage: () => Promise<string | null>;
   onSaveImage: (imageUrl: string) => Promise<void>;
@@ -46,7 +46,7 @@ export function useUpscaleFeature(
       const uri = await onSelectImage();
       if (uri) {
         setState((prev) => ({ ...prev, imageUri: uri, error: null }));
-        config?.onImageSelect?.(uri);
+        config.onImageSelect?.(uri);
       }
     } catch (error) {
       const message = error instanceof Error ? error.message : String(error);
@@ -68,21 +68,26 @@ export function useUpscaleFeature(
       error: null,
     }));
 
-    config?.onProcessingStart?.();
+    config.onProcessingStart?.();
 
     if (__DEV__) {
       // eslint-disable-next-line no-console
       console.log("[useUpscaleFeature] Starting upscale process");
     }
 
+    const imageBase64 = await config.prepareImage(state.imageUri);
+
     const result: UpscaleResult = await executeUpscale(
       {
         imageUri: state.imageUri,
+        imageBase64,
         userId,
-        options: { scaleFactor: config?.defaultScaleFactor || 2 },
+        options: { scaleFactor: config.defaultScaleFactor || 2 },
       },
       {
-        providerId: config?.providerId,
+        model: config.model,
+        buildInput: config.buildInput,
+        extractResult: config.extractResult,
         onProgress: handleProgress,
       },
     );
@@ -95,7 +100,7 @@ export function useUpscaleFeature(
         processedUrl: url,
         progress: 100,
       }));
-      config?.onProcessingComplete?.(result);
+      config.onProcessingComplete?.(result);
     } else {
       const errorMessage = result.error || "Processing failed";
       setState((prev) => ({
@@ -104,7 +109,7 @@ export function useUpscaleFeature(
         error: errorMessage,
         progress: 0,
       }));
-      config?.onError?.(errorMessage);
+      config.onError?.(errorMessage);
     }
   }, [state.imageUri, userId, config, handleProgress]);
 

package/src/index.ts

@@ -10,6 +10,9 @@
  *   } from '@umituz/react-native-ai-generation-content';
  */
 
+// eslint-disable-next-line no-console
+if (typeof __DEV__ !== "undefined" && __DEV__) console.log("📍 [LIFECYCLE] @umituz/react-native-ai-generation-content/index.ts - Module loading");
+
 // =============================================================================
 // DOMAIN LAYER - Types & Interfaces
 // =============================================================================
@@ -314,3 +317,39 @@ export * from "./features/upscaling";
 
 export * from "./features/photo-restoration";
 
+// =============================================================================
+// FEATURES - AI Hug
+// =============================================================================
+
+export * from "./features/ai-hug";
+
+// =============================================================================
+// FEATURES - AI Kiss
+// =============================================================================
+
+export * from "./features/ai-kiss";
+
+// =============================================================================
+// FEATURES - Face Swap
+// =============================================================================
+
+export * from "./features/face-swap";
+
+// =============================================================================
+// FEATURES - Anime Selfie
+// =============================================================================
+
+export * from "./features/anime-selfie";
+
+// =============================================================================
+// FEATURES - Remove Background
+// =============================================================================
+
+export * from "./features/remove-background";
+
+// =============================================================================
+// FEATURES - Remove Object
+// =============================================================================
+
+export * from "./features/remove-object";
+

package/src/types/jsx.d.ts

@@ -0,0 +1,19 @@
+/**
+ * React 19 JSX Compatibility for React Native
+ * Fixes JSX namespace issues between React 19 and react-native
+ */
+
+import type { JSX as ReactJSX } from 'react';
+
+declare global {
+  namespace JSX {
+    interface Element extends ReactJSX.Element {}
+    interface ElementClass extends ReactJSX.ElementClass {}
+    interface ElementAttributesProperty extends ReactJSX.ElementAttributesProperty {}
+    interface ElementChildrenAttribute extends ReactJSX.ElementChildrenAttribute {}
+    interface IntrinsicAttributes extends ReactJSX.IntrinsicAttributes {}
+    interface IntrinsicClassAttributes<T> extends ReactJSX.IntrinsicClassAttributes<T> {}
+  }
+}
+
+export {};

package/src/features/face-swap/domain/entities.ts

@@ -1,48 +0,0 @@
-/**
- * Face Swap Domain Entities
- */
-
-export interface FaceSwapConfig {
-    /**
-     * Enhance the face details after swapping
-     * @default true
-     */
-    enhanceFace?: boolean;
-
-    /**
-     * Preserve the skin tone of the target image
-     * @default true
-     */
-    preserveSkinTone?: boolean;
-}
-
-export interface FaceSwapRequest {
-    /**
-     * The target image where the face will be swapped TO.
-     * Can be a Base64 string or a remote URL.
-     */
-    targetImage: string;
-
-    /**
-     * The source image containing the face to swap FROM.
-     * Can be a Base64 string or a remote URL.
-     */
-    sourceImage: string;
-
-    /**
-     * Optional configuration for the swap process
-     */
-    options?: FaceSwapConfig;
-}
-
-export interface FaceSwapResult {
-    /**
-     * URL or Base64 of the resulting image
-     */
-    imageUrl: string;
-
-    /**
-     * Metadata about the generation
-     */
-    metadata?: Record<string, unknown>;
-}

package/src/features/photo-restoration/domain/types/provider.types.ts

@@ -1,23 +0,0 @@
-/**
- * Photo Restore Provider Types
- * Interface for provider implementations
- */
-
-import type { PhotoRestoreRequest, PhotoRestoreResult } from "./photo-restore.types";
-
-export interface IPhotoRestoreProvider {
-  readonly providerId: string;
-  readonly providerName: string;
-
-  isAvailable(): boolean;
-
-  restore(
-    request: PhotoRestoreRequest,
-    onProgress?: (progress: number) => void,
-  ): Promise<PhotoRestoreResult>;
-}
-
-export interface PhotoRestoreProviderEntry {
-  provider: IPhotoRestoreProvider;
-  priority: number;
-}

package/src/features/photo-restoration/infrastructure/services/photo-restore-provider-registry.ts

@@ -1,77 +0,0 @@
-/**
- * Photo Restore Provider Registry
- * Manages registered photo restore providers
- */
-
-import type {
-  IPhotoRestoreProvider,
-  PhotoRestoreProviderEntry,
-} from "../../domain/types";
-
-declare const __DEV__: boolean;
-
-class PhotoRestoreProviderRegistry {
-  private providers: Map<string, PhotoRestoreProviderEntry> = new Map();
-  private defaultProviderId: string | null = null;
-
-  register(provider: IPhotoRestoreProvider, priority = 0): void {
-    this.providers.set(provider.providerId, { provider, priority });
-
-    if (__DEV__) {
-      // eslint-disable-next-line no-console
-      console.log(
-        `[PhotoRestoreRegistry] Registered provider: ${provider.providerId}`,
-      );
-    }
-  }
-
-  unregister(providerId: string): void {
-    this.providers.delete(providerId);
-  }
-
-  setDefault(providerId: string): void {
-    if (!this.providers.has(providerId)) {
-      throw new Error(`Provider ${providerId} not registered`);
-    }
-    this.defaultProviderId = providerId;
-  }
-
-  get(providerId?: string): IPhotoRestoreProvider | null {
-    const id = providerId || this.defaultProviderId;
-    if (!id) return this.getHighestPriority();
-
-    const entry = this.providers.get(id);
-    return entry?.provider || null;
-  }
-
-  getAvailable(): IPhotoRestoreProvider | null {
-    const sorted = this.getSortedProviders();
-    return sorted.find((p) => p.isAvailable()) || null;
-  }
-
-  hasProvider(providerId: string): boolean {
-    return this.providers.has(providerId);
-  }
-
-  getProviderIds(): string[] {
-    return Array.from(this.providers.keys());
-  }
-
-  private getHighestPriority(): IPhotoRestoreProvider | null {
-    const sorted = this.getSortedProviders();
-    return sorted[0] || null;
-  }
-
-  private getSortedProviders(): IPhotoRestoreProvider[] {
-    return Array.from(this.providers.values())
-      .sort((a, b) => b.priority - a.priority)
-      .map((entry) => entry.provider);
-  }
-
-  reset(): void {
-    this.providers.clear();
-    this.defaultProviderId = null;
-  }
-}
-
-export const photoRestoreProviderRegistry = new PhotoRestoreProviderRegistry();

package/src/features/upscaling/domain/types/provider.types.ts

@@ -1,23 +0,0 @@
-/**
- * Upscale Provider Types
- * Interface for provider implementations
- */
-
-import type { UpscaleRequest, UpscaleResult } from "./upscale.types";
-
-export interface IUpscaleProvider {
-  readonly providerId: string;
-  readonly providerName: string;
-
-  isAvailable(): boolean;
-
-  upscale(
-    request: UpscaleRequest,
-    onProgress?: (progress: number) => void,
-  ): Promise<UpscaleResult>;
-}
-
-export interface UpscaleProviderEntry {
-  provider: IUpscaleProvider;
-  priority: number;
-}