Update app.py

app.py

@@ -184,6 +184,41 @@ def delete_person(name):
     except Exception as e:
         return f"Failed to delete person: {str(e)}"
 
+# Function to send message to Firebase Realtime Database
+def send_message(sender_email, receiver_email, message):
+    try:
+        sender_ref = messages_ref.child(sender_email).child(receiver_email)
+        receiver_ref = messages_ref.child(receiver_email).child(sender_email)
+        
+        sender_ref.push().set({
+            'sender_email': sender_email,
+            'message': message
+        })
+        
+        receiver_ref.push().set({
+            'sender_email': sender_email,
+            'message': message
+        })
+        
+        return True
+    except Exception as e:
+        print(f"Failed to send message: {str(e)}")
+        return False
+
+# Function to get messages from Firebase Realtime Database
+def get_messages(sender_email, receiver_email):
+    try:
+        sender_ref = messages_ref.child(sender_email).child(receiver_email)
+        receiver_ref = messages_ref.child(receiver_email).child(sender_email)
+        
+        sender_messages = sender_ref.get()
+        receiver_messages = receiver_ref.get()
+        
+        return sender_messages, receiver_messages
+    except Exception as e:
+        print(f"Failed to get messages: {str(e)}")
+        return None, None
+
 # Streamlit interface for adding a person
 def add_person_ui():
     st.title("Add Person")
@@ -272,41 +307,12 @@ def logout():
     st.session_state.auth_state["user"] = None
     st.session_state.auth_state["signed_in"] = False
 
-# Function to send message to Firebase Realtime Database
-def send_message(sender_email, receiver_email, message):
-    try:
-        sender_ref = messages_ref.child(sender_email).child(receiver_email)
-        receiver_ref = messages_ref.child(receiver_email).child(sender_email)
-        
-        sender_ref.push().set({
-            'sender_email': sender_email,
-            'message': message
-        })
-        
-        receiver_ref.push().set({
-            'sender_email': sender_email,
-            'message': message
-        })
-        
-        return True
-    except Exception as e:
-        print(f"Failed to send message: {str(e)}")
-        return False
-
-# Function to get messages from Firebase Realtime Database
-def get_messages(sender_email, receiver_email):
-    try:
-        messages = messages_ref.child(sender_email).child(receiver_email).get()
-        return messages
-    except Exception as e:
-        print(f"Failed to get messages: {str(e)}")
-        return None
-
 # Streamlit interface for chat
-def chat_ui(sender_email, receiver_email):
+def chat_ui():
     st.title("Chat")
-    message = st.text_input("Type your message", help="Type your message here")
+    sender_email = st.session_state.auth_state["user"].email
     receiver_email = st.text_input("Receiver's Email", help="Enter the receiver's email address")
+    message = st.text_input("Type your message", help="Type your message here")
     if st.button("Send"):
         if not message or not receiver_email:
             st.error("Please enter both a message and the receiver's email.")
@@ -317,9 +323,12 @@ def chat_ui(sender_email, receiver_email):
                 st.error("Failed to send message.")
 
     st.title("Messages")
-    messages = get_messages(sender_email, receiver_email)
-    if messages:
-        for key, value in messages.items():
+    sender_messages, receiver_messages = get_messages(sender_email, receiver_email)
+    if sender_messages:
+        for key, value in sender_messages.items():
+            st.write(f"{value['sender_email']}: {value['message']}")
+    if receiver_messages:
+        for key, value in receiver_messages.items():
             st.write(f"{value['sender_email']}: {value['message']}")
 
 # Update the main function to include the chat UI option
@@ -338,7 +347,7 @@ def main():
     elif option == "Recognize Face (Optimal)":
         recognize_face_optimal_ui()
     elif option == "Chat":
-        chat_ui(st.session_state.auth_state["user"].email, "RECEIVER_EMAIL")  # Change RECEIVER_EMAIL to the actual receiver's email
+        chat_ui()
 
 # Run the application
 if __name__ == "__main__":