Update app.py

app.py

@@ -18,13 +18,15 @@ def musica_enhancement(image):
     print(f"Uploaded image shape: {img.shape}, dtype: {img.dtype}")
     print(f"Image min: {img.min()}, max: {img.max()}")
 
-    # Check if the image is grayscale
-    if len(img.shape) != 2:
-        raise ValueError("Uploaded image must be a grayscale image.")
+    # Convert RGB to grayscale if necessary
+    if len(img.shape) == 3 and img.shape[2] == 3:  # Check for RGB
+        img = cv2.cvtColor(img, cv2.COLOR_RGB2GRAY)
+        print("Converted RGB image to grayscale.")
 
     # Ensure the image has 16-bit depth
     if img.dtype != np.uint16:
-        raise ValueError("Uploaded image must be a 16-bit TIFF image.")
+        img = img.astype(np.uint16)  # Scale if necessary
+        print("Converted image to 16-bit.")
 
     # Normalize to [0, 1]
     img_norm = img.astype(np.float32) / 65535.0
@@ -50,7 +52,6 @@ def musica_enhancement(image):
 
     # Convert to 8-bit
     img_gamma_8bit = (img_gamma * 255).astype(np.uint8)
-    print(f"Final image min: {img_gamma_8bit.min()}, max: {img_gamma_8bit.max()}")
 
     # Convert to RGB for better viewing compatibility
     img_rgb = cv2.cvtColor(img_gamma_8bit, cv2.COLOR_GRAY2RGB)
@@ -72,6 +73,8 @@ def musica_enhancement(image):
     return enhanced_image, histogram
 
 
+
+
 # Define Gradio interface
 with gr.Blocks() as demo:
     gr.Markdown("# Musica Image Enhancement")