@umituz/react-native-ai-generation-content 1.83.6 → 1.83.8

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@umituz/react-native-ai-generation-content",
-  "version": "1.83.6",
+  "version": "1.83.8",
   "description": "Provider-agnostic AI generation orchestration for React Native with result preview components",
   "main": "src/index.ts",
   "types": "src/index.ts",

package/src/domains/generation/wizard/presentation/hooks/usePhotoBlockingGeneration.ts

@@ -109,6 +109,8 @@ export function usePhotoBlockingGeneration(
           const duration = typeof inputData?.duration === "number" ? inputData.duration : undefined;
           const resolution = typeof inputData?.resolution === "string" ? inputData.resolution : undefined;
 
+          const aspectRatio = typeof inputData?.aspectRatio === "string" ? inputData.aspectRatio : undefined;
+
           const result = await persistence.saveAsProcessing(userId, {
             scenarioId: scenario.id,
             scenarioTitle: scenario.title || scenario.id,
@@ -116,6 +118,9 @@ export function usePhotoBlockingGeneration(
             duration,
             resolution,
             creditCost,
+            aspectRatio,
+            provider: "fal",
+            outputType: scenario.outputType,
           });
           creationIdRef.current = result.creationId;
 
@@ -132,7 +137,7 @@ export function usePhotoBlockingGeneration(
       // Start blocking generation
       await generate(input);
     },
-    [userId, scenario, persistence, generate],
+    [userId, scenario, persistence, creditCost, generate],
   );
 
   return { isGenerating, startGeneration };

package/src/domains/generation/wizard/presentation/hooks/useVideoQueueGeneration.ts

@@ -42,9 +42,10 @@ export function useVideoQueueGeneration(props: UseVideoQueueGenerationProps): Us
       isPollingRef.current = false;
       consecutiveErrorsRef.current = 0;
       pollStartTimeRef.current = null;
-      creationIdRef.current = null;
-      requestIdRef.current = null;
-      modelRef.current = null;
+      // NOTE: Do NOT null creationIdRef/requestIdRef/modelRef here.
+      // In-flight poll callbacks may still resolve after unmount and need
+      // these refs to properly save the completed generation to Firestore.
+      // They are cleaned up by resetRefs() after handleComplete/handleError.
       setIsGenerating(false);
     };
   }, [clearPolling]);

package/src/domains/generation/wizard/presentation/hooks/useWizardFlowHandlers.ts

@@ -1,4 +1,4 @@
-import { useCallback } from "react";
+import { useCallback, useRef } from "react";
 import { AlertType, AlertMode, useAlert } from "@umituz/react-native-design-system";
 import { StepType, type StepDefinition } from "../../../../../domain/entities/flow-config.types";
 import type { UploadedImage } from "../../../../../presentation/hooks/generation/useAIGenerateState";
@@ -53,9 +53,12 @@ export function useWizardFlowHandlers(props: UseWizardFlowHandlersProps) {
   } = props;
 
   const alert = useAlert();
+  // Guard: prevent multiple onGenerationStart calls from auto-advancing steps
+  const generationStartedRef = useRef(false);
 
   const handleGenerationComplete = useCallback(
     (result: unknown) => {
+      generationStartedRef.current = false;
       setResult(result);
       setCurrentCreation(isCreation(result) ? result : null);
       onGenerationComplete?.(result);
@@ -66,6 +69,7 @@ export function useWizardFlowHandlers(props: UseWizardFlowHandlersProps) {
 
   const handleGenerationError = useCallback(
     (errorMessage: string) => {
+      generationStartedRef.current = false;
       const safeErrorMessage = errorMessage?.trim() || "error.generation.unknown";
       const displayMessage = safeErrorMessage.startsWith("error.") ? t(safeErrorMessage) : safeErrorMessage;
 
@@ -112,6 +116,8 @@ export function useWizardFlowHandlers(props: UseWizardFlowHandlersProps) {
       });
     }
     if (nextStepDef?.type === StepType.GENERATING && onGenerationStart) {
+      if (generationStartedRef.current) return;
+      generationStartedRef.current = true;
       onGenerationStart(mergedData, nextStep, handleGenerationError);
       return;
     }