Update app.py

app.py

@@ -1,7 +1,7 @@
 import os
 import streamlit as st
 import firebase_admin
-from firebase_admin import credentials, db, auth
+from firebase_admin import credentials, db, auth, firestore
 import face_recognition
 from PIL import Image
 import numpy as np
@@ -20,6 +20,7 @@ if not firebase_admin._apps:
         'databaseURL': 'https://projectinsta-s-default-rtdb.firebaseio.com/',
         'projectId': 'projectinsta-s'
     })
+    db = firestore.client()
 
 # Reference to the root of your Firebase Realtime Database
 ref = db.reference('/')
@@ -57,6 +58,32 @@ def create_user(email, password):
     except Exception as e:
         print(f"User creation error: {str(e)}")
         return False, None
+
+# Function to send a message
+def send_message(sender, recipient, message):
+    try:
+        # Add a new document with a generated ID
+        db.collection("messages").add({
+            "sender": sender,
+            "recipient": recipient,
+            "message": message,
+            "timestamp": firebase_admin.firestore.SERVER_TIMESTAMP
+        })
+        return "Message sent successfully!"
+    except Exception as e:
+        return f"Failed to send message: {str(e)}"
+
+# Function to retrieve messages
+def retrieve_messages():
+    try:
+        # Retrieve all documents from the 'messages' collection
+        messages_ref = db.collection("messages").order_by("timestamp")
+        for message in messages_ref.stream():
+            message_data = message.to_dict()
+            # Display the message in the chat interface
+            st.write(f"{message_data['sender']}: {message_data['message']} ({message_data['timestamp']})")
+    except Exception as e:
+        st.error(f"Failed to retrieve messages: {str(e)}")
         
 # Update load_and_encode function to use the aligned face without normalization
 def load_and_encode(image_path):
@@ -213,6 +240,18 @@ def recognize_face_ui():
         result = recognize_face(image_path)
         st.write(result, unsafe_allow_html=True)
 
+# Streamlit interface for chat
+def chat_ui():
+    st.title("Chat")
+    sender = st.session_state.auth_state["user"]["email"]
+    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(sender, recipient, message)
+        st.write(result)
+    st.header("Messages")
+    retrieve_messages()
+
 # Streamlit interface for recognizing face with optimal ID
 def recognize_face_optimal_ui():
     st.title("Recognize Face (Optimal)")
@@ -280,31 +319,21 @@ def logout():
     st.session_state.auth_state["user"] = None
     st.session_state.auth_state["signed_in"] = False
 
-# Update the main function to include the new option
 # Update the main function to include the new option
 def main():
     st.sidebar.title("Options")
+    option = st.sidebar.radio("Select Option", ["Add Person", "Recognize Face", "Delete Person", "Recognize Face (Optimal)", "Chat"])
 
-    add_icon_path = os.path.join(current_directory, "Add.png")
-    delete_icon_path = os.path.join(current_directory, "delete.png")
-    recognize_icon_path = os.path.join(current_directory, "Explore.png")
-    recognize_optimal_icon_path = os.path.join(current_directory, "Explore+.png")
-
-    add_icon = Image.open(add_icon_path)
-    delete_icon = Image.open(delete_icon_path)
-    recognize_icon = Image.open(recognize_icon_path)
-    recognize_optimal_icon = Image.open(recognize_optimal_icon_path)
-
-    option = st.sidebar.selectbox("Select Option", [add_icon, delete_icon, recognize_icon, recognize_optimal_icon], index=0, format_func=lambda x: "")
-
-    if option == add_icon:
+    if option == "Add Person":
         add_person_ui()
-    elif option == delete_icon:
-        delete_person_ui()
-    elif option == recognize_icon:
+    elif option == "Recognize Face":
         recognize_face_ui()
-    elif option == recognize_optimal_icon:
+    elif option == "Delete Person":
+        delete_person_ui()
+    elif option == "Recognize Face (Optimal)":
         recognize_face_optimal_ui()
+    elif option == "Chat":
+        chat_ui()
 
 if __name__ == "__main__":
     authenticate_user_ui()