Update app.py

app.py

@@ -1,7 +1,7 @@
 import os
 import streamlit as st
 import firebase_admin
-from firebase_admin import credentials, auth, firestore, db as realtimedb
+from firebase_admin import credentials, db, auth
 import face_recognition
 from PIL import Image
 import numpy as np
@@ -12,23 +12,17 @@ import dlib
 current_directory = os.path.dirname(os.path.abspath(__file__))
 skp_path = os.path.join(current_directory, "projectinsta-s-firebase-adminsdk-y6vlu-9a1345f468.json")
 
-# Initialize Firebase Admin SDK for Firestore
+# Check if the app is already initialized
 if not firebase_admin._apps:
+    # Initialize Firebase Admin SDK
     cred = credentials.Certificate(skp_path)
     firebase_admin.initialize_app(cred, {
         'databaseURL': 'https://projectinsta-s-default-rtdb.firebaseio.com/',
         'projectId': 'projectinsta-s'
     })
-    db_firestore = firestore.client()
 
-# Initialize Firebase Admin SDK for Realtime Database
-if not firebase_admin._apps:
-    cred = credentials.Certificate(skp_path)
-    firebase_admin.initialize_app(cred, {
-        'databaseURL': 'https://projectinsta-s-default-rtdb.firebaseio.com/',
-        'projectId': 'projectinsta-s'
-    })
-    db_realtime = realtimedb.reference()
+# Reference to the root of your Firebase Realtime Database
+ref = db.reference('/')
 
 # Streamlit session state
 if "auth_state" not in st.session_state:
@@ -63,52 +57,6 @@ 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(db_firestore, db_realtime, sender, recipient, message):
-    try:
-        # Add a new document with a generated ID to Firestore
-        db_firestore.collection("messages").add({
-            "sender": sender,
-            "recipient": recipient,
-            "message": message,
-            "timestamp": firebase_admin.firestore.SERVER_TIMESTAMP
-        })
-        
-        # Add a new child with a generated ID to Realtime Database
-        db_realtime.child("messages").push({
-            "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 from 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")
-        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 from Firestore: {str(e)}")
-
-# Function to retrieve messages from Realtime Database
-def retrieve_messages_realtime(db_realtime):
-    try:
-        # Retrieve all documents from the 'messages' collection in Realtime Database
-        messages_ref = db_realtime.child("messages").get()
-        for message_id, message_data in messages_ref.items():
-            # 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 from Realtime Database: {str(e)}")
         
 # Update load_and_encode function to use the aligned face without normalization
 def load_and_encode(image_path):
@@ -165,8 +113,8 @@ def add_person(name, image_path, instagram_handle):
         # Convert NumPy arrays to lists for JSON serialization
         encoding = encoding[0].tolist()
 
-        # Save data to Firestore
-        db_firestore.collection("persons").document(name).set({
+        # Save data to Firebase Realtime Database
+        ref.child(name).set({
             "encoding": encoding,
             "info": {
                 "instagram_handle": instagram_handle,
@@ -174,9 +122,9 @@ def add_person(name, image_path, instagram_handle):
             }
         })
 
-        return f"Success: {name} added to Firestore!"
+        return f"Success: {name} added to the database!"
     except Exception as e:
-        return f"Failed to add person to Firestore: {str(e)}"
+        return f"Failed to add person: {str(e)}"
 
 # Recognize face from image
 def recognize_face(image_path):
@@ -189,11 +137,10 @@ def recognize_face(image_path):
             return "No face found in the provided image."
 
         matches = []
-        for person_ref in db_firestore.collection("persons").stream():
-            data = person_ref.to_dict()
+        for name, data in ref.get().items():
             known_encoding = np.array(data["encoding"])
             if face_recognition.compare_faces([known_encoding], unknown_encoding[0])[0]:
-                matches.append((person_ref.id, data["info"]))
+                matches.append((name, data["info"]))
 
         if matches:
             results = []
@@ -219,16 +166,15 @@ def recognize_face_optimal(image_path):
             return "No face found in the provided image."
 
         matches = []
-        for person_ref in db_firestore.collection("persons").stream():
-            data = person_ref.to_dict()
+        for name, data in ref.get().items():
             known_encoding = np.array(data["encoding"])
             similarity_score = face_recognition.face_distance([known_encoding], unknown_encoding[0])[0]
-            matches.append((person_ref.id, similarity_score))
+            matches.append((name, similarity_score))
 
         if matches:
             best_match = min(matches, key=lambda x: x[1])
             best_name, best_score = best_match
-            info = db_firestore.collection("persons").document(best_name).get().to_dict()["info"]
+            info = ref.child(best_name).child("info").get()
             insta_handle = info["instagram_handle"]
             insta_link = info["instagram_link"]
             insta_link_html = f'<a href="{insta_link}" target="_blank"><font color="red">{insta_handle}</font></a>'
@@ -238,13 +184,13 @@ def recognize_face_optimal(image_path):
     except Exception as e:
         return f"Failed to recognize face: {str(e)}"
 
-# Delete person from Firestore
+# Delete person from database
 def delete_person(name):
     try:
-        db_firestore.collection("persons").document(name).delete()
-        return f"{name} deleted from Firestore!"
+        ref.child(name).delete()
+        return f"{name} deleted from the database!"
     except Exception as e:
-        return f"Failed to delete person from Firestore: {str(e)}"
+        return f"Failed to delete person: {str(e)}"
 
 # Streamlit interface for adding a person
 def add_person_ui():
@@ -266,20 +212,6 @@ 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(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")
-    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 (Firestore)")
-    retrieve_messages_firestore(db_firestore)
-    st.header("Messages (Realtime Database)")
-    retrieve_messages_realtime(db_realtime)
 
 # Streamlit interface for recognizing face with optimal ID
 def recognize_face_optimal_ui():
@@ -301,7 +233,7 @@ def delete_person_ui():
             st.success(result)
 
 # Streamlit interface for user authentication
-def authenticate_user_ui(db_firestore, db_realtime):  # Pass the database references as arguments
+def authenticate_user_ui():
     st.title("Insta's EYE")
     st.sidebar.title("Options")
 
@@ -322,9 +254,7 @@ def authenticate_user_ui(db_firestore, db_realtime):  # Pass the database refere
                 else:
                     success, user = authenticate_user(email, password)
                     if success:
-                        # Extract user's email from UserRecord object
-                        user_email = user.email
-                        st.session_state.auth_state["user"] = {"email": user_email}
+                        st.session_state.auth_state["user"] = user
                         st.session_state.auth_state["signed_in"] = True
                         st.success("Authentication successful! You can now manage your set of images and profiles.")
                         main()
@@ -353,7 +283,7 @@ def logout():
 # 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"])
+    option = st.sidebar.radio("Select Option", ["Add Person", "Recognize Face", "Delete Person", "Recognize Face (Optimal)"])
 
     if option == "Add Person":
         add_person_ui()
@@ -363,8 +293,6 @@ def main():
         delete_person_ui()
     elif option == "Recognize Face (Optimal)":
         recognize_face_optimal_ui()
-    elif option == "Chat":
-        chat_ui()
 
 if __name__ == "__main__":
-     authenticate_user_ui(db_firestore, db_realtime)
+    authenticate_user_ui()