Abeg work

pages/1_Live_Detection.py

@@ -19,7 +19,7 @@ if "audio_queue" not in st.session_state:
 if "last_status" not in st.session_state:
     st.session_state.last_status = {"status": "Awake"}
 
-
+    
 # --- Load Configuration and Environment Variables ---
 @st.cache_resource
 def load_app_config():
@@ -51,11 +51,12 @@ def autoplay_audio(audio_bytes: bytes):
 
 # --- WebRTC Video Processor ---
 class VideoProcessor(VideoProcessorBase):
-    def __init__(self, status_queue: queue.Queue, audio_queue: queue.Queue):
-        self.status_queue = status_queue
-        self.audio_queue = audio_queue
+    def __init__(self):
         self._detector = get_detector(config)
         self._alerter = get_alerter(config, secrets["gemini_api_key"])
+        # Thread-safe queues for communication
+        self.status_queue = queue.Queue()
+        self.audio_queue = queue.Queue()
 
     def recv(self, frame: av.VideoFrame) -> av.VideoFrame:
         img = frame.to_ndarray(format="bgr24")
@@ -90,8 +91,7 @@ class VideoProcessor(VideoProcessorBase):
             self._alerter.reset_alert()
             
         return av.VideoFrame.from_ndarray(processed_frame, format="bgr24")
-    
-    
+
 # --- Page UI ---
 st.title("📹 Live Drowsiness Detection")
 st.info("Press 'START' to activate your camera and begin monitoring.")
@@ -136,6 +136,7 @@ if webrtc_ctx.state.playing:
         st.session_state.last_status = {"status": "Awake"}
     
     try:
+        # This line will now work because st.session_state.status_queue was initialized
         status_result = st.session_state.status_queue.get(timeout=0.1)
         st.session_state.last_status = status_result
     except queue.Empty: