Update app.py

addition of get_container_memory_limit()

app.py

@@ -20,13 +20,29 @@ def get_container_cpu_count():
                 period = int(f.read())
             if quota > 0 and period > 0:
                 cpu_count = quota / period
+        else:
+            print(f"{quota_path} or {period_path} doesn't exist")
         return int(cpu_count)
     except Exception:
         return os.cpu_count()
+
+def get_container_memory_limit():
+    try:
+        mem_limit_path = "/sys/fs/cgroup/memory/memory.limit_in_bytes"
+        if os.path.exists(mem_limit_path):
+            with open(mem_limit_path) as f:
+                mem_bytes = int(f.read())
+            return mem_bytes // (1024 * 1024)  # in MB
+        else:
+            print(f"{mem_limit_path} doesn't exist")
+        # fallback to total system memory
+        return int(psutil.virtual_memory().total // (1024 * 1024))
+    except Exception:
+        return None
         
 @app.get("/")
 def greet_json():
-    return {"Hello": "World!", "cpu_count": get_container_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": get_container_memory_limit()}
 
 @app.get("/health")
 def health():