Update app.py

app.py

@@ -1,9 +1,9 @@
-import os
 import cv2
 import numpy as np
 import tensorflow as tf
 from tensorflow.keras.models import model_from_json
 import streamlit as st
+from streamlit_webrtc import VideoTransformerBase, webrtc_streamer
 from PIL import Image
 
 # Load model
@@ -22,13 +22,14 @@ def allowed_file(filename):
     """Checks the file format when file is uploaded"""
     return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
 
+
 def Emotion_Analysis(image):
     """It does prediction of Emotions found in the Image provided, saves as Images and returns them"""
     gray_frame = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
     faces = face_haar_cascade.detectMultiScale(gray_frame, scaleFactor=1.3, minNeighbors=5)
 
     if len(faces) == 0:
-        return None
+        return image, None
 
     for (x, y, w, h) in faces:
         roi = gray_frame[y:y + h, x:x + w]
@@ -51,14 +52,11 @@ def Emotion_Analysis(image):
 
     return image, pred_emotion
 
-def video_frame_callback(frame):
-    """Callback function to process each frame of video"""
-    image = np.array(frame)
-    result = Emotion_Analysis(image)
-    if result is not None:
-        processed_image, _ = result
-        return processed_image
-    return frame
+class EmotionDetector(VideoTransformerBase):
+    def transform(self, frame):
+        image = frame.to_ndarray(format="bgr24")
+        result_image, _ = Emotion_Analysis(image)
+        return result_image
 
 st.title('Emotion Detection App')
 
@@ -84,22 +82,6 @@ if upload_option == "Image Upload":
 
 elif upload_option == "Webcam":
     st.sidebar.write("Webcam Capture")
-    run = st.checkbox('Run Webcam')
-    FRAME_WINDOW = st.image([])
-
-    camera = cv2.VideoCapture(0)
-
-    while run:
-        success, frame = camera.read()
-        if not success:
-            st.error("Unable to read from webcam. Please check your camera settings.")
-            break
-        frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
-        processed_frame = video_frame_callback(frame)
-        FRAME_WINDOW.image(processed_frame)
-
-    camera.release()
+    webrtc_streamer(key="example", video_transformer_factory=EmotionDetector)
 else:
     st.write("Please select an option to start.")
-
-