UPDATE: exportToDashboard

analyticsHub/routers/reportingTool.py

@@ -7,6 +7,7 @@ from fastapi import APIRouter, Depends
 from urllib.request import urlopen
 from typing import Annotated
 from . import pipeline
+import psutil
 import json
 import os
 import gc
@@ -25,6 +26,10 @@ async def generateChart(chartDetails: GenerateChartInput, credentials: Annotated
                 metadata = json.loads(urlopen(fileUrl).read())
             )
             gc.collect()
+            memory = psutil.virtual_memory()
+            cpuUsage = psutil.cpu_percent(interval=1)
+            print(f"RAM Usage Percentage: {memory.percent}%")
+            print(f"Total CPU Usage: {cpuUsage}%")
             return JSONResponse(status_code = 200, content = response)
         else:
             return JSONResponse(status_code = 498, content = {"status": "ERROR", "errorDetail": "Invalid Token"})    
@@ -44,6 +49,10 @@ async def generatePanelChart(panelChartDetails: PanelChartDetails, credentials:
                 dataSource = panelChartDetails.dataSource
             )
             gc.collect()
+            memory = psutil.virtual_memory()
+            cpuUsage = psutil.cpu_percent(interval=1)
+            print(f"RAM Usage Percentage: {memory.percent}%")
+            print(f"Total CPU Usage: {cpuUsage}%")
             return JSONResponse(status_code = 200, content = response)
         else:
             return JSONResponse(status_code = 498, content = {"status": "ERROR", "errorDetail": "Invalid Token"})    

requirements.txt

@@ -12,6 +12,7 @@ python-dotenv
 python-jose
 pydantic
 pandas
+psutil
 uvicorn
 psycopg2-binary
 pymongo[srv]