Update app.py

app.py

@@ -7,6 +7,7 @@ from PIL import Image
 import numpy as np
 import cv2
 import dlib
+from streamlit import cache as st_cache
 
 # Get the current working directory
 current_directory = os.path.dirname(os.path.abspath(__file__))
@@ -62,6 +63,7 @@ def create_user(email, password):
         return False, None
         
 # Update load_and_encode function to use the aligned face without normalization
+@st_cache
 def load_and_encode(image_path):
     try:
         aligned_face = detect_and_align_faces(image_path)
@@ -107,6 +109,7 @@ def detect_and_align_faces(image_path):
     return aligned_face
 
 # Add person to database
+@st_cache
 def add_person(name, image_path, instagram_handle, email=None):
     try:
         encoding = load_and_encode(image_path)
@@ -135,6 +138,7 @@ def add_person(name, image_path, instagram_handle, email=None):
         return f"Failed to add person: {str(e)}"
 
 # Recognize face from image
+@st_cache
 def recognize_face(image_path):
     if not image_path:
         return "Please upload an image."
@@ -166,6 +170,7 @@ def recognize_face(image_path):
         return f"Failed to recognize face: {str(e)}"
         
 # Recognize face from image and return optimal or highest matching ID
+@st_cache
 def recognize_face_optimal(image_path):
     if not image_path:
         return "Please upload an image."
@@ -197,6 +202,7 @@ def recognize_face_optimal(image_path):
         return f"Failed to recognize face: {str(e)}"
 
 # Delete person from database
+@st_cache
 def delete_person(name):
     try:
         ref.child(name).delete()
@@ -303,6 +309,7 @@ def send_message(sender_email, receiver_email, message_content):
         return f"Failed to send message: {str(e)}"
 
 # Function to retrieve messages for a user
+@st_cache
 def get_messages(user_email):
     try:
         messages = db_firestore.collection('messages').where('receiver_email', '==', user_email).order_by('timestamp', direction=firestore.Query.DESCENDING).stream()