Upload 6 files

.gitignore

@@ -0,0 +1,2 @@
+/credentials.py
+/old_data

credentials.py

@@ -0,0 +1,2 @@
+mongo_url = "mongodb+srv://kushagrasharma_13:[email protected]/?retryWrites=true&w=majority&appName=Cluster0"
+password = "Kushagra@13"

medical-app.py

@@ -0,0 +1,234 @@
+import re
+import cv2
+import time
+import bcrypt
+import numpy as np
+import streamlit as st
+import tensorflow as tf
+from pymongo import MongoClient
+from credentials import mongo_url
+
+@st.cache
+def load_data():
+    client = MongoClient(mongo_url)
+    db = client['medical-app-auth']
+    collection = db['login-data']
+    brain_tumor_model = tf.keras.models.load_model('models/brain_tumor.h5')
+    alzheimer_model = tf.keras.models.load_model('models/alzheimer.h5')
+    return collection, brain_tumor_model, alzheimer_model
+
+collection, brain_tumor_model, alzheimer_model = load_data()
+
+if 'current_page' not in st.session_state:
+    st.session_state.current_user = None
+    st.session_state.current_page = 'login'
+    
+def clear_cache():
+        keys = list(st.session_state.keys())
+        for key in keys:
+            st.session_state.pop(key)
+
+def login():
+    st.set_page_config(layout='centered', page_title="Brain MRI", page_icon="mri_of_brain.jpg")
+    col1, col2 = st.columns([5,1])
+    with col2:
+        if st.button("Register", use_container_width=True):
+            st.session_state.current_page = 'register'
+            st.rerun()
+
+    def reset_passowrd_input():
+        st.session_state.password   = ""    
+
+    def reset_username_inputs():
+        st.session_state.username   = ""
+        reset_passowrd_input()
+
+    with st.form(key='login', clear_on_submit=True):
+        st.subheader("Login")
+
+        username = st.text_input("Username", placeholder="Enter Username")
+        password = st.text_input("Enter Password", type="password")
+        username = username.lower().strip()
+
+        user = collection.find_one({"username":username})
+
+        submit = st.form_submit_button("Login")
+        if submit:
+            with st.spinner('Checking credentials...'):
+                if user==None:
+                    st.warning("Username is does not exits, please register")
+                    reset_username_inputs()
+                elif bcrypt.checkpw(password.encode('utf-8'), user.get("password")):
+                    st.warning("**Credential Matched**: Redirecting...") 
+                    st.session_state.current_user = username
+                    st.session_state.current_page = 'medical'
+                    st.rerun()
+                else:
+                    st.warning("Password is incorrect")
+                    reset_passowrd_input()
+
+def register():
+    st.set_page_config(layout='centered', page_title="Brain MRI", page_icon="mri_of_brain.jpg")
+    col1, col2 = st.columns([5,1])
+    with col2:
+        if st.button("Login", use_container_width=True):
+            st.session_state.current_page = 'login'
+            st.rerun()
+
+    document = {}
+    pattern = r'^[a-zA-Z0-9._%+-]+@[a-zA-Z0-9.-]+\.[a-zA-Z]{2,}$'
+
+    def reset_password_inputs():
+        st.session_state.password1  = ""
+        st.session_state.password2  = ""
+
+    def reset_username_inputs():
+        st.session_state.username   = ""
+        reset_password_inputs()
+
+    def reset_email_inputs():
+        st.session_state.email      = ""
+        reset_username_inputs()
+
+    with st.form(key='register', clear_on_submit=True):
+        st.subheader("Register")
+
+        email = st.text_input("Email", placeholder="[email protected]", max_chars=64)
+        username = st.text_input("Username", placeholder="Enter Username", max_chars=17)
+        name = st.text_input("Name", placeholder="Enter Your Full Name")
+        password1 = st.text_input("Enter Password", type="password", max_chars=17)
+        password2 = st.text_input("Confirm Password", type="password", max_chars=17)
+        
+        email = email.lower().strip()
+        username = username.lower().strip()
+        name = name.title().strip()
+        
+        submit = st.form_submit_button("Register")
+        if submit:
+            with st.spinner('Checking credentials...'):
+                if re.match(pattern, email)==None:
+                    st.warning(":red[Please provide a valid email.]", icon="⚠️")
+                    reset_email_inputs()
+                elif collection.find_one({"email":email}):
+                    st.warning("Email already exits, please Login!")
+                    reset_email_inputs()
+                elif not name.replace(" ", "").isalpha():
+                    st.warning("Don't use number or special characters for name")
+                    st.session_state.name = ""
+                elif len(username)<5:
+                    st.warning("Username must atleast be of 5 characters")
+                    reset_username_inputs()
+                elif len(password1)<6:
+                    st.warning("Password must atleast be of 6 characters")
+                    reset_password_inputs()
+                elif collection.find_one({"username": username}):
+                    st.warning("Username already exits, please try a different one.")
+                    reset_username_inputs()
+                elif password1 != password2:
+                    st.warning(":red[Passwords do not match. Please try again.]", icon="⚠️")
+                    reset_password_inputs()
+                elif not (email and name and username and password1 and password2):
+                    st.warning(":red[Please complete all the fields above.]", icon="⚠️")
+                else:
+                    salt = bcrypt.gensalt(rounds=13)
+                    hashed_password = bcrypt.hashpw(password1.encode('utf-8'), salt)
+                    salt = bcrypt.gensalt(rounds=13)
+                    hashed_password = bcrypt.hashpw(password1.encode('utf-8'), salt)
+                    document = {"name":name, "username":username, "email":email, "password":hashed_password, "salt":salt}
+                    collection.insert_one(document)
+                    st.warning("**Successfully Registered**: Redirecting...")
+                    st.session_state.current_user = username
+                    st.session_state.current_page = 'medical'
+                    st.rerun()
+
+
+
+def medical_page():
+    def alzheimer():
+        col1, col2, col3 = st.columns([6, 6, 1])
+        with col1:
+            if st.button(f"Welcome! {st.session_state.current_user}"):
+                st.rerun()
+        with col3:
+            st.button("Logout", use_container_width=True, on_click=clear_cache)
+                
+        st.markdown("***")         
+        st.subheader("Here's your Alzheimer's Scan")
+        
+        uploaded_file = st.file_uploader("Upload MRI scan image for detecting alzheimer's", type=['png', 'jpg'])
+        if uploaded_file is not None:
+            file_bytes = np.asarray(bytearray(uploaded_file.read()), dtype=np.uint8)
+            opencv_image = cv2.imdecode(file_bytes, cv2.IMREAD_COLOR)
+            resized_img = cv2.resize(opencv_image, (208, 176))
+            resized_img = resized_img[np.newaxis, ...]
+            class_names = ['ModerateDemented', 'NonDemented', 'VeryMildDemented', 'MildDemented']
+            
+            pred = class_names[np.argmax(st.session_state.alzheimer_model.predict(resized_img))]
+            st.text(pred)
+
+    def brain_tumor():
+        col1, col2, col3 = st.columns([6, 6, 1])
+        with col1:
+            if st.button(f"Welcome! {st.session_state.current_user}"):
+                st.rerun()
+        with col3:
+            st.button("Logout", use_container_width=True, on_click=clear_cache)
+                
+        st.markdown("***")        
+        st.subheader("Here's your Brain Tumor Scan")
+        
+        uploaded_file = st.file_uploader("Upload MRI scan image for detecting Brain Tumor", type=['png', 'jpg'])
+        if uploaded_file is not None:
+            file_bytes = np.asarray(bytearray(uploaded_file.read()), dtype=np.uint8)
+            opencv_image = cv2.imdecode(file_bytes, cv2.IMREAD_COLOR)
+            resized_img = cv2.resize(opencv_image, (168, 150))
+            resized_img = resized_img[np.newaxis, ...]
+            class_names = ['Pituitary', 'No-Tumor', 'Meningioma', 'Glioma']
+            
+            pred = class_names[np.argmax(st.session_state.brain_tumor_model.predict(resized_img))]
+            st.text(pred)
+
+    def abscesses():
+        col1, col2, col3 = st.columns([6, 6, 1])
+        with col1:
+            if st.button(f"Welcome! {st.session_state.current_user}"):
+                st.rerun()
+        with col3:
+            st.button("Logout", use_container_width=True, on_click=clear_cache)
+        st.markdown("***")        
+        st.subheader("Here's your Abscesses Scan")
+        uploaded_file = st.file_uploader("Upload MRI scan image for detecting Abscesses", type=['png', 'jpg'], disabled=True)
+        st.write("Feature currently unavailable")
+        if uploaded_file is not None:
+            file_bytes = np.asarray(bytearray(uploaded_file.read()), dtype=np.uint8)
+            opencv_image = cv2.imdecode(file_bytes, cv2.IMREAD_COLOR)
+            resized_img = cv2.resize(opencv_image, (208, 176))
+            resized_img = resized_img[np.newaxis, ...]
+            class_names = ['ModerateDemented', 'NonDemented', 'VeryMildDemented', 'MildDemented']
+            pred = class_names[np.argmax(abscesses_model.predict(resized_img))]
+            st.text(pred)
+    st.set_page_config(layout='wide', page_title="Brain MRI", page_icon="mri_of_brain.jpg")
+    st.sidebar.image("mri_of_brain.jpg")
+    st.sidebar.title("Navigation")
+    page_options = ["Alzheimer", "Brain Tumor", "Abscesses"]
+    selected_page = st.sidebar.selectbox("Select a Scan", page_options)
+
+    if selected_page == "Alzheimer":
+        alzheimer()
+    elif selected_page == "Brain Tumor":
+        brain_tumor()
+    elif selected_page == "Abscesses":
+        abscesses()
+
+
+def main():
+    if st.session_state.current_page == 'login':
+        login()
+    elif st.session_state.current_page == 'register':
+        register()
+    elif st.session_state.current_page == 'medical':
+        medical_page()
+
+if __name__ == "__main__":
+
+    main()

requirements.txt

@@ -0,0 +1,12 @@
+pip==24.0
+numpy==1.26.4
+keras==2.15.0
+altair==5.2.0
+pandas==2.2.1
+pydeck==0.8.0
+streamlit==1.32.0
+tensorflow==2.15.0
+opencv-python-headless==4.9.0.80
+bcrypt==4.1.2
+pymongo==3.11.0
+dnspython==1.16.0

utils.py

@@ -0,0 +1,28 @@
+# Copyright (c) Streamlit Inc. (2018-2022) Snowflake Inc. (2022)
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import inspect
+import textwrap
+
+import streamlit as st
+
+
+def show_code(demo):
+    """Showing the code of the demo."""
+    show_code = st.sidebar.checkbox("Show code", True)
+    if show_code:
+        # Showing the code of the demo.
+        st.markdown("## Code")
+        sourcelines, _ = inspect.getsourcelines(demo)
+        st.code(textwrap.dedent("".join(sourcelines[1:])))