@umituz/react-native-ai-generation-content 1.82.6 → 1.82.8

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@umituz/react-native-ai-generation-content",
-  "version": "1.82.6",
+  "version": "1.82.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/creations/presentation/hooks/useProcessingJobsPoller.ts

@@ -85,7 +85,8 @@ export function useProcessingJobsPoller(
       try {
         await repository.update(userId, creation.id, {
           status: CREATION_STATUS.FAILED,
-          metadata: { error: "Generation timed out" },
+          metadata: { ...creation.metadata, error: "Generation timed out" },
+          completedAt: new Date(),
         });
       } catch (e) {
         if (typeof __DEV__ !== "undefined" && __DEV__) {
@@ -129,7 +130,8 @@ export function useProcessingJobsPoller(
           }
           await repository.update(userId, creation.id, {
             status: CREATION_STATUS.FAILED,
-            metadata: { error: "No valid result URL received" },
+            metadata: { ...creation.metadata, error: "No valid result URL received" },
+            completedAt: new Date(),
           });
           return;
         }
@@ -143,6 +145,7 @@ export function useProcessingJobsPoller(
           status: CREATION_STATUS.COMPLETED,
           uri,
           output,
+          completedAt: new Date(),
         });
       } else if (status.status === QUEUE_STATUS.FAILED) {
         if (typeof __DEV__ !== "undefined" && __DEV__) console.log("[ProcessingJobsPoller] Failed:", creation.id);
@@ -151,7 +154,8 @@ export function useProcessingJobsPoller(
 
         await repository.update(userId, creation.id, {
           status: CREATION_STATUS.FAILED,
-          metadata: { error: "Generation failed" },
+          metadata: { ...creation.metadata, error: "Generation failed" },
+          completedAt: new Date(),
         });
       }
     } catch (error) {

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

@@ -53,7 +53,12 @@ export const executeWizardGeneration = async (params: ExecuteGenerationParams):
     await generationFn(input, prompt);
 
     if (isMountedRef.current) {
-      dispatch({ type: "COMPLETE" });
+      // For video queue mode, don't dispatch COMPLETE — the queue polling handles
+      // completion separately via onSuccess. The function above just submitted to queue.
+      // For photo (blocking) mode, the generation is truly done here.
+      if (!isVideoMode) {
+        dispatch({ type: "COMPLETE" });
+      }
     }
   } catch (error: unknown) {
     if (!isMountedRef.current) return;

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

@@ -25,13 +25,16 @@ export function useVideoQueueGeneration(props: UseVideoQueueGenerationProps): Us
   const consecutiveErrorsRef = useRef(0);
   const [isGenerating, setIsGenerating] = useState(false);
 
+  const clearPolling = useCallback(() => {
+    if (pollingRef.current) {
+      clearInterval(pollingRef.current);
+      pollingRef.current = null;
+    }
+  }, []);
+
   useEffect(() => {
     return () => {
-      if (pollingRef.current) {
-        clearInterval(pollingRef.current);
-        pollingRef.current = null;
-      }
-      // Reset all refs on unmount
+      clearPolling();
       isGeneratingRef.current = false;
       isPollingRef.current = false;
       consecutiveErrorsRef.current = 0;
@@ -40,9 +43,10 @@ export function useVideoQueueGeneration(props: UseVideoQueueGenerationProps): Us
       modelRef.current = null;
       setIsGenerating(false);
     };
-  }, []);
+  }, [clearPolling]);
 
   const resetRefs = useCallback(() => {
+    clearPolling();
     creationIdRef.current = null;
     requestIdRef.current = null;
     modelRef.current = null;
@@ -50,10 +54,13 @@ export function useVideoQueueGeneration(props: UseVideoQueueGenerationProps): Us
     isPollingRef.current = false;
     consecutiveErrorsRef.current = 0;
     setIsGenerating(false);
-  }, []);
+  }, [clearPolling]);
 
   const handleComplete = useCallback(
     async (urls: GenerationUrls) => {
+      // Stop polling immediately on completion
+      clearPolling();
+
       const creationId = creationIdRef.current;
       const uri = (urls.videoUrl || urls.imageUrl) ?? "";
 
@@ -72,9 +79,12 @@ export function useVideoQueueGeneration(props: UseVideoQueueGenerationProps): Us
         if (typeof __DEV__ !== "undefined" && __DEV__) {
           console.error("[VideoQueue] ❌ Invalid completion data:", { creationId, userId, uri });
         }
+        resetRefs();
+        onError?.("Invalid completion data - no valid URL received");
         return;
       }
 
+      let persistenceSucceeded = true;
       if (creationId && userId) {
         try {
           await persistence.updateToCompleted(userId, creationId, {
@@ -87,27 +97,32 @@ export function useVideoQueueGeneration(props: UseVideoQueueGenerationProps): Us
             console.log("[VideoQueue] ✅ Updated completion status in Firestore");
           }
         } catch (error) {
+          persistenceSucceeded = false;
           if (typeof __DEV__ !== "undefined" && __DEV__) {
             console.error("[VideoQueue] ❌ Failed to update completion status:", error);
           }
         }
       }
 
+      resetRefs();
+
+      // Still call onSuccess even if persistence failed - the generation itself succeeded
+      // The video/image URL is valid, user should still see the result
       if (typeof __DEV__ !== "undefined" && __DEV__) {
-        console.log("[VideoQueue] 🎯 Calling onSuccess callback now...");
+        console.log("[VideoQueue] 🎯 Calling onSuccess callback now...", { persistenceSucceeded });
       }
-      resetRefs();
       onSuccess?.(urls);
 
       if (typeof __DEV__ !== "undefined" && __DEV__) {
         console.log("[VideoQueue] ✅ onSuccess callback completed");
       }
     },
-    [userId, persistence, onSuccess, resetRefs],
+    [userId, persistence, onSuccess, onError, resetRefs, clearPolling],
   );
 
   const handleError = useCallback(
     async (errorMsg: string) => {
+      clearPolling();
       const creationId = creationIdRef.current;
       if (creationId && userId) {
         try {
@@ -121,7 +136,7 @@ export function useVideoQueueGeneration(props: UseVideoQueueGenerationProps): Us
       resetRefs();
       onError?.(errorMsg);
     },
-    [userId, persistence, onError, resetRefs],
+    [userId, persistence, onError, resetRefs, clearPolling],
   );
 
   const pollStatus = useCallback(async () => {
@@ -129,15 +144,21 @@ export function useVideoQueueGeneration(props: UseVideoQueueGenerationProps): Us
     const model = modelRef.current;
     if (!requestId || !model) return;
 
-    await pollQueueStatus({
-      requestId,
-      model,
-      isPollingRef,
-      pollingRef,
-      consecutiveErrorsRef,
-      onComplete: handleComplete,
-      onError: handleError,
-    });
+    try {
+      await pollQueueStatus({
+        requestId,
+        model,
+        isPollingRef,
+        pollingRef,
+        consecutiveErrorsRef,
+        onComplete: handleComplete,
+        onError: handleError,
+      });
+    } catch (error) {
+      if (typeof __DEV__ !== "undefined" && __DEV__) {
+        console.error("[VideoQueue] Unexpected poll error:", error);
+      }
+    }
   }, [handleComplete, handleError]);
 
   const startGeneration = useCallback(
@@ -176,14 +197,34 @@ export function useVideoQueueGeneration(props: UseVideoQueueGenerationProps): Us
           if (typeof __DEV__ !== "undefined" && __DEV__) {
             console.error("[VideoQueue] Failed to save processing creation:", error);
           }
+          // Continue without creation tracking - generation can still proceed
+          // The video will be generated but won't appear in gallery history
+        }
+      }
+
+      let queueResult;
+      try {
+        queueResult = await strategy.submitToQueue(input);
+      } catch (error) {
+        // Queue submission threw - reset state and report error
+        if (creationId && userId) {
+          try {
+            await persistence.updateToFailed(userId, creationId, error instanceof Error ? error.message : "Queue submission failed");
+          } catch { /* best effort */ }
         }
+        isGeneratingRef.current = false;
+        setIsGenerating(false);
+        onError?.(error instanceof Error ? error.message : "Queue submission failed");
+        return;
       }
 
-      const queueResult = await strategy.submitToQueue(input);
       if (!queueResult.success || !queueResult.requestId || !queueResult.model) {
         if (creationId && userId) {
-          await persistence.updateToFailed(userId, creationId, queueResult.error || "Queue submission failed");
+          try {
+            await persistence.updateToFailed(userId, creationId, queueResult.error || "Queue submission failed");
+          } catch { /* best effort */ }
         }
+        isGeneratingRef.current = false;
         setIsGenerating(false);
         onError?.(queueResult.error || "Queue submission failed");
         return;
@@ -205,7 +246,7 @@ export function useVideoQueueGeneration(props: UseVideoQueueGenerationProps): Us
       pollingRef.current = setInterval(() => void pollStatus(), DEFAULT_POLL_INTERVAL_MS);
       void pollStatus();
     },
-    [userId, scenario, persistence, strategy, pollStatus, onError],
+    [userId, scenario, persistence, strategy, creditCost, pollStatus, onError],
   );
 
   return { isGenerating, startGeneration };

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

@@ -136,6 +136,7 @@ export const useGenerationOrchestrator = <TInput, TResult>(
       // Create new AbortController for this generation
       abortControllerRef.current = new AbortController();
       isGeneratingRef.current = true;
+      let moderationPending = false;
       setState({ ...INITIAL_STATE, status: "checking", isGenerating: true });
 
       if (typeof __DEV__ !== "undefined" && __DEV__) {
@@ -167,7 +168,7 @@ export const useGenerationOrchestrator = <TInput, TResult>(
           console.log("[Orchestrator] Starting moderation check");
         }
 
-        return await handleModeration({
+        const result = await handleModeration({
           input,
           moderation,
           alertMessages,
@@ -180,6 +181,14 @@ export const useGenerationOrchestrator = <TInput, TResult>(
           onError,
           handleLifecycleComplete,
         });
+
+        // If handleModeration returned undefined, the warning dialog was shown.
+        // The dialog's proceed/cancel callbacks now own isGeneratingRef — don't reset in finally.
+        if (result === undefined && moderation) {
+          moderationPending = true;
+        }
+
+        return result;
       } catch (err) {
         // Don't show error if aborted
         if (abortControllerRef.current?.signal.aborted) {
@@ -197,7 +206,10 @@ export const useGenerationOrchestrator = <TInput, TResult>(
         handleLifecycleComplete("error", undefined, error);
         throw error;
       } finally {
-        isGeneratingRef.current = false;
+        // Only reset if moderation dialog did not take ownership
+        if (!moderationPending) {
+          isGeneratingRef.current = false;
+        }
         abortControllerRef.current = null;
       }
     },