Update app.py

app.py

@@ -4,6 +4,9 @@ from tensorflow.keras.preprocessing import image
 import numpy as np
 from PIL import Image
 import base64
+import cv2
+import numpy as np
+import matplotlib.pyplot as plt
 
 H = 256
 W = 256
@@ -36,16 +39,52 @@ custom_style = """
 st.markdown(custom_style, unsafe_allow_html=True)
 
 
+# Function to perform inference
+def perform_inference(img):
+    image = cv2.imread(img, cv2.IMREAD_COLOR)
+    original_shape = image.shape[:2]
+    original_image = image.copy()
+    image = cv2.resize(image, (W, H))
+    image = image/255.0
+    image = np.expand_dims(image, axis=0)
+
+    mask = model.predict(image, verbose=0)[0]
+    mask = cv2.resize(mask, (original_shape[1], original_shape[0]))
+    mask = mask >= 0.5
+    mask = np.expand_dims(mask, axis=-1)
+    segmented_image = original_image * mask
+    return original_image, mask, segmented_image
+
+# Function to display images using Matplotlib
+def show_image(image, title="Image"):
+    plt.imshow(image, cmap='gray')
+    plt.title(title)
+    plt.axis('off')
+    st.pyplot()
+
+# Streamlit app
 def main():
-    st.title("Brain Tumor Segmentation")
+    st.title("Brain Tumor Segmentation App")
 
-    uploaded_file = st.file_uploader("Upload an MRI image for tumor segmentation...", type=["jpg", "png", "jpeg"])
+    # Allow user to upload an image
+    uploaded_file = st.file_uploader("Upload a brain tumor image...", type=["jpg", "png", "jpeg"])
 
     if uploaded_file is not None:
-        original_image = Image.open(uploaded_file)
-        st.image(original_image, caption="Uploaded Image", use_column_width=True)
+        # Perform inference on the uploaded image
+        original_image, mask, segmented_image = perform_inference(uploaded_file)
 
-        st.markdown("## Tumor Segmentation Result")
+        # Display original image
+        st.subheader("Original Image")
+        show_image(original_image, title="Original Image")
+
+        # Display mask
+        st.subheader("Mask")
+        show_image(mask, title="Mask")
+
+        # Display segmented image
+        st.subheader("Segmented Brain Tumor")
+        show_image(segmented_image, title="Segmented Brain Tumor")
 
 if __name__ == "__main__":
     main()
+