Update app.py

app.py

@@ -113,9 +113,13 @@ def process_frame(frame):
                     "emotion": emotion,
                     "timestamp": timestamp
                 }
-                # Insert the data into MongoDB
-                collection.insert_one(document)
-                print(f"Data inserted into MongoDB: {document}")
+                
+                # Check if the same record already exists (to prevent duplicates)
+                existing_record = collection.find_one({"name": name, "timestamp": timestamp})
+                if not existing_record:
+                    # Insert the data into MongoDB
+                    collection.insert_one(document)
+                    print(f"Data inserted into MongoDB: {document}")
 
             # Draw bounding box and label
             cv2.rectangle(frame, (x, y), (x + w, y + h), (0, 255, 0), 2)
@@ -126,7 +130,6 @@ def process_frame(frame):
 
     return frame, result_text
 
-
 # Video feed display
 def video_feed(video_source):
     frame_placeholder = st.empty()  # Placeholder for displaying video frames
@@ -174,16 +177,18 @@ elif upload_choice == "Upload Video":
             video_source = cv2.VideoCapture(tfile.name)
             video_feed(video_source)
 
-# Display the records stored in MongoDB with latest records on top
+# Display the latest 5 records stored in MongoDB with timestamp and proper alignment
 st.markdown("### MongoDB Records")
-records = collection.find().sort("timestamp", -1)  # Sort records by timestamp in descending order
+records = collection.find().sort("timestamp", -1).limit(5)  # Limit to 5 records
 
 for record in records:
-    col1, col2 = st.columns([3, 3])
+    col1, col2, col3 = st.columns([3, 3, 2])
     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
     delete_button = st.button("Delete", key=f"delete_{record['_id']}")