Update app.py

app.py

@@ -62,15 +62,20 @@ def process_image(input_img):
     return input_img, enhanced_img
 
 # Gradio Interface
-demo = gr.Interface(
-    fn=process_image,
-    inputs=gr.Image(label="Upload X-ray (TIFF or JPEG/PNG)"),
-    outputs=[
-        gr.Image(label="Original Image"), 
-        gr.Image(label="Enhanced Image")
-    ],
-    title="X-ray Enhancement (MUSICA®-Style)",
-    description="Upload a TIFF, JPEG, or PNG X-ray image to enhance it using multi-scale wavelet processing and adaptive contrast adjustment."
-)
+with gr.Blocks() as demo:
+    gr.Markdown("# X-ray Enhancement (MUSICA®-Style)")
+    gr.Markdown("Upload a TIFF, JPEG, or PNG X-ray image to enhance it using multi-scale wavelet processing and adaptive contrast adjustment.")
+    
+    with gr.Row():
+        input_image = gr.Image(label="Upload X-ray (TIFF or JPEG/PNG)")
+        output_original = gr.Image(label="Original Image")
+        output_enhanced = gr.Image(label="Enhanced Image")
+    
+    submit_button = gr.Button("Enhance")
+    submit_button.click(
+        fn=process_image,
+        inputs=input_image,
+        outputs=[output_original, output_enhanced]
+    )
 
 demo.launch()