Update app.py

app.py

@@ -697,12 +697,29 @@ def download_image_to_numpy(url):
     else:
         raise Exception(f"Failed to download image. Status code: {response.status_code}")
 
+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']
     message = data['message']
     session_id = request.sid
-    image_array = download_image_to_numpy(img_url)
+    image_array = base64_to_numpy(img_url)
     response = get_answer(image=image_array, message=message, sessionID=request.sid)
     emit('response', response, room=session_id)
     return response