Normalize float types to uint8 in image_to_base64 function

app.py

@@ -70,6 +70,11 @@ def base64_to_image(base64_str):
     return img
 
 def image_to_base64(image):
+    # Convert float types to uint8
+    if image.dtype == np.float64 or image.dtype == np.float32:
+        # Normalize to 0-255 range and convert to uint8
+        image = np.clip(image * 255 if image.max() <= 1.0 else image, 0, 255).astype(np.uint8)
+    
     # Convert to BGR if it's RGB
     if len(image.shape) > 2 and image.shape[2] == 3:
         image = cv2.cvtColor(image, cv2.COLOR_RGB2BGR)