@umituz/react-native-ai-pruna-provider 1.0.20 → 1.0.22

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@umituz/react-native-ai-pruna-provider",
-  "version": "1.0.20",
+  "version": "1.0.22",
   "description": "Pruna AI provider for React Native - implements IAIProvider interface for unified AI generation",
   "main": "./src/index.ts",
   "types": "./src/index.ts",
@@ -33,6 +33,7 @@
   },
   "peerDependencies": {
     "@umituz/react-native-ai-generation-content": ">=1.70.0",
+    "@umituz/react-native-design-system": ">=1.0.0",
     "expo": ">=54.0.0",
     "react": ">=19.0.0",
     "react-native": ">=0.81.4"

package/src/infrastructure/services/pruna-api-client.ts

@@ -16,12 +16,16 @@ import { PRUNA_BASE_URL, PRUNA_PREDICTIONS_URL, PRUNA_FILES_URL, UPLOAD_CONFIG }
 import { generationLogCollector } from "../utils/log-collector";
 import { detectMimeType } from "../utils/mime-detection.util";
 import { getExtensionForMime } from "../utils/constants/mime.constants";
+import {
+  base64ToTempFile,
+  deleteTempFile,
+} from "@umituz/react-native-design-system/filesystem";
 
 const TAG = 'pruna-api';
 
 /**
  * Upload a base64 file (image or audio) to Pruna's file storage.
- * p-video requires file URLs (not raw base64).
+ * Uses design system filesystem for React Native compatibility.
  * Returns the HTTPS file URL to use in predictions.
  */
 export async function uploadFileToStorage(
@@ -42,70 +46,67 @@ export async function uploadFileToStorage(
 
   generationLogCollector.log(sessionId, TAG, 'Uploading file to Pruna storage...');
 
-  // Strip data URI prefix if present to get raw base64
-  const raw = base64Data.includes('base64,') ? base64Data.split('base64,')[1] : base64Data;
+  // Use design system filesystem to create temp file (React Native compatible)
+  const tempUri = await base64ToTempFile(base64Data);
 
-  // Decode base64 to get binary data
-  let binaryStr: string;
-  try {
-    binaryStr = atob(raw);
-  } catch {
-    throw new Error("Invalid file format. Please provide base64 or a valid URL.");
-  }
-
-  // Convert to byte array for MIME detection
-  const bytes = new Uint8Array(binaryStr.length);
-  for (let i = 0; i < binaryStr.length; i++) {
-    bytes[i] = binaryStr.charCodeAt(i);
+  if (!tempUri) {
+    throw new Error("Failed to create temporary file from base64 data");
   }
 
-  const mime = detectMimeType(bytes);
-  const ext = getExtensionForMime(mime);
+  try {
+    // Fetch the temp file to get a Blob
+    const response = await fetch(tempUri);
+    const blob = await response.blob();
+    generationLogCollector.log(sessionId, TAG, `Temp file created (${blob.size} bytes), uploading to Pruna...`);
 
-  // React Native FormData: use object format with uri (data URI)
-  const formData = new FormData();
-  formData.append('content', {
-    uri: base64Data.startsWith('data:') ? base64Data : `data:${mime};base64,${raw}`,
-    type: mime,
-    name: `upload.${ext}`,
-  } as any);
+    // Create FormData with the blob
+    const formData = new FormData();
+    formData.append('content', blob, 'upload');
 
-  const startTime = Date.now();
+    const startTime = Date.now();
 
-  // Apply timeout to prevent indefinite hangs
-  const uploadController = new AbortController();
-  const timeoutId = setTimeout(() => uploadController.abort(), UPLOAD_CONFIG.timeoutMs);
+    // Apply timeout to prevent indefinite hangs
+    const uploadController = new AbortController();
+    const timeoutId = setTimeout(() => uploadController.abort(), UPLOAD_CONFIG.timeoutMs);
 
-  try {
-    const response = await fetch(PRUNA_FILES_URL, {
-      method: 'POST',
-      headers: { 'apikey': apiKey },
-      body: formData,
-      signal: uploadController.signal,
-    });
+    try {
+      const uploadResponse = await fetch(PRUNA_FILES_URL, {
+        method: 'POST',
+        headers: { 'apikey': apiKey },
+        body: formData,
+        signal: uploadController.signal,
+      });
 
-    if (!response.ok) {
-      const err = await response.json().catch(() => ({ message: response.statusText }));
-      const errorMessage = (err as { message?: string }).message || `File upload error: ${response.status}`;
-      generationLogCollector.error(sessionId, TAG, `File upload failed: ${errorMessage}`);
-      throw new Error(errorMessage);
-    }
+      if (!uploadResponse.ok) {
+        const err = await uploadResponse.json().catch(() => ({ message: uploadResponse.statusText }));
+        const errorMessage = (err as { message?: string }).message || `File upload error: ${uploadResponse.status}`;
+        generationLogCollector.error(sessionId, TAG, `File upload failed: ${errorMessage}`);
+        throw new Error(errorMessage);
+      }
 
-    const data: PrunaFileUploadResponse = await response.json();
-    const fileUrl = data.urls?.get || `${PRUNA_FILES_URL}/${data.id}`;
+      const data: PrunaFileUploadResponse = await uploadResponse.json();
+      const fileUrl = data.urls?.get || `${PRUNA_FILES_URL}/${data.id}`;
 
-    const elapsed = Date.now() - startTime;
-    generationLogCollector.log(sessionId, TAG, `File upload completed in ${elapsed}ms → ${fileUrl}`);
+      const elapsed = Date.now() - startTime;
+      generationLogCollector.log(sessionId, TAG, `File upload completed in ${elapsed}ms → ${fileUrl}`);
 
-    return fileUrl;
-  } catch (error) {
-    if (error instanceof Error && error.name === 'AbortError') {
-      generationLogCollector.error(sessionId, TAG, `File upload timed out after ${UPLOAD_CONFIG.timeoutMs}ms`);
-      throw new Error(`File upload timed out after ${UPLOAD_CONFIG.timeoutMs}ms`);
+      return fileUrl;
+    } catch (error) {
+      if (error instanceof Error && error.name === 'AbortError') {
+        generationLogCollector.error(sessionId, TAG, `File upload timed out after ${UPLOAD_CONFIG.timeoutMs}ms`);
+        throw new Error(`File upload timed out after ${UPLOAD_CONFIG.timeoutMs}ms`);
+      }
+      throw error;
+    } finally {
+      clearTimeout(timeoutId);
     }
-    throw error;
   } finally {
-    clearTimeout(timeoutId);
+    // Cleanup temp file
+    try {
+      await deleteTempFile(tempUri);
+    } catch (cleanupError) {
+      generationLogCollector.warn(sessionId, TAG, `Failed to delete temp file: ${cleanupError}`);
+    }
   }
 }
 

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

@@ -106,7 +106,6 @@ async function buildImageEditInput(
     images: imageUrls,
     prompt,
     aspect_ratio: aspectRatio,
-    reference_image: "1", // Required by Pruna API for p-image-edit
   };
 
   if (input.seed !== undefined) payload.seed = input.seed;