Update app.py

app.py

@@ -335,6 +335,10 @@ def get_messages(user_email):
 def messaging_ui():
     st.title("💬 Messaging")
 
+    # Initialize session variable for tracking new messages
+    if "new_messages" not in st.session_state:
+        st.session_state.new_messages = 0
+
     if st.session_state.auth_state["signed_in"]:
         sender_email = st.session_state.auth_state["user"].email
         receiver_email = st.text_input("Receiver's Email", help="Enter the receiver's email address")
@@ -343,6 +347,17 @@ def messaging_ui():
         if st.button("Send Message"):
             result = send_message(sender_email, receiver_email, message_content)
             st.write(result)
+            # Increment new_messages when a message is sent
+            st.session_state.new_messages += 1
+
+        # Get the current number of new messages
+        new_messages = st.session_state.new_messages
+
+        # Update the messaging title to show the number of new messages
+        title = "💬 Messaging"
+        if new_messages > 0:
+            title += f" ({new_messages})"
+        st.title(title)
 
         messages = get_messages(sender_email)
         if messages: