|
10 | 10 | from app.errors.handlers import register_handlers |
11 | 11 | from app.services import tts_service, clothing_detector |
12 | 12 |
|
13 | | -# ── Logging config ───────────────────────────────────────────────────────────── |
14 | | -# Show INFO+ from our code; suppress noisy third-party logs |
| 13 | +# ── Silence noisy libraries (Python logging layer) ───────────────────────────── |
| 14 | +for _lib in ("tensorflow", "absl", "h5py", "keras", "kokoro", "torch", |
| 15 | + "urllib3", "httpx", "httpcore"): |
| 16 | + logging.getLogger(_lib).setLevel(logging.ERROR) |
| 17 | + |
| 18 | +# ── Suppress TF stderr noise via env vars (must be set before TF imports) ────── |
| 19 | +os.environ.setdefault("TF_CPP_MIN_LOG_LEVEL", "3") # suppress TF C++ logs |
| 20 | +os.environ.setdefault("TF_ENABLE_ONEDNN_OPTS", "0") # suppress oneDNN warning |
| 21 | +os.environ.setdefault("PYTHONWARNINGS", "ignore") # suppress Python warnings |
| 22 | + |
| 23 | +# ── App logger ───────────────────────────────────────────────────────────────── |
15 | 24 | logging.basicConfig( |
16 | 25 | level=logging.INFO, |
17 | | - format="%(asctime)s [%(levelname)s] %(name)s: %(message)s", |
| 26 | + format="%(asctime)s [%(levelname)s] %(message)s", |
18 | 27 | datefmt="%H:%M:%S", |
19 | 28 | ) |
20 | | -# Suppress library noise |
21 | | -for _noisy in ("tensorflow", "absl", "h5py", "keras", "kokoro", "torch", |
22 | | - "urllib3", "httpx", "httpcore", "uvicorn.access"): |
23 | | - logging.getLogger(_noisy).setLevel(logging.ERROR) |
24 | | - |
25 | 29 | logger = logging.getLogger("rizzvision") |
26 | 30 |
|
27 | 31 |
|
| 32 | +# ── Filter health-check pings from access log ─────────────────────────────────── |
| 33 | +class _HealthFilter(logging.Filter): |
| 34 | + def filter(self, record: logging.LogRecord) -> bool: |
| 35 | + msg = record.getMessage() |
| 36 | + return not any(p in msg for p in ("/health", "HEAD /", "GET / ")) |
| 37 | + |
| 38 | +logging.getLogger("uvicorn.access").addFilter(_HealthFilter()) |
| 39 | + |
| 40 | + |
28 | 41 | @asynccontextmanager |
29 | 42 | async def lifespan(app: FastAPI): |
30 | | - logger.info("━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━") |
31 | | - logger.info("🚀 Rizzvision backend starting up") |
32 | | - logger.info("━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━") |
| 43 | + logger.info("----------------------------------------") |
| 44 | + logger.info("Rizzvision backend starting") |
| 45 | + logger.info("----------------------------------------") |
33 | 46 |
|
34 | | - # 1. TTS warmup |
35 | | - logger.info("[1/2] Loading Kokoro TTS pipelines (en + hi)…") |
| 47 | + logger.info("[1/2] Loading Kokoro TTS (en + hi)...") |
36 | 48 | try: |
37 | 49 | await asyncio.to_thread(tts_service.warmup) |
38 | | - logger.info("[1/2] ✓ TTS ready") |
| 50 | + logger.info("[1/2] TTS loaded OK") |
39 | 51 | except Exception as exc: |
40 | | - logger.error("[1/2] ✗ TTS warmup failed: %s", exc) |
| 52 | + logger.error("[1/2] TTS load FAILED: %s", exc) |
41 | 53 |
|
42 | | - # 2. Clothing classifier |
43 | | - logger.info("[2/2] Loading clothing classifier v3 (EfficientNetB3, 115 MB)…") |
| 54 | + logger.info("[2/2] Loading clothing classifier v3 (115 MB)...") |
44 | 55 | try: |
45 | 56 | await asyncio.to_thread(clothing_detector._load) |
46 | | - logger.info("[2/2] ✓ Clothing classifier ready — thresholds: %s", clothing_detector._thresholds) |
| 57 | + t = clothing_detector._thresholds |
| 58 | + logger.info( |
| 59 | + "[2/2] Classifier loaded OK | tops=%.3f bottoms=%.3f " |
| 60 | + "footwear=%.3f outerwear=%.3f dress=%.3f", |
| 61 | + t.get("tops", 0), t.get("bottoms", 0), |
| 62 | + t.get("footwear", 0), t.get("outerwear", 0), t.get("dress", 0), |
| 63 | + ) |
47 | 64 | except Exception as exc: |
48 | | - logger.error("[2/2] ✗ Clothing classifier failed to load: %s", exc) |
| 65 | + logger.error("[2/2] Classifier load FAILED: %s", exc) |
49 | 66 |
|
50 | | - logger.info("━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━") |
51 | | - logger.info("✅ All models loaded — accepting requests") |
52 | | - logger.info("━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━") |
| 67 | + logger.info("----------------------------------------") |
| 68 | + logger.info("All models ready — accepting requests") |
| 69 | + logger.info("----------------------------------------") |
53 | 70 | yield |
54 | | - logger.info("🛑 Rizzvision backend shutting down") |
| 71 | + logger.info("Backend shutting down") |
55 | 72 |
|
56 | 73 |
|
57 | 74 | app = FastAPI(title="rizzvision-v2", version="2.0.0", lifespan=lifespan) |
|
0 commit comments