import streamlit as st
import cv2
import numpy as np
from PIL import Image
from utils import *

st.title("Medical Document Image Quality Enhancement System")

uploaded_file = st.file_uploader("Upload Medical Image", type=["png","jpg","jpeg"])

if uploaded_file is not None:

    image = Image.open(uploaded_file)
    image = np.array(image)

    st.subheader("Original Image")
    st.image(image, use_column_width=True)

    blur_score = detect_blur(image)
    st.write("Blur Score:", blur_score)

    enhanced = enhance_image(image)

    st.subheader("Enhanced Image")
    st.image(enhanced, use_column_width=True)

    psnr_value = calculate_psnr(image, enhanced)
    ssim_value = calculate_ssim(image, enhanced)

    st.subheader("Quality Metrics")
    st.write("PSNR:", psnr_value)
    st.write("SSIM:", ssim_value)