Update main.py

main.py

@@ -819,7 +819,7 @@ class SafeDrivingAssistant:
         self.alert_manager.play_energetic_music()
         self.add_chat_log("System", "Energetic synthwave music started. Stay alert!")
 
-    # Core Drowsiness Evaluation & Loop
+   # Core Drowsiness Evaluation & Loop
 
     def run(self):
         """Main camera acquisition loop that drives the safe assistant."""
@@ -830,46 +830,77 @@ class SafeDrivingAssistant:
         cap.set(cv2.CAP_PROP_FRAME_WIDTH, FRAME_WIDTH)
         cap.set(cv2.CAP_PROP_FRAME_HEIGHT, FRAME_HEIGHT)
         
+        # Detect if we are in a headless cloud environment without a webcam
+        use_simulation = False
         if not cap.isOpened():
-            print("[CoreEngine] FATAL: Could not access web camera.")
-            return
+            print("[CoreEngine] WARNING: Could not access physical web camera.")
+            print("[CoreEngine] Pivoting to Cloud Simulation Mode to keep web HUD alive...")
+            use_simulation = True
+        else:
+            print("[CoreEngine] Camera stream operational. System fully active.")
 
-        print("[CoreEngine] Camera stream operational. System fully active.")
-        print("[CoreEngine] Press 'q' in CV window or click Dashboard Reset to quit.")
+        print("[CoreEngine] System loop running. Use the Web Dashboard to monitor telemetry.")
         
         prev_time = time.time()
         
         try:
             while True:
-                ret, frame = cap.read()
-                if not ret:
-                    time.sleep(0.01)
-                    continue
-
-                # Mirror frame for intuitive pilot HUD overlay
-                frame = cv2.flip(frame, 1)
+                current_time = time.time()
+                
+                if use_simulation:
+                    # Generate an animated cyberpunk grid frame for the headless dashboard
+                    frame = np.zeros((FRAME_HEIGHT, FRAME_WIDTH, 3), dtype=np.uint8)
+                    # Create a scrolling scan line
+                    scan_y = int(current_time * 120) % FRAME_HEIGHT
+                    cv2.line(frame, (0, scan_y), (FRAME_WIDTH, scan_y), (40, 40, 40), 2)
+                    cv2.putText(frame, "CLOUD SIMULATION FEED (NO PHYSICAL CAM)", (20, FRAME_HEIGHT - 20), 
+                                cv2.FONT_HERSHEY_SIMPLEX, 0.4, (0, 255, 0), 1)
+                    ret = True
+                else:
+                    ret, frame = cap.read()
+                    if not ret:
+                        time.sleep(0.01)
+                        continue
+                    # Mirror frame for intuitive pilot HUD overlay
+                    frame = cv2.flip(frame, 1)
                 
                 # Check if tracking is active (controlled from Dashboard)
                 with dashboard_state.lock:
                     active = dashboard_state.detection_active
                     
                 if active:
-                    # Calculate EAR and overlay glow contours on frame
-                    ear, landmarks, processed_frame = self.detector.process_frame(frame)
+                    if use_simulation:
+                        # Cloud Demo Mode: Automatically simulate a drowsy event cycle every 25 seconds
+                        # to let you test your Flask dashboard overlays and system responses safely!
+                        cycle = int(current_time) % 25
+                        if cycle > 18:  # Simulate closed eyes for 7 seconds
+                            ear = 0.16
+                            cv2.putText(frame, "SIMULATING DROWSINESS (EYES CLOSED)", (FRAME_WIDTH // 2 - 180, FRAME_HEIGHT // 2),
+                                        cv2.FONT_HERSHEY_SIMPLEX, 0.6, (0, 0, 255), 2)
+                        else:
+                            ear = 0.28
+                        
+                        processed_frame = frame.copy()
+                        # Draw virtual telemetry eye dots onto the matrix background
+                        cv2.circle(processed_frame, (int(FRAME_WIDTH * 0.4), int(FRAME_HEIGHT * 0.45)), 8, (0, 255, 0) if ear > EAR_THRESHOLD else (0, 0, 255), -1)
+                        cv2.circle(processed_frame, (int(FRAME_WIDTH * 0.6), int(FRAME_HEIGHT * 0.45)), 8, (0, 255, 0) if ear > EAR_THRESHOLD else (0, 0, 255), -1)
+                        if ear is not None:
+                            cv2.putText(processed_frame, f"EAR: {ear:.2f}", (30, 40), cv2.FONT_HERSHEY_SIMPLEX, 0.7, (0, 255, 0) if ear > EAR_THRESHOLD else (0, 0, 255), 2)
+                        landmarks = {}
+                    else:
+                        # Calculate EAR and overlay glow contours on frame via physical camera
+                        ear, landmarks, processed_frame = self.detector.process_frame(frame)
                 else:
                     processed_frame = frame.copy()
                     cv2.putText(processed_frame, "TRACKING PAUSED", (50, 50), 
                                 cv2.FONT_HERSHEY_SIMPLEX, 0.8, (0, 165, 255), 2)
                     ear = None
 
-                current_time = time.time()
-                
                 # Calculate processing frame rate (FPS)
-                fps = int(1.0 / (current_time - prev_time))
+                fps = int(1.0 / (current_time - prev_time)) if (current_time - prev_time) > 0 else 30
                 prev_time = current_time
 
                 # Drowsiness Logic Decision Engine
-
                 if ear is not None and active:
                     if ear < EAR_THRESHOLD:
                         self.consec_closed_frames += 1
@@ -925,7 +956,7 @@ class SafeDrivingAssistant:
                                 self.alert_manager.trigger_level3_advisory()
                                 self.add_chat_log("System", "FREQUENT DROWSINESS DETECTED. Prompting driver to pull over.")
                             else:
-                                # Normal recovery (driver had a brief blink/close and is now normal)
+                                # Normal recovery
                                 current_state = self.get_system_state()
                                 if current_state not in ["WAITING_REST_RESPONSE", "WAITING_SONG_RESPONSE"]:
                                     self.set_system_state("NORMAL")
@@ -935,7 +966,6 @@ class SafeDrivingAssistant:
                     self.eyes_closed_start_time = None
 
                 # Update Global Telemetry Buffer for Flask Server
-
                 with dashboard_state.lock:
                     dashboard_state.latest_frame = processed_frame.copy()
                     if ear is not None:
@@ -944,25 +974,31 @@ class SafeDrivingAssistant:
                         dashboard_state.ear = 0.30  # Default baseline when no face present
                     dashboard_state.fps = fps
 
-                # OpenCV display output fallback (allows running headless or with double screens)
-                cv2.imshow("DriveSafe HUD AI Console", processed_frame)
-                
-                # Check for keyboard inputs on CV Window ('q' to quit, 'r' to reset)
-                key = cv2.waitKey(1) & 0xFF
-                if key == ord('q') or key == 27:  # ESC or Q
-                    print("[CoreEngine] Exit key received. Terminating system.")
-                    break
-                elif key == ord('r'):
-                    self.reset_warnings()
+                # OpenCV display output fallback (wrapped safely to prevent headless display context drops)
+                try:
+                    cv2.imshow("DriveSafe HUD AI Console", processed_frame)
+                    key = cv2.waitKey(1) & 0xFF
+                    if key == ord('q') or key == 27:
+                        print("[CoreEngine] Exit key received. Terminating system.")
+                        break
+                    elif key == ord('r'):
+                        self.reset_warnings()
+                except Exception:
+                    # Prevents crashes on platforms where standard desktop window pipelines are fully restricted
+                    time.sleep(0.03)
 
         except KeyboardInterrupt:
             print("[CoreEngine] Keyboard interrupt. Shutting down.")
         finally:
             print("[CoreEngine] Releasing resources...")
             cap.release()
-            cv2.destroyAllWindows()
+            try:
+                cv2.destroyAllWindows()
+            except Exception:
+                pass
             self.assistant.stop()
             sys.exit(0)
+                 
 
 if __name__ == "__main__":
     assistant_app = SafeDrivingAssistant()