Update main.py

main.py

@@ -5,7 +5,6 @@ import time
 import requests
 import asyncio
 import threading
-import platform
 from fastapi import FastAPI, Request, HTTPException, Depends
 from fastapi.responses import JSONResponse
 from fastapi.security import HTTPBearer, HTTPAuthorizationCredentials
@@ -13,10 +12,12 @@ import uvicorn
 from pydantic import BaseModel
 from typing import List, Optional
 
+# Import shared metrics helper (cùng folder, không cần path manipulation)
 try:
-    import psutil
+    from shared_metrics import get_hf_metrics
 except ImportError:
-    psutil = None
+    def get_hf_metrics():
+        return {"total_memory_mb": 16384, "used_memory_mb": 4096, "cpu_count": 2, "cpu_percent": 0, "platform": "Linux", "max_workers": 6, "hardware_tier": "cpu-basic", "gpu": None}
 
 try:
     from vietnamese_ai.serving.batch_server import MayChuBatch  # type: ignore
@@ -60,25 +61,6 @@ batch_server = MayChuBatch(
     so_worker=4
 )
 
-def get_metrics():
-    """Thu thập chỉ số phần cứng thực tế của server"""
-    metrics = {
-        "total_memory_mb": 0,
-        "used_memory_mb": 0,
-        "cpu_percent": 0,
-        "cpu_count": os.cpu_count() or 1,
-        "platform": platform.system(),
-        "max_workers": 2,
-    }
-    if psutil:
-        mem = psutil.virtual_memory()
-        metrics["total_memory_mb"] = round(mem.total / 1024 / 1024)
-        metrics["used_memory_mb"] = round(mem.used / 1024 / 1024)
-        metrics["cpu_percent"] = psutil.cpu_percent(interval=0.1)
-        # Ước tính workers: mỗi worker cần ~2GB RAM
-        metrics["max_workers"] = max(1, int(mem.available / 1024 / 1024 / 2048))
-    return metrics
-
 def start_heartbeat(port):
     while True:
         try:
@@ -86,7 +68,7 @@ def start_heartbeat(port):
                 "role": "Gateway",
                 "port": port,
                 "public_url": os.environ.get("SPACE_HOST", ""),
-                "metrics": get_metrics()
+                "metrics": get_hf_metrics()
             }, timeout=5)
         except Exception:
             pass