app.py

import streamlit as st 

import tensorflow as tf
from PIL import Image
import numpy as np
import cv2
from Utils import *

model_path="C:/Users/pc/Desktop/Merchant_Landmark_V1_30122021/30122021SavedModels/SavedModel100Epochs"
model = tf.keras.models.load_model(model_path,custom_objects=None)




st.subheader("Upload Merchant Knee View")
image_file = st.file_uploader("Upload Images", type=["dcm"])


examples=["C:/Users/pc/Desktop/Merchant_Landmark_V1_30122021/Dicoms/1.2.392.200036.9107.500.304.423.20111205.94459.10423.dcm"
          ,"C:/Users/pc/Desktop/Merchant_Landmark_V1_30122021/Dicoms/1.2.392.200036.9107.500.304.423.20120225.115452.10423.dcm",
          "C:/Users/pc/Desktop/Merchant_Landmark_V1_30122021/Dicoms/1.2.392.200036.9107.500.304.423.20120509.174735.10423.dcm"]

colx1, colx2, colx3 = st.columns(3)
with colx1:
    st.text("Example  Dicom ")

    if st.button('Example 1'):
        image_file=examples[0]

with colx2:
    st.text("Example  Dicom ")

    if st.button('Example 2'):
        image_file=examples[1]


with colx3:
    st.text("Example  Dicom ")

    if st.button('Example 3'):
        image_file=examples[2]
    

if image_file is not None:
    st.text("Making A Prediction ....")

    
    try:
        data,PatientName,PatientID,SOPInstanceUID,StudyDate,InstitutionAddress,PatientAge,PatientSex=read_dicom(image_file,False,True)
    except:
        data,PatientName,PatientID,SOPInstanceUID,StudyDate,InstitutionAddress,PatientAge,PatientSex=read_dicom(image_file,True,True)
        pass
    


    img = np.copy(data)
    
    #Denoise Image 
    kernel =( np.ones((5,5), dtype=np.float32))
    img2=cv2.morphologyEx(img, cv2.MORPH_OPEN, kernel,iterations=2 )
    img2=cv2.erode(img2,kernel,iterations =2)
    if len(img2.shape)==3:
        img2=img2[:,:,0]
        
    #Threshhold 100- 4096
    ret,thresh = cv2.threshold(img2,100, 4096, cv2.THRESH_BINARY)
    
    #To Thresh uint8 becasue "findContours" doesnt accept uint16
    thresh =((thresh/np.max(thresh))*255).astype('uint8')
    a1,b1=thresh.shape
    #Find Countours
    contours, hierarchy = cv2.findContours(thresh.copy(), cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
    
    #If There is no countour
    if len(contours)==0:
            roi= thresh
        
    else:
        #Get Areas 
        c_area=np.zeros([len(contours)])
        for i in range(len(contours)):
            c_area[i]= cv2.contourArea(contours[i]) 
            
        #Find Max Countour    
        cnts=contours[np.argmax(c_area)]
        x, y, w, h = cv2.boundingRect(cnts)
        
        #Posibble Square 
        roi = croping(data, x, y, w, h)
        
        # Absolute Square 
        roi=modification_cropping(roi)
        
        # Resize to 256x256 with Inter_Nearest
        roi=cv2.resize(roi,(256,256),interpolation=cv2.INTER_NEAREST)
        
    pre=predict(roi,model)
    heatpoint=points_max_value(pre)
    output=put_text_point(roi,heatpoint)
    output,PatellerCongruenceAngle,ParalelTiltAngle=draw_angle(output,heatpoint)
    data_text = {'PatientID': PatientID, 'PatientName': PatientName,
            'Pateller_Congruence_Angle': PatellerCongruenceAngle,
            'Paralel_Tilt_Angle':ParalelTiltAngle,
            'SOP_Instance_UID':SOPInstanceUID,
            "StudyDate" :StudyDate,
            "InstitutionName" :InstitutionAddress,
            "PatientAge" :PatientAge ,
            "PatientSex" :PatientSex,
            }
    
    
    
    col1, col2 = st.columns(2)
    with col1:
        st.text("Original Dicom Image")

        st.image(np.uint8((data/np.max(data)*255)),width=350)

    
    with col2:
        st.text("Predicted Image ")

        st.image(np.uint8(output),width=350)

    
    
    st.write(data_text)