Update app.py

app.py

@@ -68,11 +68,25 @@ def process_live_frame(frame):
     score = indicators.get("details", {}).get("Score", 0)
 
     # Build the status text
+    # Determine drowsiness level based on the UI slider's value
+    drowsiness_level = "Awake"
+    if score >= VERY_DROWSY_DEFAULT:  # Use a fixed upper threshold
+        drowsiness_level = "Very Drowsy"
+    elif score >= sensitivity_threshold: # Use the slider for slight drowsiness
+        drowsiness_level = "Slightly Drowsy"
+
+    # Build the status text with explicit details
     status_text = f"Lighting: {lighting}\n"
     if lighting == "Low":
         status_text += "Detection paused due to low light."
     else:
-        status_text += f"Status: {drowsiness_level}\nScore: {score:.2f}"
+        status_text += f"Status: {drowsiness_level}\nScore: {score:.2f} (Threshold: {sensitivity_threshold:.2f})"
+        # Explicitly show what is being detected
+        if score > 0:
+            if indicators.get('eye_closure'): status_text += "\n- Eyes Closed Detected"
+            if indicators.get('yawning'): status_text += "\n- Yawn Detected"
+            if indicators.get('head_nod'): status_text += "\n- Head Nod Detected"
+            if indicators.get('looking_away'): status_text += "\n- Looking Away Detected"
 
     # Handle alerts
     audio_output = None
@@ -83,6 +97,7 @@ def process_live_frame(frame):
     else:
         alerter.reset_alert()
 
+    # Return all the values needed to update the UI
     return processed_frame, status_text, audio_output
 
 # --- UI Definition for the Live Detection Page ---
@@ -98,12 +113,23 @@ def create_live_detection_page():
                 status_output = gr.Textbox(label="Live Status", lines=3, interactive=False)
                 # Audio player is now visible for debugging and user feedback.
                 audio_alert_output = gr.Audio(autoplay=True, visible=True, label="Alert Sound")
-
+    # --- Added Sensitivity Slider ---
+                sensitivity_slider = gr.Slider(
+                    minimum=0.1, 
+                    maximum=1.0, 
+                    value=SLIGHTLY_DROWSY_DEFAULT, 
+                    step=0.05, 
+                    label="Alert Sensitivity Threshold",
+                    info="Lower value = more sensitive to drowsiness signs."
+                )
+    
+        # Link the inputs (webcam and slider) to the processing function and its outputs
         webcam_input.stream(
             fn=process_live_frame,
-            inputs=[webcam_input],
-            outputs=[processed_output, status_output, audio_alert_output]
-        )
+            inputs=[webcam_input, sensitivity_slider],
+            outputs=[processed_output, status_output, audio_alert_output],
+            every=0.1
+    )
     return live_detection_page
 
 # --- UI Definition for the Home Page ---