Update main.py

main.py

@@ -1,11 +1,28 @@
 from fastapi import FastAPI
+import os
 
 
 app = FastAPI(
     title="Test Selenium"
 )
 
+
 @app.get("/")
-def root ():
-    result = {message: "This is test"}
+def root(
+    cpu_load: Optional[str] = Query(
+        False,
+        description='True/False depending your needs, gets average CPU load value',
+        regex='^(True|False)$')
+) -> Any:
+    result: Dict[Any, Any] = {
+        "message": "Your first endpoint is working"
+    }
+
+    if cpu_load == 'True':
+        result["cpu_average_load"] = os.getloadavg()
+        result["cpu"] = os.cpu_count()
+        
+        mem_bytes = os.sysconf('SC_PAGE_SIZE') * os.sysconf('SC_PHYS_PAGES')
+        result["memory"] = mem_bytes/(1024)
+        
     return result