Update app.py

app.py

@@ -5,7 +5,7 @@ import warnings
 from fastapi import FastAPI, File, UploadFile, HTTPException
 from contextlib import asynccontextmanager
 import torch
-from ultralytics import YOLO, settings
+from ultralytics import YOLO
 import cv2
 import requests
 import json
@@ -17,21 +17,15 @@ import logging
 import pandas as pd
 
 # --- Initial Configuration ---
-# Completely suppress all warnings
+# Suppress all warnings
 warnings.filterwarnings("ignore")
-os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'
 
-# Set up YOLO configuration directory
+# Clean up and set YOLO_CONFIG_DIR
 yolo_config_dir = "/tmp/Ultralytics"
-
-# Clean and recreate directory
 if os.path.exists(yolo_config_dir):
     shutil.rmtree(yolo_config_dir)
 os.makedirs(yolo_config_dir, exist_ok=True)
-
-# Set environment variable and update Ultralytics settings
 os.environ["YOLO_CONFIG_DIR"] = yolo_config_dir
-settings.update({'config_dir': yolo_config_dir})
 
 # --- Logging Configuration ---
 # Configure logging before any other operations
@@ -44,13 +38,11 @@ logging.basicConfig(
 # Get logger instance
 logger = logging.getLogger(__name__)
 
-# Suppress all third-party logging
-logging.getLogger("ultralytics").setLevel(logging.CRITICAL)
-logging.getLogger("PIL").setLevel(logging.CRITICAL)
-logging.getLogger("matplotlib").setLevel(logging.CRITICAL)
-logging.getLogger("urllib3").setLevel(logging.CRITICAL)
-logging.getLogger("requests").setLevel(logging.CRITICAL)
-logging.getLogger("torch").setLevel(logging.CRITICAL)
+# Suppress third-party logging
+logging.getLogger("ultralytics").setLevel(logging.ERROR)
+logging.getLogger("PIL").setLevel(logging.WARNING)
+logging.getLogger("matplotlib").setLevel(logging.WARNING)
+logging.getLogger("urllib3").setLevel(logging.WARNING)
 
 # Log environment variable once
 logger.info(f"YOLO_CONFIG_DIR set to: {os.getenv('YOLO_CONFIG_DIR')}")
@@ -71,19 +63,15 @@ class YOLOv8Model:
         try:
             logger.info("Initializing YOLOv8 model")
             
-            # Completely suppress stdout/stderr during model loading
+            # Temporarily suppress stdout for model loading
             original_stdout = sys.stdout
-            original_stderr = sys.stderr
             sys.stdout = open(os.devnull, 'w')
-            sys.stderr = open(os.devnull, 'w')
             
             try:
                 self.model = YOLO(model_path)
             finally:
                 sys.stdout.close()
-                sys.stderr.close()
                 sys.stdout = original_stdout
-                sys.stderr = original_stderr
             
             logger.info("YOLOv8 model loaded successfully")
         except Exception as e:
@@ -92,15 +80,8 @@ class YOLOv8Model:
 
     def predict(self, image):
         try:
-            # Suppress output during prediction
-            original_stdout = sys.stdout
-            sys.stdout = open(os.devnull, 'w')
-            try:
-                results = self.model(image)
-                return results.pandas().xyxy[0]
-            finally:
-                sys.stdout.close()
-                sys.stdout = original_stdout
+            results = self.model(image)
+            return results.pandas().xyxy[0]
         except Exception as e:
             logger.error(f"Prediction error: {e}")
             raise
@@ -194,25 +175,13 @@ async def lifespan(app: FastAPI):
     global yolo_model
     logger.info("FastAPI application starting up")
     try:
-        # Double suppression during model initialization
-        original_stdout = sys.stdout
-        original_stderr = sys.stderr
-        sys.stdout = open(os.devnull, 'w')
-        sys.stderr = open(os.devnull, 'w')
-        
-        try:
-            yolo_model = YOLOv8Model()
-        finally:
-            sys.stdout.close()
-            sys.stderr.close()
-            sys.stdout = original_stdout
-            sys.stderr = original_stderr
-        
+        yolo_model = YOLOv8Model()
         logger.info("YOLOv8 model initialized successfully")
     except Exception as e:
         logger.error(f"Startup error: {e}")
         raise
     yield
+    logger    	    
     logger.info("FastAPI application shutting down")
     yolo_model = None