Update app.py

app.py

@@ -20,15 +20,19 @@ import pandas as pd
 # Suppress all warnings
 warnings.filterwarnings("ignore")
 
-# Clean up and set YOLO_CONFIG_DIR
+# Set up YOLO_CONFIG_DIR
 yolo_config_dir = "/tmp/Ultralytics"
-if os.path.exists(yolo_config_dir):
-    shutil.rmtree(yolo_config_dir)
-os.makedirs(yolo_config_dir, exist_ok=True)
-os.environ["YOLO_CONFIG_DIR"] = yolo_config_dir
+try:
+    if os.path.exists(yolo_config_dir):
+        shutil.rmtree(yolo_config_dir)
+    os.makedirs(yolo_config_dir, exist_ok=True)
+    os.chmod(yolo_config_dir, 0o777)  # Ensure directory is writable
+    os.environ["YOLO_CONFIG_DIR"] = yolo_config_dir
+except Exception as e:
+    print(f"Failed to set up YOLO_CONFIG_DIR: {e}")
+    raise
 
 # --- Logging Configuration ---
-# Configure logging before any other operations
 logging.basicConfig(
     level=logging.INFO,
     format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
@@ -62,18 +66,14 @@ class YOLOv8Model:
     def __init__(self, model_path='yolov8n.pt'):
         try:
             logger.info("Initializing YOLOv8 model")
-            
-            # Temporarily suppress stdout for model loading
             original_stdout = sys.stdout
             sys.stdout = open(os.devnull, 'w')
-            
             try:
                 self.model = YOLO(model_path)
+                logger.info("YOLOv8 model loaded successfully")
             finally:
                 sys.stdout.close()
                 sys.stdout = original_stdout
-            
-            logger.info("YOLOv8 model loaded successfully")
         except Exception as e:
             logger.error(f"Failed to load YOLOv8 model: {e}")
             raise
@@ -98,6 +98,7 @@ def preprocess_frame(frame):
 
 def capture_rtsp_frames(rtsp_url: str):
     try:
+        logger.info(f"Attempting to connect to RTSP stream: {rtsp_url}")
         cap = cv2.VideoCapture(rtsp_url)
         if not cap.isOpened():
             logger.error(f"Failed to open RTSP stream: {rtsp_url}")
@@ -122,6 +123,7 @@ def save_snapshot(frame):
         filename = f"snapshot_{int(time.time())}.jpg"
         snapshot_path = Path("/snapshots") / filename
         os.makedirs("/snapshots", exist_ok=True)
+        os.chmod("/snapshots", 0o777)  # Ensure directory is writable
         cv2.imwrite(str(snapshot_path), frame)
         return f"/snapshots/{filename}"
     except Exception as e:
@@ -181,7 +183,6 @@ async def lifespan(app: FastAPI):
         logger.error(f"Startup error: {e}")
         raise
     yield
-    logger    	    
     logger.info("FastAPI application shutting down")
     yolo_model = None