Update main.py

main.py

@@ -1,6 +1,7 @@
 import os
 import shutil
 import logging
+import psutil
 from fastapi import FastAPI, UploadFile, File, Form
 from fastapi.responses import JSONResponse
 from deepface import DeepFace
@@ -11,6 +12,19 @@ logger = logging.getLogger(__name__)
 
 app = FastAPI()
 
+def log_ram_usage(context=""):
+    ram = psutil.virtual_memory()
+    process = psutil.Process(os.getpid())
+    process_ram = process.memory_info().rss / (1024 ** 3)
+
+    logger.info(
+        f"[RAM MONITOR] {context} | "
+        f"Total RAM: {ram.total / (1024 ** 3):.2f} GB | "
+        f"RAM Available: {ram.available / (1024 ** 3):.2f} GB | "
+        f"RAM Usage: {ram.percent}% | "
+        f"App RAM: {process_ram:.2f} GB"
+    )
+
 # Path database lokal (dipertahankan untuk kompatibilitas ke belakang)
 DB_PATH = "/app/face_db"
 if not os.path.exists(DB_PATH):
@@ -18,6 +32,8 @@ if not os.path.exists(DB_PATH):
 
 @app.get("/")
 async def root():
+    log_ram_usage("Health Check")
+    
     return {"status": "AI Face Recognition System is Running", "mode": "hybrid-stateless"}
 
 @app.post("/api/represent")
@@ -47,6 +63,8 @@ async def represent_face(
             enforce_detection=False
         )
 
+        log_ram_usage("After Face Representation")
+
         if len(representations) > 0:
             embedding = representations[0]["embedding"]
             return {"success": True, "embedding": embedding}