@umituz/react-native-ai-pruna-provider 1.0.34 → 1.0.36

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@umituz/react-native-ai-pruna-provider",
-  "version": "1.0.34",
+  "version": "1.0.36",
   "description": "Pruna AI provider for React Native - implements IAIProvider interface for unified AI generation",
   "main": "./src/index.ts",
   "types": "./src/index.ts",

package/src/infrastructure/services/pruna-input-builder.ts

@@ -69,8 +69,8 @@ async function buildImageEditInput(
   apiKey: string,
   sessionId: string,
 ): Promise<Record<string, unknown>> {
-  // p-image-edit requires uploaded file URLs, not base64
-  // First upload all images to Pruna storage, then use the URLs
+  // p-image-edit: Like video-studio, use raw base64 (not file upload)
+  // Get base64 image(s) from input
   let rawImages: string[];
 
   if (Array.isArray(input.images)) {
@@ -89,55 +89,34 @@ async function buildImageEditInput(
     throw new Error("Image is required for p-image-edit. Provide 'image', 'images', 'image_url', or 'image_urls'.");
   }
 
-  generationLogCollector.log(sessionId, TAG, `p-image-edit: uploading ${rawImages.length} image(s) to Pruna storage...`);
-
-  // Upload all images and get URLs
-  const imageUrls: string[] = [];
-  for (let i = 0; i < rawImages.length; i++) {
-    const rawImage = rawImages[i];
-    generationLogCollector.log(sessionId, TAG, `p-image-edit: uploading image ${i + 1}/${rawImages.length}...`);
-
-    // __DEV__ log image details before upload
-    if (typeof __DEV__ !== 'undefined' && __DEV__) {
-      console.log(`[DEV] [${TAG}] Image ${i + 1}/${rawImages.length}:`, {
-        sizeKB: Math.round(rawImage.length / 1024),
-        isUrl: rawImage.startsWith('http'),
-        hasDataUri: rawImage.startsWith('data:'),
-        preview: rawImage.substring(0, 50) + '...',
-      });
-    }
-
-    const url = await uploadFileToStorage(rawImage, apiKey, sessionId);
-    imageUrls.push(url);
+  generationLogCollector.log(sessionId, TAG, `p-image-edit: processing ${rawImages.length} image(s)...`);
 
-    // __DEV__ log successful upload
-    if (typeof __DEV__ !== 'undefined' && __DEV__) {
-      console.log(`[DEV] [${TAG}] Image ${i + 1}/${rawImages.length} uploaded successfully:`, {
-        url,
-        urlPrefix: url.substring(0, 50) + '...',
-      });
+  // Strip data URI prefixes (like video-studio does)
+  const strippedImages = rawImages.map(img => {
+    if (img.startsWith('http')) {
+      return img; // Already a URL, pass through
     }
-  }
+    // Strip data URI prefix to get raw base64
+    return img.includes('base64,') ? img.split('base64,')[1] : img;
+  });
 
-  generationLogCollector.log(sessionId, TAG, `p-image-edit: image upload completed, using first image`);
+  // Use first image (like video-studio)
+  const payload: Record<string, unknown> = {
+    images: [strippedImages[0]],  // Array with first image, raw base64
+    prompt,
+    aspect_ratio: aspectRatio,
+  };
 
   // __DEV__ log final payload
   if (typeof __DEV__ !== 'undefined' && __DEV__) {
     console.log(`[DEV] [${TAG}] Final p-image-edit payload:`, {
-      imageCount: imageUrls.length,
+      imageCount: strippedImages.length,
       promptLength: prompt.length,
       aspectRatio,
-      imageUrl: imageUrls[0].substring(0, 60) + '...',
+      base64Preview: strippedImages[0].substring(0, 50) + '...',
     });
   }
 
-  // Pruna API expects singular 'image' (not 'images' array)
-  const payload: Record<string, unknown> = {
-    image: imageUrls[0],  // Use first uploaded image
-    prompt,
-    aspect_ratio: aspectRatio,
-  };
-
   if (input.seed !== undefined) payload.seed = input.seed;
   if (input.disable_safety_checker !== undefined) payload.disable_safety_checker = input.disable_safety_checker;
   if (input.width !== undefined) payload.width = input.width;