Update app.py

app.py

@@ -65,7 +65,7 @@ def create_user(email, password):
         return False, None
 
 # Function to send a message
-def send_message(sender, recipient, message):
+def send_message(db_firestore, db_realtime, sender, recipient, message):
     try:
         # Add a new document with a generated ID to Firestore
         db_firestore.collection("messages").add({
@@ -88,7 +88,7 @@ def send_message(sender, recipient, message):
         return f"Failed to send message: {str(e)}"
 
 # Function to retrieve messages from Firestore
-def retrieve_messages_firestore():
+def retrieve_messages_firestore(db_firestore):
     try:
         # Retrieve all documents from the 'messages' collection in Firestore
         messages_ref = db_firestore.collection("messages").order_by("timestamp")
@@ -100,7 +100,7 @@ def retrieve_messages_firestore():
         st.error(f"Failed to retrieve messages from Firestore: {str(e)}")
 
 # Function to retrieve messages from Realtime Database
-def retrieve_messages_realtime():
+def retrieve_messages_realtime(db_realtime):
     try:
         # Retrieve all documents from the 'messages' collection in Realtime Database
         messages_ref = db_realtime.child("messages").get()
@@ -266,9 +266,9 @@ def recognize_face_ui():
     if st.button("Recognize Face"):
         result = recognize_face(image_path)
         st.write(result, unsafe_allow_html=True)
-
+        
 # Streamlit interface for chat (Retrieve messages from both Firestore and Realtime Database)
-def chat_ui():
+def chat_ui(db_firestore, db_realtime):
     st.title("Chat")
     sender = st.session_state.auth_state["user"]["email"]
     recipient = st.text_input("Recipient", help="Enter the recipient's email address")
@@ -277,9 +277,9 @@ def chat_ui():
         result = send_message(sender, recipient, message)
         st.write(result)
     st.header("Messages (Firestore)")
-    retrieve_messages_firestore()
+    retrieve_messages_firestore(db_firestore)
     st.header("Messages (Realtime Database)")
-    retrieve_messages_realtime()
+    retrieve_messages_realtime(db_realtime)
 
 # Streamlit interface for recognizing face with optimal ID
 def recognize_face_optimal_ui():
@@ -367,4 +367,4 @@ def main():
         chat_ui()
 
 if __name__ == "__main__":
-    authenticate_user_ui()
+     authenticate_user_ui(db_firestore, db_realtime)