Pushing code for rtc loading

pages/1_Live_Detection.py

@@ -17,8 +17,6 @@ from src.alerting.alert_system import get_alerter
 def load_app_config():
     """Loads config from yaml and .env files."""
     load_dotenv()
-    gemini_api_key = os.getenv("GEMINI_API_KEY")
-    # Navigate up to the root to find the config file
     config_path = "/config.yaml" if os.path.exists("/config.yaml") else "config.yaml"
     with open(config_path, 'r') as f:
         config = yaml.safe_load(f)
@@ -32,11 +30,11 @@ def load_app_config():
 
 config, secrets = load_app_config()
 
-# --- Initialize Session State (if not already done in main.py) ---
-if "play_audio" not in st.session_state:
-    st.session_state.play_audio = None
-if "active_alerts" not in st.session_state:
-    st.session_state.active_alerts = {"status": "Awake"}
+# --- Initialize Session State for Queues at the TOP of the script ---
+if "status_queue" not in st.session_state:
+    st.session_state.status_queue = queue.Queue()
+if "audio_queue" not in st.session_state:
+    st.session_state.audio_queue = queue.Queue()
 
 # --- Client-Side Audio Playback Function ---
 def autoplay_audio(audio_bytes: bytes):
@@ -51,7 +49,10 @@ def autoplay_audio(audio_bytes: bytes):
 
 # --- WebRTC Video Processor ---
 class VideoProcessor(VideoProcessorBase):
-    def __init__(self):
+    # The __init__ method now accepts the queues as arguments
+    def __init__(self, status_queue: queue.Queue, audio_queue: queue.Queue):
+        self.status_queue = status_queue
+        self.audio_queue = audio_queue
         self._detector = get_detector(config)
         self._alerter = get_alerter(config, secrets["gemini_api_key"])
 
@@ -61,24 +62,24 @@ class VideoProcessor(VideoProcessorBase):
         strategy = config.get('detection_strategy')
         if strategy == 'hybrid':
             processed_frame, alert_triggered, active_alerts = self._detector.process_frame(img)
-            st.session_state.active_alerts = active_alerts if alert_triggered else {"status": "Awake"}
-        else: # Fallback for simpler strategies
+            # Use self.status_queue.put() instead of st.session_state
+            self.status_queue.put(active_alerts if alert_triggered or 'Low Light' in active_alerts else {"status": "Awake"})
+        else:
             processed_frame, indicators = self._detector.process_frame(img)
-            alert_triggered = any(indicators.values())
-            st.session_state.active_alerts = indicators if alert_triggered else {"status": "Awake"}
+            alert_triggered = any(v for k, v in indicators.items() if k != 'low_light')
+            self.status_queue.put(indicators)
 
         if alert_triggered:
             audio_data = self._alerter.trigger_alert()
             if audio_data:
-                st.session_state.play_audio = audio_data
+                # Use self.audio_queue.put()
+                self.audio_queue.put(audio_data)
         else:
             self._alerter.reset_alert()
             
         return av.VideoFrame.from_ndarray(processed_frame, format="bgr24")
 
 # --- Page UI ---
-# The st.set_page_config() call has been removed from this file.
-# The configuration from main.py will apply to this page.
 st.title("📹 Live Drowsiness Detection")
 st.info("Press 'START' to activate your camera and begin monitoring.")
 
@@ -87,16 +88,8 @@ ice_servers = [{'urls': 'stun:global.stun.twilio.com:3478'}]
 if secrets["turn_username"] and secrets["turn_credential"]:
     print("TURN credentials found, adding TURN servers to config.")
     turn_servers = [
-        {
-            'urls': 'turn:global.turn.twilio.com:3478?transport=udp',
-            'username': secrets["turn_username"],
-            'credential': secrets["turn_credential"]
-        },
-        {
-            'urls': 'turn:global.turn.twilio.com:3478?transport=tcp',
-            'username': secrets["turn_username"],
-            'credential': secrets["turn_credential"]
-        }
+        {'urls': 'turn:global.turn.twilio.com:3478?transport=udp', 'username': secrets["turn_username"], 'credential': secrets["turn_credential"]},
+        {'urls': 'turn:global.turn.twilio.com:3478?transport=tcp', 'username': secrets["turn_username"], 'credential': secrets["turn_credential"]}
     ]
     ice_servers.extend(turn_servers)
 
@@ -108,8 +101,12 @@ col1, col2 = st.columns([3, 1])
 with col1:
     webrtc_ctx = webrtc_streamer(
         key="drowsiness-detection",
-        video_processor_factory=VideoProcessor,
-        rtc_configuration=RTC_CONFIGURATION, # Use the new robust configuration
+        # The factory now correctly passes the queues from session_state
+        video_processor_factory=lambda: VideoProcessor(
+            status_queue=st.session_state.status_queue,
+            audio_queue=st.session_state.audio_queue
+        ),
+        rtc_configuration=RTC_CONFIGURATION,
         media_stream_constraints={"video": True, "audio": False},
         async_processing=True,
     )
@@ -126,25 +123,18 @@ with col2:
     audio_placeholder = st.empty()
 
 if webrtc_ctx.state.playing:
-    # --- Polling Loop ---
+    if "last_status" not in st.session_state:
+        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:
-        status_result = None
+        pass
 
-    # Check for new audio alerts
-    try:
-        audio_data = st.session_state.audio_queue.get(timeout=0.1)
-    except queue.Empty:
-        audio_data = None
-    
     with status_placeholder.container():
-        # Persist the last known status if there's no new one
-        if status_result:
-            st.session_state.last_status = status_result
-        
-        last_status = getattr(st.session_state, 'last_status', {"status": "Awake"})
-        
+        last_status = st.session_state.last_status
         if last_status.get("Low Light"):
              st.warning("⚠️ Low Light Detected! Accuracy may be affected.")
         elif last_status.get("status") == "Awake":
@@ -152,17 +142,18 @@ if webrtc_ctx.state.playing:
         else:
             st.error("🚨 DROWSINESS DETECTED!")
             for key, value in last_status.items():
-                if key != "Low Light":
+                if key != "Low Light" and key != "status":
                     st.warning(f"-> {key}: {value:.2f}" if isinstance(value, float) else f"-> {key}")
     
-    if audio_data:
+    try:
+        audio_data = st.session_state.audio_queue.get(timeout=0.1)
         with audio_placeholder.container():
             autoplay_audio(audio_data)
-    
-    # Force a rerun to keep the polling active
+    except queue.Empty:
+        pass
+
     time.sleep(0.1)
     st.rerun()
-
 else:
     with status_placeholder.container():
         st.info("✔️ Driver is Awake")