Update app.py

app.py

@@ -265,6 +265,49 @@ def feedback_ui():
             send_feedback(feedback_data)
             st.success("Feedback submitted successfully! Thank you for your feedback.")
 
+#section for messaging 
+# Function to send a message
+def send_message(sender_email, receiver_email, message_content):
+    try:
+        db_firestore.collection('messages').add({
+            'sender_email': sender_email,
+            'receiver_email': receiver_email,
+            'message_content': message_content,
+            'timestamp': firestore.SERVER_TIMESTAMP
+        })
+        return "Message sent successfully."
+    except Exception as e:
+        return f"Failed to send message: {str(e)}"
+
+# Function to retrieve messages for a user
+def get_messages(user_email):
+    try:
+        messages = db_firestore.collection('messages').where('receiver_email', '==', user_email).order_by('timestamp').get()
+        return messages
+    except Exception as e:
+        return f"Failed to retrieve messages: {str(e)}"
+
+# Streamlit interface for sending messages
+def send_message_ui():
+    st.title("Send Message")
+    sender_email = st.session_state.auth_state["user"]["email"]
+    receiver_email = st.selectbox("Receiver's Email", [user["email"] for user in get_all_users()])
+    message_content = st.text_area("Message Content")
+    if st.button("Send Message"):
+        result = send_message(sender_email, receiver_email, message_content)
+        st.write(result)
+
+# Streamlit interface for viewing messages
+def view_messages_ui():
+    st.title("Messages")
+    user_email = st.session_state.auth_state["user"]["email"]
+    messages = get_messages(user_email)
+    for message in messages:
+        st.write(f"From: {message['sender_email']}")
+        st.write(f"Message: {message['message_content']}")
+        st.write("---")
+#end of messaging section 
+
 def tour_guide_ui():
     st.title("Tour Guide")
     st.markdown("This tour will guide you through the application.")
@@ -381,7 +424,7 @@ def display_tour_steps(steps):
 # Update the main function to include the feedback option
 def main():
     st.sidebar.title("Options")
-    option = st.sidebar.radio("Select Option", ["Add Person", "Recognize Face", "Delete Person", "Recognize Face (Optimal)", "Tour Guide", "Feedback"])
+    option = st.sidebar.radio("Select Option", ["Add Person", "Recognize Face", "Delete Person", "Recognize Face (Optimal)", "Tour Guide", "Feedback", "Send Message","Messages"])
     
     if option == "Add Person":
         add_person_ui()
@@ -395,6 +438,10 @@ def main():
         tour_guide_ui()
     elif option == "Feedback":
         feedback_ui()
+    elif option == "Send Message":
+        send_message_ui()
+    elif option == "Messages":
+        view_messages_ui()
 
 # Run the tour guide
 if __name__ == "__main__":