Update main.py

main.py

@@ -1,8 +1,9 @@
-from fastapi import FastAPI
+from fastapi import FastAPI, HTTPException
 from fastapi.responses import JSONResponse
 import time
 import psutil
 import os
+import asyncio
 
 app = FastAPI()
 
@@ -10,8 +11,11 @@ process = psutil.Process(os.getpid())
 last_cpu_times = process.cpu_times()
 last_time = time.time()
 
-def get_cpu_percent():
+# Limit concurrent requests
+MAX_CONCURRENT_REQUESTS = 20
+semaphore = asyncio.Semaphore(MAX_CONCURRENT_REQUESTS)
 
+def get_cpu_percent():
     global last_cpu_times, last_time
     now = time.time()
     times = process.cpu_times()
@@ -19,19 +23,27 @@ def get_cpu_percent():
     cpu_used = (times.user - last_cpu_times.user) + (times.system - last_cpu_times.system)
     last_cpu_times = times
     last_time = now
-
     return round((cpu_used / elapsed) * 100, 1)
 
 def simulate_work():
-    
     total = 0
     for i in range(500000):
         total += i
+    return total
+
+async def limited_worker():
+    async with semaphore:
+        # Actual work happens here
+        simulate_work()
+        return {"ok": True, "time": int(time.time() * 1000)}
 
 @app.get("/")
 async def root():
-    simulate_work()
-    return {"ok": True, "time": int(time.time() * 1000)}
+    try:
+        result = await limited_worker()
+        return result
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
 
 @app.get("/metrics")
 async def metrics():