Update main.py

main.py

@@ -2,7 +2,7 @@ from fastapi.responses import HTMLResponse
 from fastapi.templating import Jinja2Templates
 from fastapi import FastAPI, Request, HTTPException
 from fastapi.middleware.cors import CORSMiddleware
-from stdb import get_equity_data, get_stocks_data
+from stdb import DataManager
 
 app = FastAPI()
 
@@ -17,6 +17,7 @@ app.add_middleware(
 )
 
 templates = Jinja2Templates(directory="templates")
+data_manager = DataManager()
 
 @app.get("/", response_class=HTMLResponse)
 async def read_root(request: Request):
@@ -35,7 +36,7 @@ async def read_root(request: Request):
 @app.get('/data')
 async def get_data(ticker: str, date: str, qty: int):
     try:
-        response = get_equity_data(ticker, date, qty)
+        response = data_manager.get_equity_data(ticker, date, qty)
         return response
     except:
         return {"Timeout" : "Error"}
@@ -43,7 +44,7 @@ async def get_data(ticker: str, date: str, qty: int):
 @app.get('/stocks')
 async def get_stocks_data():
     try:
-        response = get_stocks_data()
+        response = data_manager.get_stocks_data()
         return response
     except:
         return {"Timeout" : "Error"}