Update app.py

app.py

@@ -350,6 +350,13 @@ def messaging_ui():
             # Increment new_messages when a message is sent
             st.session_state.new_messages += 1
 
+        # Check for new messages when the page is loaded
+        messages = get_messages(sender_email)
+        if messages:
+            new_messages_count = sum(1 for _ in messages)
+            # Increment new_messages count
+            st.session_state.new_messages += new_messages_count
+
         # Get the current number of new messages
         new_messages = st.session_state.new_messages
 
@@ -359,7 +366,6 @@ def messaging_ui():
             title += f" ({new_messages})"
         st.title(title)
 
-        messages = get_messages(sender_email)
         if messages:
             st.write("Your messages:")
             for message in messages: