@umituz/react-native-ai-generation-content 1.61.25 → 1.61.26

package/src/infrastructure/utils/video-result-extractor.util.ts

@@ -0,0 +1,38 @@
+/**
+ * Video Result Extractor Utility
+ * Shared extractor for video generation results
+ */
+
+export interface ExtractedVideoResult {
+  videoUrl?: string;
+  thumbnailUrl?: string;
+}
+
+/**
+ * Default extractor for video generation results.
+ * Handles common provider response formats.
+ */
+export function defaultExtractVideoResult(
+  result: unknown,
+): ExtractedVideoResult | undefined {
+  if (typeof result !== "object" || result === null) return undefined;
+
+  const r = result as Record<string, unknown>;
+
+  if (typeof r.video === "string") {
+    return { videoUrl: r.video };
+  }
+
+  if (r.video && typeof r.video === "object") {
+    const video = r.video as Record<string, unknown>;
+    if (typeof video.url === "string") {
+      return {
+        videoUrl: video.url,
+        thumbnailUrl:
+          typeof r.thumbnail === "string" ? r.thumbnail : undefined,
+      };
+    }
+  }
+
+  return undefined;
+}

package/src/presentation/constants/alert-messages.ts

@@ -0,0 +1,14 @@
+/**
+ * Default Alert Messages
+ * Shared constant for generation error messages
+ */
+
+import type { AlertMessages } from "../hooks/generation/types";
+
+export const DEFAULT_ALERT_MESSAGES: AlertMessages = {
+  networkError: "No internet connection. Please check your network.",
+  policyViolation: "Content not allowed. Please try again.",
+  saveFailed: "Failed to save. Please try again.",
+  creditFailed: "Credit operation failed. Please try again.",
+  unknown: "An error occurred. Please try again.",
+};

package/src/presentation/hooks/generation/errors.ts

@@ -58,11 +58,15 @@ export const parseError = (err: unknown): GenerationError => {
   return createGenerationError("unknown", "Generation failed");
 };
 
+const VALID_ERROR_TYPES: readonly string[] = ["network", "credits", "policy", "save", "unknown"];
+
 const isGenerationError = (err: unknown): err is GenerationError => {
   return (
     typeof err === "object" &&
     err !== null &&
     "type" in err &&
-    "message" in err
+    "message" in err &&
+    typeof (err as GenerationError).type === "string" &&
+    VALID_ERROR_TYPES.includes((err as GenerationError).type)
   );
 };

package/src/presentation/hooks/generation/useImageGeneration.ts

@@ -4,7 +4,7 @@
  * Uses centralized orchestrator for credit/error handling
  */
 
-import { useMemo, useCallback } from "react";
+import { useMemo, useCallback, useRef } from "react";
 import { useGenerationOrchestrator } from "./orchestrator";
 import type { GenerationStrategy, AlertMessages } from "./types";
 import { executeImageFeature } from "../../../infrastructure/services";
@@ -105,9 +105,13 @@ export const useImageGeneration = <
     [],
   );
 
+  const lastInputRef = useRef<TInput | null>(null);
+
   const strategy: GenerationStrategy<TInput, TResult> = useMemo(
     () => ({
       execute: async (input) => {
+        lastInputRef.current = input;
+
         // Build executor input
         const executorInput = buildExecutorInput
           ? buildExecutorInput(input)
@@ -129,7 +133,7 @@ export const useImageGeneration = <
       getCreditCost: () => creditCost,
       save: buildCreation
         ? async (result, uid) => {
-            const creation = buildCreation(result, {} as TInput);
+            const creation = buildCreation(result, lastInputRef.current ?? {} as TInput);
             if (creation) {
               await repository.create(uid, creation);
             }

package/src/presentation/hooks/generation/useVideoGeneration.ts

@@ -4,7 +4,7 @@
  * Uses centralized orchestrator for credit/error handling
  */
 
-import { useMemo, useCallback } from "react";
+import { useMemo, useCallback, useRef } from "react";
 import { useGenerationOrchestrator } from "./orchestrator";
 import type { GenerationStrategy, AlertMessages } from "./types";
 import { executeVideoFeature } from "../../../infrastructure/services";
@@ -60,9 +60,13 @@ export const useVideoGeneration = <TResult>(
     [],
   );
 
+  const lastInputRef = useRef<DualImageVideoInput | null>(null);
+
   const strategy: GenerationStrategy<DualImageVideoInput, TResult> = useMemo(
     () => ({
       execute: async (input) => {
+        lastInputRef.current = input;
+
         const result = await executeVideoFeature(
           featureType,
           {
@@ -81,7 +85,7 @@ export const useVideoGeneration = <TResult>(
       getCreditCost: () => creditCost,
       save: buildCreation
         ? async (result, uid) => {
-            const creation = buildCreation(result, {} as DualImageVideoInput);
+            const creation = buildCreation(result, lastInputRef.current ?? {} as DualImageVideoInput);
             if (creation) {
               await repository.create(uid, creation);
             }

package/src/presentation/hooks/useGenerationFlow.ts

@@ -84,29 +84,26 @@ export const useGenerationFlow = ({
   const goNext = useCallback(() => {
     if (!canGoNext) return;
 
-    setState((prev) => {
-      const nextIndex = prev.currentStepIndex + 1;
-
-      if (nextIndex >= config.photoSteps.length) {
-        const newState = { ...prev, isComplete: true };
-        onComplete?.(newState);
-        return newState;
-      }
+    const nextIndex = state.currentStepIndex + 1;
+    const isLastStep = nextIndex >= config.photoSteps.length;
 
+    if (isLastStep) {
+      const newState = { ...state, isComplete: true };
+      setState(newState);
+      onComplete?.(newState);
+    } else {
+      setState((prev) => ({ ...prev, currentStepIndex: nextIndex }));
       onStepChange?.(nextIndex, config.photoSteps[nextIndex]);
-      return { ...prev, currentStepIndex: nextIndex };
-    });
-  }, [canGoNext, config.photoSteps, onComplete, onStepChange]);
+    }
+  }, [canGoNext, state, config.photoSteps, onComplete, onStepChange]);
 
   const goBack = useCallback(() => {
     if (!canGoBack) return;
 
-    setState((prev) => {
-      const prevIndex = prev.currentStepIndex - 1;
-      onStepChange?.(prevIndex, config.photoSteps[prevIndex]);
-      return { ...prev, currentStepIndex: prevIndex, isComplete: false };
-    });
-  }, [canGoBack, config.photoSteps, onStepChange]);
+    const prevIndex = state.currentStepIndex - 1;
+    setState((prev) => ({ ...prev, currentStepIndex: prevIndex, isComplete: false }));
+    onStepChange?.(prevIndex, config.photoSteps[prevIndex]);
+  }, [canGoBack, state.currentStepIndex, config.photoSteps, onStepChange]);
 
   const updatePhoto = useCallback((imageUri: string, previewUrl?: string) => {
     setState((prev) =>
@@ -159,12 +156,10 @@ export const useGenerationFlow = ({
   }, [config]);
 
   const complete = useCallback(() => {
-    setState((prev) => {
-      const newState = { ...prev, isComplete: true };
-      onComplete?.(newState);
-      return newState;
-    });
-  }, [onComplete]);
+    const newState = { ...state, isComplete: true };
+    setState(newState);
+    onComplete?.(newState);
+  }, [state, onComplete]);
 
   return {
     state,

package/src/presentation/layouts/DualImageFeatureLayout.tsx

@@ -122,7 +122,6 @@ export const DualImageFeatureLayout: React.FC<DualImageFeatureLayoutProps> = ({
         onGenerate={handleProcess}
         isGenerating={feature.isProcessing}
         isDisabled={!feature.sourceImageUri || !feature.targetImageUri}
-        progress={feature.progress}
         translations={{
           generateButton: translations.processButtonText,
           generatingButton: translations.processingText,

package/src/presentation/layouts/DualImageVideoFeatureLayout.tsx

@@ -115,7 +115,7 @@ export const DualImageVideoFeatureLayout: React.FC<DualImageVideoFeatureLayoutPr
       <AIGenerationForm
         onGenerate={handleProcess}
         isGenerating={feature.isProcessing}
-        progress={feature.progress}
+        isDisabled={!feature.sourceImageUri || !feature.targetImageUri}
         translations={{
           generateButton: translations.processButtonText,
           generatingButton: translations.processingText,

package/src/presentation/layouts/SingleImageFeatureLayout.tsx

@@ -138,7 +138,7 @@ export const SingleImageFeatureLayout: React.FC<SingleImageFeatureLayoutProps> =
       <AIGenerationForm
         onGenerate={handleProcess}
         isGenerating={feature.isProcessing}
-        progress={feature.progress}
+        isDisabled={!feature.imageUri}
         translations={{
           generateButton: translations.processButtonText,
           generatingButton: translations.processingText,

package/src/presentation/layouts/SingleImageWithPromptFeatureLayout.tsx

@@ -125,7 +125,7 @@ export const SingleImageWithPromptFeatureLayout: React.FC<SingleImageWithPromptF
       <AIGenerationForm
         onGenerate={handleProcess}
         isGenerating={feature.isProcessing}
-        progress={feature.progress}
+        isDisabled={!feature.imageUri}
         translations={{
           generateButton: translations.processButtonText,
           generatingButton: translations.processingText,