Update app.py

app.py

@@ -533,7 +533,23 @@ def handle_message(data):
             emit('json_response', response, room=session_id)
         session_store.pop(session_id, None)
 
-        
+import base64
+import numpy as np
+from io import BytesIO
+from PIL import Image
+
+def base64_to_numpy(base64_string):
+    # Decode the base64 string
+    image_data = base64.b64decode(base64_string)
+    
+    # Convert the byte data to a PIL image
+    image = Image.open(BytesIO(image_data))
+    
+    # Convert the PIL image to a NumPy array
+    image_np = np.array(image)
+    
+    return image_np
+    
 @socketio.on('example')
 def handle_message(data):
     img_url = data['img_url']