Update app.py

app.py

@@ -3,8 +3,12 @@ import numpy as np
 import gradio as gr
 import pywt
 from skimage import exposure
+from PIL import Image
 
 def musica_enhancement(img):
+    # Convert PIL Image to NumPy array
+    img = np.array(img)
+    
     # Convert from RGB to grayscale if needed
     if len(img.shape) == 3:
         img = cv2.cvtColor(img, cv2.COLOR_RGB2GRAY)
@@ -46,20 +50,27 @@ def musica_enhancement(img):
     return img_sharp
 
 def process_image(input_img):
+    # Convert Gradio input (PIL Image) to NumPy array
+    input_img = np.array(input_img)
+    
+    # Enhance the image
     enhanced_img = musica_enhancement(input_img)
+    
+    # Convert enhanced image back to PIL for display
+    enhanced_img = Image.fromarray(enhanced_img)
+    
     return input_img, enhanced_img
 
 # Gradio Interface
 demo = gr.Interface(
     fn=process_image,
-    inputs=gr.Image(label="Upload X-ray"),
+    inputs=gr.Image(label="Upload X-ray (TIFF or JPEG/PNG)"),
     outputs=[
         gr.Image(label="Original Image"), 
         gr.Image(label="Enhanced Image")
     ],
-    examples=[["./sample_xray.jpg"]],
     title="X-ray Enhancement (MUSICA®-Style)",
-    description="Medical X-ray enhancement using multi-scale wavelet processing and adaptive contrast adjustment"
+    description="Upload a TIFF, JPEG, or PNG X-ray image to enhance it using multi-scale wavelet processing and adaptive contrast adjustment."
 )
 
 demo.launch()