logging

Dockerfile

@@ -1,20 +1,32 @@
 # Use official Python 3.11 as base image
 FROM python:3.11-slim
 
-# Install system dependencies
+# Install system dependencies including timezone data
 RUN apt-get update && apt-get install -y \
     build-essential \
     curl \
     git \
+    tzdata \
     && rm -rf /var/lib/apt/lists/*
 
+# Set timezone to Asia/Kolkata (IST)
+ENV TZ=Asia/Kolkata
+RUN ln -snf /usr/share/zoneinfo/Asia/Kolkata /etc/localtime && echo Asia/Kolkata > /etc/timezone
+
 # Set up non-root user (HF Standard)
 RUN useradd -m -u 1000 user
 ENV HOME=/home/user \
-	PATH=/home/user/.local/bin:$PATH
+	PATH=/home/user/.local/bin:$PATH \
+	MPLCONFIGDIR=/tmp/.matplotlib \
+	PYTHONUNBUFFERED=1 \
+	PYTHONDONTWRITEBYTECODE=1 \
+	TZ=Asia/Kolkata
 
 WORKDIR $HOME/app
 
+# Create matplotlib config directory
+RUN mkdir -p /tmp/.matplotlib && chown -R user:user /tmp/.matplotlib
+
 # Copy and install basic deps
 COPY --chown=user requirements.txt requirements.txt
 RUN pip install --no-cache-dir --upgrade pip && \
@@ -31,9 +43,7 @@ RUN mkdir -p /tmp/tmp && chown -R user:user /tmp/tmp
 RUN ln -s /tmp/tmp tmp
 
 # Hugging Face PORT is 7860
-ENV PORT=7860 \
-    PYTHONUNBUFFERED=1 \
-    PYTHONDONTWRITEBYTECODE=1
+ENV PORT=7860
 
 USER user
 EXPOSE 7860

main.py

@@ -2,21 +2,72 @@ import pandas as pd
 import numpy as np
 import joblib
 import shap
-from fastapi import FastAPI, HTTPException
+from fastapi import FastAPI, HTTPException, Request
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.staticfiles import StaticFiles
 from fastapi.responses import FileResponse
 from pydantic import BaseModel, Field
 from typing import Optional, List
 import os
+import logging
+from datetime import datetime
+from pytz import timezone
+from starlette.middleware.base import BaseHTTPMiddleware
+
+# Setup IST timezone logging
+IST = timezone('Asia/Kolkata')
+
+# Configure logging with IST timezone
+class ISTFormatter(logging.Formatter):
+    """Custom formatter for IST timestamps"""
+    def formatTime(self, record, datefmt=None):
+        dt = datetime.fromtimestamp(record.created, tz=IST)
+        if datefmt:
+            return dt.strftime(datefmt)
+        else:
+            return dt.strftime("%Y-%m-%d %H:%M:%S %Z")
+
+# Only configure logs from main, not uvicorn multiple times
+if not logging.getLogger("uvicorn").handlers:
+    logging.basicConfig(
+        level=logging.INFO,
+        format='[%(asctime)s] %(levelname)s - %(name)s - %(message)s',
+        datefmt='%Y-%m-%d %H:%M:%S %Z'
+    )
+    # Apply IST formatter
+    for handler in logging.root.handlers:
+        handler.setFormatter(ISTFormatter('[%(asctime)s] %(levelname)s - %(message)s'))
+
+logger = logging.getLogger(__name__)
+
+def get_client_ip(request: Request) -> str:
+    """Extract client IP from request headers or socket"""
+    if 'x-forwarded-for' in request.headers:
+        return request.headers['x-forwarded-for'].split(',')[0].strip()
+    elif 'x-real-ip' in request.headers:
+        return request.headers['x-real-ip']
+    else:
+        return request.client.host if request.client else "0.0.0.0"
+
+class LoggingMiddleware(BaseHTTPMiddleware):
+    async def dispatch(self, request: Request, call_next):
+        client_ip = get_client_ip(request)
+        method = request.method
+        path = request.url.path
+        logger.info(f"📨 Incoming | IP: {client_ip} | {method} {path}")
+        
+        response = await call_next(request)
+        
+        logger.info(f"✅ Response | IP: {client_ip} | {method} {path} | Status: {response.status_code}")
+        return response
 
 try:
     from llm import start_chat_session, get_chat_response
     CHAT_AVAILABLE = True
-    print("[OK] LLM Chat module loaded")
+    logger.info("[OK] LLM Chat module loaded")
 except Exception as _llm_err:
     CHAT_AVAILABLE = False
-    print(f"[!] LLM Chat unavailable: {_llm_err}")
+    logger.warning(f"[!] LLM Chat unavailable: {_llm_err}")
 
 app = FastAPI(
     title="PlacementPredictor+",
@@ -24,6 +75,9 @@ app = FastAPI(
     version="1.0.0"
 )
 
+# Add logging middleware first
+app.add_middleware(LoggingMiddleware)
+
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
@@ -55,14 +109,14 @@ def load_artifacts():
     feature_names = artifacts['feature_names']
     routing_engine = artifacts.get('routing_engine')
     explainer = shap.TreeExplainer(model)
-    print("[OK] All artifacts loaded successfully!")
+    logger.info("[OK] All artifacts loaded successfully!")
 
 @app.on_event("startup")
 async def startup_event():
     try:
         load_artifacts()
     except Exception as e:
-        print(f"[!] Warning: {e}")
+        logger.error(f"[!] Warning: {e}")
 
 class StudentData(BaseModel):
     Age: int = Field(..., ge=15, le=40)

requirements.txt

@@ -22,4 +22,5 @@ python-multipart
 sh
 jinja2
 python-slugify
-sqlalchemy
+sqlalchemy
+pytz