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(db_firestore, db_realtime, sender, recipient, message):
+def send_message(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(db_firestore, db_realtime, sender, recipient, message):
         return f"Failed to send message: {str(e)}"
 
 # Function to retrieve messages from Firestore
-def retrieve_messages_firestore(db_firestore):
+def retrieve_messages_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(db_firestore):
         st.error(f"Failed to retrieve messages from Firestore: {str(e)}")
 
 # Function to retrieve messages from Realtime Database
-def retrieve_messages_realtime(db_realtime):
+def retrieve_messages_realtime():
     try:
         # Retrieve all documents from the 'messages' collection in Realtime Database
         messages_ref = db_realtime.child("messages").get()
@@ -274,12 +274,12 @@ def chat_ui():
     recipient = st.text_input("Recipient", help="Enter the recipient's email address")
     message = st.text_area("Message", help="Type your message here")
     if st.button("Send"):
-        result = send_message(db_firestore, db_realtime, sender, recipient, message)
+        result = send_message(sender, recipient, message)
         st.write(result)
     st.header("Messages (Firestore)")
-    retrieve_messages_firestore(db_firestore)
+    retrieve_messages_firestore()
     st.header("Messages (Realtime Database)")
-    retrieve_messages_realtime(db_realtime)
+    retrieve_messages_realtime()
 
 # Streamlit interface for recognizing face with optimal ID
 def recognize_face_optimal_ui():