avoid hash collision between users using the same image

client/src/hooks/useMainStore.ts

@@ -18,7 +18,7 @@ export type ImageState = ImageStateValues & {
   setIsFollowingCursor: (isFollowingCursor: boolean) => void
   setIsGazingAtCursor: (isGazingAtCursor: boolean) => void
   setOriginalImage: (url: string) => void
-  setOriginalImageHash: (hash: string) => void
+  setOriginalImageUuid: (uuid: string) => void
   setPreviewImage: (url: string) => void
   resetImage: () => void
   setAverageLatency: (averageLatency: number) => void
@@ -41,7 +41,7 @@ export const getDefaultState = (): ImageStateValues => ({
   isFollowingCursor: false,
   isGazingAtCursor: false,
   originalImage: '',
-  originalImageHash: '',
+  originalImageUuid: '',
   previewImage: '',
   minLatency: 20, // min time between requests
   averageLatency: 190, // this should be the average for most people
@@ -98,7 +98,7 @@ export const useMainStore = create<ImageState>((set, get) => ({
   setIsFollowingCursor: (isFollowingCursor: boolean) => set({ isFollowingCursor }),
   setIsGazingAtCursor: (isGazingAtCursor: boolean) => set({ isGazingAtCursor }),
   setOriginalImage: (url) => set({ originalImage: url }),
-  setOriginalImageHash: (originalImageHash) => set({ originalImageHash }),
+  setOriginalImageUuid: (originalImageUuid) => set({ originalImageUuid }),
   setPreviewImage: (url) => set({ previewImage: url }),
   resetImage: () => {
     const { originalImage } = get()
@@ -121,11 +121,11 @@ export const useMainStore = create<ImageState>((set, get) => ({
     }})
   },
   handleServerResponse: async (params: OnServerResponseParams) => {
-    const { originalImage, setMetadata, setPreviewImage, setOriginalImageHash, applyModifiedHeadToCanvas, modifyImage } = useMainStore.getState();
+    const { originalImage, setMetadata, setPreviewImage, setOriginalImageUuid, applyModifiedHeadToCanvas, modifyImage } = useMainStore.getState();
     if (typeof params.error === "string") {
       console.error(`handleServerResponse: failed to perform the request, resetting the app (${params.error})`)
       setPreviewImage(originalImage)
-      setOriginalImageHash('')
+      setOriginalImageUuid('')
     } else if (typeof params.image !== "undefined") {
 
       // this is where we decide to paste back the image as a whole,
@@ -141,7 +141,7 @@ export const useMainStore = create<ImageState>((set, get) => ({
       setPreviewImage(image);
     } else if (typeof params.loaded !== "undefined") {
       //console.log(`handleServerResponse: received a json`, params)
-      setOriginalImageHash(params.loaded.h)
+      setOriginalImageUuid(params.loaded.u)
       setMetadata({
         center: params.loaded.c, // center - 2x1
         size: params.loaded.s, // size - scalar
@@ -149,7 +149,7 @@ export const useMainStore = create<ImageState>((set, get) => ({
         angle: params.loaded.a,  //angle - rad, counterclockwise
       })
 
-      // right after we received the hash, we perform a first blank request
+      // right after we received the id, we perform a first blank request
       await modifyImage({
         landmark: {
           group: 'background',
@@ -264,7 +264,7 @@ export const useMainStore = create<ImageState>((set, get) => ({
 
     const {
       originalImage,
-      originalImageHash,
+      originalImageUuid,
       params: previousParams,
       setParams,
       setError,
@@ -424,8 +424,8 @@ export const useMainStore = create<ImageState>((set, get) => ({
 
     try {
 
-      if (originalImageHash) {
-        facePoke.transformImage(originalImageHash, params);
+      if (originalImageUuid) {
+        facePoke.transformImage(originalImageUuid, params);
       }
 
     } catch (error) {

client/src/lib/facePoke.ts

@@ -135,8 +135,8 @@ export class FacePoke {
     this.sendBlobMessage(await blob.arrayBuffer());
   }
 
-  public transformImage(hash: string, params: Partial<ImageModificationParams>): void {
-    this.sendJsonMessage({ hash, params });
+  public transformImage(uuid: string, params: Partial<ImageModificationParams>): void {
+    this.sendJsonMessage({ uuid, params });
   }
 
   private sendBlobMessage(buffer: ArrayBuffer): void {

client/src/types.ts

@@ -30,7 +30,7 @@ export interface Metadata {
  */
 export interface ModifyImageMessage {
   image?: string;
-  hash?: string;
+  uuid?: string;
   params: Partial<ImageModificationParams>;
 }
 
@@ -38,7 +38,7 @@ export type OnServerResponseParams = {
   image?: Blob
   error?: string
   loaded?: {
-    h: string
+    i: string
   } & {
     c: number[] //center - 2x1
     s: number // size - scalar
@@ -80,7 +80,7 @@ export interface ImageStateValues {
   isGazingAtCursor: boolean
   originalImage: string
   previewImage: string
-  originalImageHash: string
+  originalImageUuid: string
   minLatency: number
   averageLatency: number
   maxLatency: number

engine.py

@@ -1,3 +1,4 @@
+import uuid
 import logging
 import hashlib
 import os
@@ -61,37 +62,10 @@ class Engine:
 
         logger.info("✅ FacePoke Engine initialized successfully.")
 
-    async def get_image_hash(self, image: Union[Image.Image, str, bytes]) -> str:
-        """
-        Compute or retrieve the hash for an image.
-
-        Args:
-            image (Union[Image.Image, str, bytes]): The input image, either as a PIL Image,
-                base64 string, or bytes.
-
-        Returns:
-            str: The computed hash of the image.
-        """
-        if isinstance(image, str):
-            # Assume it's already a hash if it's a string of the right length
-            if len(image) == 32:
-                return image
-            # Otherwise, assume it's a base64 string
-            image = base64_data_uri_to_PIL_Image(image)
-
-        if isinstance(image, Image.Image):
-            return hashlib.md5(image.tobytes()).hexdigest()
-        elif isinstance(image, bytes):
-            return hashlib.md5(image).hexdigest()
-        else:
-            raise ValueError("Unsupported image type")
-
     @alru_cache(maxsize=512)
     async def load_image(self, data):
         image = Image.open(io.BytesIO(data))
-
-        image_hash = await self.get_image_hash(image)
-
+        uuid = uuid.uuid4()
         img_rgb = np.array(image)
 
         inference_cfg = self.live_portrait.live_portrait_wrapper.cfg
@@ -113,13 +87,13 @@ class Engine:
             'inference_cfg': inference_cfg
         }
 
-        self.processed_cache[image_hash] = processed_data
+        self.processed_cache[uuid] = processed_data
 
         # Calculate the bounding box
         bbox_info = parse_bbox_from_landmark(processed_data['crop_info']['lmk_crop'], scale=1.0)
 
         return {
-            'h': image_hash,
+            'u': uuid,
 
             # those aren't easy to serialize
             'c': bbox_info['center'], # 2x1
@@ -129,12 +103,12 @@ class Engine:
             # 'bbox_rot': bbox_info['bbox_rot'].toList(),  # 4x2
         }
 
-    async def transform_image(self, image_hash: str, params: Dict[str, float]) -> bytes:
+    async def transform_image(self, uuid: str, params: Dict[str, float]) -> bytes:
         # If we don't have the image in cache yet, add it
-        if image_hash not in self.processed_cache:
+        if uuid not in self.processed_cache:
             raise ValueError("cache miss")
 
-        processed_data = self.processed_cache[image_hash]
+        processed_data = self.processed_cache[uuid]
 
         try:
             # Apply modifications based on params

public/index.js

@@ -29754,8 +29754,8 @@ class FacePoke {
     const blob = new Blob([buffer], { type: "application/octet-binary" });
     this.sendBlobMessage(await blob.arrayBuffer());
   }
-  transformImage(hash, params) {
-    this.sendJsonMessage({ hash, params });
+  transformImage(uuid, params) {
+    this.sendJsonMessage({ uuid, params });
   }
   sendBlobMessage(buffer) {
     if (!this.ws || this.ws.readyState !== WebSocketState.OPEN) {
@@ -29888,7 +29888,7 @@ var getDefaultState = () => ({
   isFollowingCursor: false,
   isGazingAtCursor: false,
   originalImage: "",
-  originalImageHash: "",
+  originalImageUuid: "",
   previewImage: "",
   minLatency: 20,
   averageLatency: 190,
@@ -29943,7 +29943,7 @@ var useMainStore = create((set, get) => ({
   setIsFollowingCursor: (isFollowingCursor) => set({ isFollowingCursor }),
   setIsGazingAtCursor: (isGazingAtCursor) => set({ isGazingAtCursor }),
   setOriginalImage: (url) => set({ originalImage: url }),
-  setOriginalImageHash: (originalImageHash) => set({ originalImageHash }),
+  setOriginalImageUuid: (originalImageUuid) => set({ originalImageUuid }),
   setPreviewImage: (url) => set({ previewImage: url }),
   resetImage: () => {
     const { originalImage } = get();
@@ -29966,16 +29966,16 @@ var useMainStore = create((set, get) => ({
     } });
   },
   handleServerResponse: async (params) => {
-    const { originalImage, setMetadata, setPreviewImage, setOriginalImageHash, applyModifiedHeadToCanvas, modifyImage } = useMainStore.getState();
+    const { originalImage, setMetadata, setPreviewImage, setOriginalImageUuid, applyModifiedHeadToCanvas, modifyImage } = useMainStore.getState();
     if (typeof params.error === "string") {
       console.error(`handleServerResponse: failed to perform the request, resetting the app (${params.error})`);
       setPreviewImage(originalImage);
-      setOriginalImageHash("");
+      setOriginalImageUuid("");
     } else if (typeof params.image !== "undefined") {
       const image = await convertImageToBase64(params.image);
       setPreviewImage(image);
     } else if (typeof params.loaded !== "undefined") {
-      setOriginalImageHash(params.loaded.h);
+      setOriginalImageUuid(params.loaded.u);
       setMetadata({
         center: params.loaded.c,
         size: params.loaded.s,
@@ -30041,7 +30041,7 @@ var useMainStore = create((set, get) => ({
   modifyImage: async ({ landmark, vector, mode }) => {
     const {
       originalImage,
-      originalImageHash,
+      originalImageUuid,
       params: previousParams,
       setParams,
       setError,
@@ -30139,8 +30139,8 @@ var useMainStore = create((set, get) => ({
     }
     setParams(params);
     try {
-      if (originalImageHash) {
-        facePoke.transformImage(originalImageHash, params);
+      if (originalImageUuid) {
+        facePoke.transformImage(originalImageUuid, params);
       }
     } catch (error) {
       setError("Failed to modify image");