Update app.py

app.py

@@ -61,6 +61,8 @@ def process_frame(frame):
     gray_frame = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
     faces = face_cascade.detectMultiScale(gray_frame, scaleFactor=1.1, minNeighbors=5, minSize=(30, 30))
 
+    result_text = ""  # Initialize the result text for display
+
     for (x, y, w, h) in faces:
         roi_gray = gray_frame[y:y+h, x:x+w]
         roi_color = frame[y:y+h, x:x+w]
@@ -79,26 +81,33 @@ def process_frame(frame):
         if confidence < 100:
             name = known_names[label]
 
+        # Format the result text as "Name is feeling Emotion"
+        result_text = f"{name} is feeling {emotion}"
+
         # Draw bounding box and label on the frame
         cv2.rectangle(frame, (x, y), (x + w, y + h), (0, 255, 0), 2)
-        cv2.putText(frame, f"{name} - {emotion}", (x, y - 10), cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 255, 0), 2)
+        cv2.putText(frame, result_text, (x, y - 10), cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 255, 0), 2)
 
-    return frame
+    return frame, result_text
 
 # Video feed
 def video_feed(video_source):
     frame_placeholder = st.empty()  # This placeholder will be used to replace frames in-place
+    text_placeholder = st.empty()  # This placeholder will display the result text
 
     while True:
         ret, frame = video_source.read()
         if not ret:
             break
 
-        frame = process_frame(frame)
+        frame, result_text = process_frame(frame)
 
         # Display the frame in the placeholder
         frame_placeholder.image(frame, channels="BGR", use_column_width=True)
 
+        # Display the result text in the text placeholder
+        text_placeholder.markdown(f"<h3 style='text-align: center;'>{result_text}</h3>", unsafe_allow_html=True)
+
 # Sidebar for video or image upload
 upload_choice = st.sidebar.radio("Choose input source", ["Upload Video", "Upload Image", "Camera"])
 
@@ -121,7 +130,8 @@ elif upload_choice == "Upload Image":
     if uploaded_image:
         image = Image.open(uploaded_image)
         frame = np.array(image)
-        frame = process_frame(frame)
+        frame, result_text = process_frame(frame)
         st.image(frame, caption='Processed Image', use_column_width=True)
+        st.markdown(f"<h3 style='text-align: center;'>{result_text}</h3>", unsafe_allow_html=True)
 
 st.sidebar.write("Emotion Labels: Angry, Fear, Happy, Neutral, Sad, Surprise")