Update app.py

app.py

@@ -67,7 +67,7 @@ def verify_password(stored_password, provided_password):
 
 def auth():
     if st.session_state.user is None:
-        st.subheader("Welcome to the HeadPeace")
+        st.subheader("Welcome to the HeadPeace 🧠")
         
         # Instructions
         with st.expander("How to Use This App", expanded=True):
@@ -110,6 +110,11 @@ def auth():
             reg_password = st.text_input("Choose a Password", type="password", key="reg_password")
             confirm_password = st.text_input("Confirm Password", type="password", key="confirm_password")
             user_type = st.selectbox("User Type", ["Patient", "Doctor"])
+
+            doctor_code = ""
+            if user_type == "Doctor":
+                doctor_code = st.text_input("Enter Doctor Registration Code", type="password")
+
             register_button = st.button("Register")
             
             if register_button:
@@ -120,6 +125,8 @@ def auth():
                     st.error("Passwords do not match.")
                 elif len(reg_password) < 8:
                     st.error("Password must be at least 8 characters long.")
+                elif user_type == "Doctor" and doctor_code != st.secrets["DOCTOR_CODE"]:
+                    st.error("Invalid doctor registration code.")
                 else:
                     hashed_password = hash_password(reg_password)
                     save_user_data(reg_username, hashed_password, user_type)