added middleware

main.py

@@ -1,11 +1,15 @@
 # main.py
-from fastapi import FastAPI,status,Response
+from fastapi import FastAPI,status,Response,Request
 from fastapi.responses import StreamingResponse
 from models import load_text_model,generate_text,load_audio_model,generate_audio,load_image_model, generate_image
 from schemas import VoicePresets
 from utils import audio_array_to_buffer,img_to_bytes
 from contextlib import asynccontextmanager
-from typing import AsyncIterator
+from typing import AsyncIterator,Callable,Awaitable
+from uuid import uuid4
+import time
+from datetime import datetime, timezone
+import csv
 
 models = {} 
 
@@ -17,6 +21,43 @@ async def lifespan(_: FastAPI) -> AsyncIterator[None]:
 
 app = FastAPI(lifespan=lifespan)
 
+csv_header = [
+    "Request ID", "Datetime", "Endpoint Triggered", "Client IP Address",
+    "Response Time", "Status Code", "Successful"
+]
+
+
+@app.middleware("http") 
+async def monitor_service(
+    req: Request, call_next: Callable[[Request], Awaitable[Response]]
+) -> Response: 
+    request_id = uuid4().hex 
+    request_datetime = datetime.now(timezone.utc).isoformat()
+    start_time = time.perf_counter()
+    response: Response = await call_next(req)
+    response_time = round(time.perf_counter() - start_time, 4) 
+    response.headers["X-Response-Time"] = str(response_time)
+    response.headers["X-API-Request-ID"] = request_id 
+    with open("usage.csv", "a", newline="") as file:
+        writer = csv.writer(file)
+        if file.tell() == 0:
+            writer.writerow(csv_header)
+        writer.writerow( 
+            [
+                request_id,
+                request_datetime,
+                req.url,
+                req.client.host,
+                response_time,
+                response.status_code,
+                response.status_code < 400,
+            ]
+        )
+    return response
+
+
+
+
 # app = FastAPI() 
 
 @app.get("/")