Update app.py

app.py

@@ -191,11 +191,16 @@ def recognize_face_optimal(image_path):
         if matches:
             best_match = min(matches, key=lambda x: x[1])
             best_name, best_score = best_match
-            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>'
-            return f"Best match: {best_name} with a similarity score of {1 - best_score:.2%}. Insta handle: {insta_link_html}"
+
+            # Check if the best match is within a reasonable range of similarity scores
+            if best_score < 0.6:  # Adjust the threshold as needed
+                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>'
+                return f"Best match: {best_name} with a similarity score of {1 - best_score:.2%}. Insta handle: {insta_link_html}"
+            else:
+                return "No suitable match found."
         else:
             return "Face not found in the database."
     except Exception as e:
@@ -317,6 +322,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()
@@ -326,7 +332,7 @@ def get_messages(user_email):
 
 # Streamlit interface for messaging
 def messaging_ui():
-    st.title("Messaging")
+    st.title("💬 Messaging")
 
     if st.session_state.auth_state["signed_in"]:
         sender_email = st.session_state.auth_state["user"].email