Update app.py

app.py

@@ -14,7 +14,7 @@ import matplotlib.pyplot as plt
 
 import io
 import streamlit as st
-from streamlit.server.server import Server
+import streamlit_webrtc as webrtc
 
 
 
@@ -82,19 +82,22 @@ def cam():
     return render_template('attendance.html')
 
 def main():
-    server = Server.get_current()
-    img_file_buffer = server.uploaded_file_mgr.get_file('img_file_buffer')
-    bytes_data = None
-    if img_file_buffer is not None:
-        test_image = Image.open(img_file_buffer)
-        st.image(test_image, use_column_width=True)
-        st.write(type(test_image))
-        image = np.asarray(test_image)
-        gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
-        st.image(cv2.cvtColor(image, cv2.COLOR_BGR2RGB))
-        if bytes_data is None:
-            st.stop()
+    webrtc_ctx = webrtc.client(ctx=st)
+    webrtc_ctx.create_stream()
     
+    while True:
+        if webrtc_ctx.video_receiver:
+            frame = webrtc_ctx.video_receiver.get_frame()
+            if frame is not None:
+                image = frame.to_ndarray(format="bgr24")
+                # Process the image with OpenCV as needed
+                
+                # Display the processed image
+                st.image(image, channels="BGR")
+        else:
+            st.write("Waiting for video...")
+        st.sleep(0.1)
+
     return render_template('attendance.html')