@zodic/shared 0.0.346 → 0.0.348

package/app/api/index.ts

@@ -1,4 +1,5 @@
 import OpenAI from 'openai';
+import { v4 } from 'uuid';
 import {
   BackendBindings,
   BatchInputItem,
@@ -785,35 +786,51 @@ export const Api = (env: BackendBindings) => ({
         'x-api-key': `${env.PIAPI_API_KEY}`,
         'Content-Type': 'application/json',
       };
-    
+
       // Validate image URLs before making the request
-      const validateImageUrl = async (url: string, label: string): Promise<void> => {
+      const validateImageUrl = async (
+        url: string,
+        label: string
+      ): Promise<void> => {
         try {
           const response = await fetch(url, { method: 'HEAD' });
           if (!response.ok) {
-            throw new Error(`${label} is inaccessible: ${response.status} ${response.statusText}`);
+            throw new Error(
+              `${label} is inaccessible: ${response.status} ${response.statusText}`
+            );
           }
           const contentType = response.headers.get('Content-Type');
           if (!contentType || !contentType.startsWith('image/')) {
-            throw new Error(`${label} is not a valid image: Content-Type is ${contentType}`);
+            throw new Error(
+              `${label} is not a valid image: Content-Type is ${contentType}`
+            );
           }
         } catch (err: any) {
           console.error(`Error validating ${label}:`, err.message);
           throw new Error(`Invalid ${label}: ${err.message}`);
         }
       };
-    
+
       // Helper function to resize an image by calling the Express server
-      const resizeImage = async (url: string, label: string): Promise<ArrayBuffer> => {
-        const resizeEndpoint = `https://zodiako-image-mounter.onrender.com/resize-image?url=${encodeURIComponent(url)}`; // Replace with your Express server URL
+      const resizeImage = async (
+        url: string,
+        label: string
+      ): Promise<ArrayBuffer> => {
+        const resizeEndpoint = `https://zodiako-image-mounter.onrender.com/resize-image?url=${encodeURIComponent(
+          url
+        )}`; // Replace with your Express server URL
         try {
           const response = await fetch(resizeEndpoint);
           if (!response.ok) {
-            throw new Error(`${label} resize failed: ${response.status} ${response.statusText}`);
+            throw new Error(
+              `${label} resize failed: ${response.status} ${response.statusText}`
+            );
           }
           const contentType = response.headers.get('Content-Type');
           if (!contentType || !contentType.startsWith('image/')) {
-            throw new Error(`${label} resize response is not an image: Content-Type is ${contentType}`);
+            throw new Error(
+              `${label} resize response is not an image: Content-Type is ${contentType}`
+            );
           }
           return await response.arrayBuffer();
         } catch (err: any) {
@@ -821,34 +838,49 @@ export const Api = (env: BackendBindings) => ({
           throw new Error(`Failed to resize ${label}: ${err.message}`);
         }
       };
-    
+
+      const getKey = (label: string) => {
+        const timestamp = Date.now();
+        const key = `${label.toLowerCase().replace(/ /g, '_')}_${timestamp}.jpg`;
+        return key;
+      };
+
       // Helper function to upload an image to storage and get a public URL
-      const uploadImageToStorage = async (imageData: ArrayBuffer, label: string): Promise<string> => {
-        const fileName = `${label.toLowerCase().replace(/\s+/g, '-')}-${Date.now()}.jpg`;
-        // Replace this with your actual storage service implementation (e.g., Cloudflare R2, AWS S3)
-        // Example for Cloudflare R2:
-        // const r2Response = await env.R2_BUCKET.put(fileName, imageData, {
-        //   httpMetadata: { contentType: 'image/jpeg' },
-        // });
-        // return `https://your-r2-bucket-url/${fileName}`;
-    
-        // For now, we'll throw an error if this isn't implemented
-        throw new Error(`Storage service not implemented for ${label}`);
+      const uploadImageToStorage = async (
+        imageData: ArrayBuffer,
+        label: string
+      ): Promise<string> => {
+        const key = getKey(label);
+        await env.PHOTOS_BUCKET.put(key, imageData);
+        const imageUrl = `https://pub-c4bb4b3e4ccd46b59eed7fa434c59b60.r2.dev/${key}`;
+        return imageUrl;
       };
-    
+
       try {
         // Validate image URLs
         await validateImageUrl(sourceImageUrl, 'Source image URL');
         await validateImageUrl(targetImageUrl, 'Target image URL');
-    
+
         // Resize images by calling the Express server
-        const resizedSourceData = await resizeImage(sourceImageUrl, 'Source image');
-        const resizedTargetData = await resizeImage(targetImageUrl, 'Target image');
-    
+        const resizedSourceData = await resizeImage(
+          sourceImageUrl,
+          'Source image'
+        );
+        const resizedTargetData = await resizeImage(
+          targetImageUrl,
+          'Target image'
+        );
+
         // Upload resized images to storage to get public URLs
-        const resizedSourceUrl = await uploadImageToStorage(resizedSourceData, 'Source image');
-        const resizedTargetUrl = await uploadImageToStorage(resizedTargetData, 'Target image');
-    
+        const resizedSourceUrl = await uploadImageToStorage(
+          resizedSourceData,
+          'user_photo'
+        );
+        const resizedTargetUrl = await uploadImageToStorage(
+          resizedTargetData,
+          'generated_image'
+        );
+
         const body = JSON.stringify({
           model: 'Qubico/image-toolkit',
           type: 'face-swap',
@@ -858,28 +890,34 @@ export const Api = (env: BackendBindings) => ({
           },
           config: {
             webhook_config: {
-              endpoint: 'https://zodic-backend.lucdelbel.workers.dev/api/webhook/faceswap',
+              endpoint:
+                'https://zodic-backend.lucdelbel.workers.dev/api/webhook/faceswap',
               // secret: '',
             },
           },
         });
-    
-        console.log('Sending FaceSwap request:', { resizedSourceUrl, resizedTargetUrl });
-    
+
+        console.log('Sending FaceSwap request:', {
+          resizedSourceUrl,
+          resizedTargetUrl,
+        });
+
         const response = await fetch(endpoint, {
           method: 'POST',
           headers,
           body,
         });
-    
+
         // Log the HTTP status and headers for debugging
         console.log('PiAPI FaceSwap Response Status:', response.status);
-        console.log('PiAPI FaceSwap Response Headers:', [...response.headers.entries()]);
-    
+        console.log('PiAPI FaceSwap Response Headers:', [
+          ...response.headers.entries(),
+        ]);
+
         // Read the response body as text first to inspect it
         const responseText = await response.text();
         console.log('PiAPI FaceSwap Raw Response Body:', responseText);
-    
+
         // Check for the expected 201 Created status
         if (response.status !== 201) {
           let errorData;
@@ -889,32 +927,45 @@ export const Api = (env: BackendBindings) => ({
             errorData = { message: responseText || 'Unknown error' };
           }
           console.error('Error from PiAPI FaceSwap:', errorData);
-          throw new Error(`PiAPI FaceSwap Error: ${response.status} - ${errorData.message || 'Unknown error'}`);
+          throw new Error(
+            `PiAPI FaceSwap Error: ${response.status} - ${
+              errorData.message || 'Unknown error'
+            }`
+          );
         }
-    
+
         // Check if the response body is empty
         if (!responseText) {
           console.error('PiAPI FaceSwap response body is empty');
           throw new Error('PiAPI FaceSwap Error: Empty response body');
         }
-    
+
         // Try to parse the response as JSON
         let data;
         try {
-          data = JSON.parse(responseText) as { task_id: string; message: string };
+          data = JSON.parse(responseText) as {
+            task_id: string;
+            message: string;
+          };
         } catch (err) {
-          console.error('Failed to parse PiAPI FaceSwap response as JSON:', responseText);
+          console.error(
+            'Failed to parse PiAPI FaceSwap response as JSON:',
+            responseText
+          );
           throw new Error('PiAPI FaceSwap Error: Invalid JSON response');
         }
-    
+
         // Validate the response structure
         if (!data.task_id || !data.message) {
-          console.error('PiAPI FaceSwap response missing required fields:', data);
+          console.error(
+            'PiAPI FaceSwap response missing required fields:',
+            data
+          );
           throw new Error('PiAPI FaceSwap Error: Invalid response structure');
         }
-    
+
         console.log('FaceSwap task created successfully:', data);
-    
+
         return {
           taskId: data.task_id,
           message: data.message,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@zodic/shared",
-  "version": "0.0.346",
+  "version": "0.0.348",
   "module": "index.ts",
   "type": "module",
   "publishConfig": {