Update pages/Camera.py

pages/Camera.py

@@ -7,10 +7,9 @@ from streamlit_webrtc import WebRtcMode, webrtc_streamer
 
 from utils import SLInference
 
-
 logger = logging.getLogger(__name__)
 
-def main():
+def display():
     """
     Main function of the app.
     """
@@ -27,9 +26,9 @@ def main():
     inference_thread.start()
 
     webrtc_ctx = webrtc_streamer(
-        key="video-sendonly",
-        mode=WebRtcMode.SENDONLY,
-        media_stream_constraints={"video": True},
+        key="video-sendrecv",
+        mode=WebRtcMode.SENDRECV,  # Используем SENDRECV для отправки и получения видео
+        media_stream_constraints={"video": True, "audio": False},  # Включаем только видео
     )
 
     gestures_deque = deque(maxlen=5)
@@ -40,9 +39,8 @@ def main():
     text_output = st.empty()
     last_5_gestures = st.empty()
 
-
-    while True:
-        if webrtc_ctx.video_receiver:
+    if webrtc_ctx.video_receiver:
+        while True:
             try:
                 video_frame = webrtc_ctx.video_receiver.get_frame(timeout=1)
             except queue.Empty:
@@ -60,11 +58,6 @@ def main():
                 elif gesture != gestures_deque[-1]:
                     gestures_deque.append(gesture)
 
-            text_output.markdown(f'<p style="font-size:20px"> Current gesture: {gesture}</p>',
-                                 unsafe_allow_html=True)
-            last_5_gestures.markdown(f'<p style="font-size:20px"> Last 5 gestures: {" ".join(gestures_deque)}</p>',
-                                 unsafe_allow_html=True)
-            print(gestures_deque)
-
-if __name__ == "__main__":
-    main()
+            text_output.markdown(f'<p style="font-size:20px"> Current gesture: {gesture}</p>', unsafe_allow_html=True)
+            last_5_gestures.markdown(f'<p style="font-size:20px"> Last 5 gestures: {" ".join(gestures_deque)}</p>', unsafe_allow_html=True)
+            print(gestures_deque)