Update app.py

app.py

@@ -15,6 +15,9 @@ EMOTION_MODEL_FILE = "CNN_Model_acc_75.h5"
 HUGGING_FACE_TOKEN = os.getenv("HUGGING_FACE_TOKEN")
 HUGGING_FACE_REPO = "username/repo_name"
 
+# Emotion Labels
+EMOTION_LABELS = ["Angry", "Disgust", "Fear", "Happy", "Sad", "Surprise", "Neutral"]
+
 # Create directories
 os.makedirs(KNOWN_FACES_DIR, exist_ok=True)
 
@@ -80,7 +83,7 @@ def detect_faces_and_emotions(image):
         reshaped_face = np.reshape(normalized_face, (1, 48, 48, 1))
         emotion_prediction = emotion_model.predict(reshaped_face)
         emotion_label = np.argmax(emotion_prediction)
-        return emotion_label
+        return EMOTION_LABELS[emotion_label]
 
     return None
 
@@ -134,17 +137,16 @@ elif choice == "Face Recognition and Emotion Detection":
                 st.error(f"Error: {e}")
 
     elif action == "Use Webcam":
-        cap = cv2.VideoCapture(0)
-        if st.button("Start Webcam"):
-            while True:
-                ret, frame = cap.read()
-                if not ret:
-                    break
-                emotion_label = detect_faces_and_emotions(frame)
+        st.info("Use the camera input widget to capture an image.")
+        camera_image = st.camera_input("Take a picture")
+        if camera_image:
+            try:
+                img = Image.open(camera_image)
+                img_array = np.array(img)
+                emotion_label = detect_faces_and_emotions(img_array)
                 if emotion_label is not None:
                     st.success(f"Emotion Detected: {emotion_label}")
-                cv2.imshow("Webcam", frame)
-                if cv2.waitKey(1) & 0xFF == ord('q'):
-                    break
-            cap.release()
-            cv2.destroyAllWindows()
+                else:
+                    st.warning("No face detected.")
+            except Exception as e:
+                st.error(f"Error: {e}")