Update app.py

addition of get_container_cpu_count()

app.py

@@ -5,9 +5,28 @@ from fastapi import FastAPI
 
 app = FastAPI()
 
+def get_container_cpu_count():
+    try:
+        # default CPU count from Python
+        cpu_count = os.cpu_count()
+
+        # read cgroup quota (if available)
+        quota_path = "/sys/fs/cgroup/cpu/cpu.cfs_quota_us"
+        period_path = "/sys/fs/cgroup/cpu/cpu.cfs_period_us"
+        if os.path.exists(quota_path) and os.path.exists(period_path):
+            with open(quota_path) as f:
+                quota = int(f.read())
+            with open(period_path) as f:
+                period = int(f.read())
+            if quota > 0 and period > 0:
+                cpu_count = quota / period
+        return int(cpu_count)
+    except Exception:
+        return os.cpu_count()
+        
 @app.get("/")
 def greet_json():
-    return {"Hello": "World!", "cpu_count": os.cpu_count(), "total_ram_in_MB": psutil.virtual_memory().total // (1024*1024)}
+    return {"Hello": "World!", "cpu_count": get_container_cpu_count(), "total_ram_in_MB": psutil.virtual_memory().total // (1024*1024)}
 
 @app.get("/health")
 def health():