Update app.py

app.py

@@ -42,7 +42,6 @@ emotion_labels = ['angry', 'fear', 'happy', 'neutral', 'sad', 'surprise']
 # Initialize MTCNN for face detection
 mtcnn = MTCNN()
 
-# Load known faces and names
 # Load known faces and names
 known_faces = []
 known_names = []
@@ -122,6 +121,9 @@ def process_frame(frame):
             cv2.rectangle(frame, (x, y), (x + w, y + h), (0, 255, 0), 2)
             cv2.putText(frame, result_text, (x, y - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.9, (0, 255, 0), 2)
 
+    else:
+        result_text = "No face detected!"
+
     return frame, result_text
 
 
@@ -177,16 +179,14 @@ st.markdown("### MongoDB Records")
 records = collection.find().sort("timestamp", -1)  # Sort records by timestamp in descending order
 
 for record in records:
-    col1, col2, col3 = st.columns([3, 3, 1])
+    col1, col2 = st.columns([3, 3])
     with col1:
         st.write(f"**Name**: {record['name']}")
     with col2:
         st.write(f"**Emotion**: {record['emotion']}")
-    with col3:
-        st.write(f"**Timestamp**: {record['timestamp']}")
     
-    # Delete record button
-    delete_button = st.button(f"Delete {record['_id']}", key=record['_id'])
+    # Delete record button without showing ID
+    delete_button = st.button(f"Delete Record", key=record['_id'])
     if delete_button:
         collection.delete_one({"_id": record["_id"]})
-        st.success(f"Record with ID {record['_id']} has been deleted.") 
+        st.success(f"Record with ID {record['_id']} has been deleted.")