Update app.py

app.py

@@ -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():
+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()
@@ -277,9 +277,9 @@ def chat_ui():
         result = send_message(db_firestore, db_realtime, 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():