@umituz/react-native-ai-generation-content 1.25.18 → 1.25.20

package/package.json

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

package/src/domains/wizard/presentation/hooks/useWizardGeneration.ts

@@ -14,6 +14,8 @@ import { executeVideoFeature } from "../../../../infrastructure/services/video-f
 import { createCreationsRepository } from "../../../../domains/creations/infrastructure/adapters";
 import type { VideoFeatureType } from "../../../../domain/interfaces";
 import type { AlertMessages } from "../../../../presentation/hooks/generation/orchestrator.types";
+import { enhanceCouplePrompt } from "../../../../features/couple-future/infrastructure/couplePromptEnhancer";
+import type { CoupleFeatureSelection } from "../../../../features/couple-future/domain/types";
 
 declare const __DEV__: boolean;
 
@@ -192,11 +194,37 @@ async function buildGenerationInput(
     return null;
   }
 
-  const prompt = scenario.aiPrompt || `Generate ${scenario.id} scene`;
+  let prompt = scenario.aiPrompt || `Generate ${scenario.id} scene`;
   const outputType = scenario.outputType || "video"; // Default to video for backward compatibility
 
-  // Build input based on output type
+  // For image generation, enhance prompt with style selections
   if (outputType === "image") {
+    // Extract style selections from wizard data
+    const selections: CoupleFeatureSelection = {};
+
+    // Romantic mood (multi-select array)
+    const romanticMoods = wizardData.selection_romantic_mood as string[] | undefined;
+    if (romanticMoods && romanticMoods.length > 0) {
+      selections.romanticMoods = romanticMoods;
+    }
+
+    // Art style (single select)
+    const artStyle = wizardData.selection_art_style as string | undefined;
+    if (artStyle && artStyle !== "original") {
+      selections.artStyle = artStyle;
+    }
+
+    // Artist style (single select)
+    const artist = wizardData.selection_artist_style as string | undefined;
+    if (artist && artist !== "original") {
+      selections.artist = artist;
+    }
+
+    // Enhance prompt with selected styles (only if not "original")
+    if (Object.keys(selections).length > 0) {
+      prompt = enhanceCouplePrompt(prompt, selections);
+    }
+
     return {
       partnerABase64: photo1Base64,
       partnerBBase64: photo2Base64,
@@ -280,21 +308,26 @@ export const useWizardGeneration = (
         const input = lastInputRef.current;
         if (!input) return;
 
-        const videoResult = result as { videoUrl?: string };
-        const imageResult = result as { imageUrl?: string };
-
         // Both input types have prompt field
         const prompt = 'prompt' in input ? input.prompt : '';
 
-        const creation = {
-          videoUrl: videoResult.videoUrl,
-          imageUrl: imageResult.imageUrl,
+        // Build creation object with only the relevant URL field
+        const creation: any = {
           scenarioId: scenario.id,
           scenarioTitle: scenario.title || scenario.id,
           prompt,
           createdAt: Date.now(),
         };
 
+        // Add only the relevant URL based on output type
+        if (outputType === "image") {
+          const imageResult = result as { imageUrl?: string };
+          creation.imageUrl = imageResult.imageUrl;
+        } else {
+          const videoResult = result as { videoUrl?: string };
+          creation.videoUrl = videoResult.videoUrl;
+        }
+
         await repository.create(uid, creation);
 
         if (typeof __DEV__ !== "undefined" && __DEV__) {