Update app.py

app.py

@@ -2,6 +2,7 @@
 import os
 os.environ['TF_ENABLE_ONEDNN_OPTS'] = '0'
 os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
+os.environ['CUDA_VISIBLE_DEVICES'] = '-1'  # Force CPU usage to avoid GPU issues
 
 # Standard Library Imports
 import base64
@@ -26,9 +27,17 @@ try:
     from PIL import Image
     import requests
     from dotenv import load_dotenv
-    from deepface import DeepFace
     from retrying import retry
     from simple_salesforce import Salesforce
+    
+    # Import DeepFace with fallback
+    try:
+        from deepface import DeepFace
+        DEEPFACE_AVAILABLE = True
+    except ImportError as e:
+        print(f"DeepFace not available: {e}")
+        DEEPFACE_AVAILABLE = False
+        
 except ImportError as e:
     print(f"Missing dependency: {e}")
     print("Please install required packages using: pip install -r requirements.txt")
@@ -334,6 +343,9 @@ class AttendanceSystem:
         if image is None or not name.strip():
             return "❌ Please provide both image and name!", self.get_registered_workers_info()
         
+        if not DEEPFACE_AVAILABLE:
+            return "❌ DeepFace is not available. Please check installation.", self.get_registered_workers_info()
+        
         try:
             # Validate image quality
             image_array = np.array(image)
@@ -375,6 +387,9 @@ class AttendanceSystem:
 
     def _register_worker_auto(self, face_image: np.ndarray) -> Optional[Tuple[str, str]]:
         """Auto-register a new worker with improved validation."""
+        if not DEEPFACE_AVAILABLE:
+            return None
+            
         try:
             # Skip very small faces
             if face_image.shape[0] < 50 or face_image.shape[1] < 50:
@@ -503,6 +518,12 @@ class AttendanceSystem:
 
     def process_frame(self, frame: np.ndarray) -> np.ndarray:
         """Enhanced frame processing with better error handling."""
+        if not DEEPFACE_AVAILABLE:
+            # Draw a message on the frame
+            cv2.putText(frame, "DeepFace not available", (50, 50), 
+                       cv2.FONT_HERSHEY_SIMPLEX, 1, (0, 0, 255), 2)
+            return frame
+            
         try:
             self.processing_stats['frames_processed'] += 1
             
@@ -784,10 +805,12 @@ class AttendanceSystem:
     def get_system_status(self) -> str:
         """Get comprehensive system status."""
         sf_status = "✅ Connected" if self.sf else "❌ Offline"
+        deepface_status = "✅ Available" if DEEPFACE_AVAILABLE else "❌ Not Available"
         db_workers = len(self.db.get_all_workers())
         
         status = f"""🔧 **System Status:**
 - Salesforce: {sf_status}
+- DeepFace: {deepface_status}
 - Registered Workers: {db_workers}
 - Recognition Threshold: {FACE_RECOGNITION_THRESHOLD}
 - Confidence Threshold: {CONFIDENCE_THRESHOLD}