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

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
 // =============================================================================
@@ -308,3 +311,45 @@ export * from "./features/background";
 
 export * from "./features/upscaling";
 
+// =============================================================================
+// FEATURES - Photo Restoration
+// =============================================================================
+
+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/entities.ts

@@ -1,48 +0,0 @@
-/**
- * Photo Restoration Domain Entities
- */
-
-export interface PhotoRestorationConfig {
-    /**
-     * Whether to fix scratches and creases
-     * @default true
-     */
-    fixScratches?: boolean;
-
-    /**
-     * Whether to enhance faces in the photo
-     * @default true
-     */
-    enhanceFaces?: boolean;
-
-    /**
-     * Upscale factor for the result
-     * @default 2
-     */
-    upscaleFactor?: 1 | 2 | 4;
-}
-
-export interface PhotoRestorationRequest {
-    /**
-     * The image to restore.
-     * Can be a Base64 string or a remote URL.
-     */
-    image: string;
-
-    /**
-     * Optional configuration
-     */
-    options?: PhotoRestorationConfig;
-}
-
-export interface PhotoRestorationResult {
-    /**
-     * The restored image URL or Base64
-     */
-    imageUrl: string;
-
-    /**
-     * Metadata about the generation
-     */
-    metadata?: Record<string, unknown>;
-}

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;
-}

package/src/features/upscaling/infrastructure/services/upscale-provider-registry.ts

@@ -1,77 +0,0 @@
-/**
- * Upscale Provider Registry
- * Manages registered upscale providers
- */
-
-import type {
-  IUpscaleProvider,
-  UpscaleProviderEntry,
-} from "../../domain/types";
-
-declare const __DEV__: boolean;
-
-class UpscaleProviderRegistry {
-  private providers: Map<string, UpscaleProviderEntry> = new Map();
-  private defaultProviderId: string | null = null;
-
-  register(provider: IUpscaleProvider, priority = 0): void {
-    this.providers.set(provider.providerId, { provider, priority });
-
-    if (__DEV__) {
-      // eslint-disable-next-line no-console
-      console.log(
-        `[UpscaleRegistry] 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): IUpscaleProvider | null {
-    const id = providerId || this.defaultProviderId;
-    if (!id) return this.getHighestPriority();
-
-    const entry = this.providers.get(id);
-    return entry?.provider || null;
-  }
-
-  getAvailable(): IUpscaleProvider | 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(): IUpscaleProvider | null {
-    const sorted = this.getSortedProviders();
-    return sorted[0] || null;
-  }
-
-  private getSortedProviders(): IUpscaleProvider[] {
-    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 upscaleProviderRegistry = new UpscaleProviderRegistry();