Files
mars_toolkit/main.py
2025-01-05 17:43:06 +08:00

28 lines
670 B
Python

"""
Author: Yutang LI
Institution: SIAT-MIC
Contact: yt.li2@siat.ac.cn
"""
from fastapi import FastAPI
import logging
from database.material_project_router import router as material_project_router
from database.oqmd_router import router as oqmd_router
from model.fairchem_router import router as fairchem_router, init_model
logging.basicConfig(
level=logging.INFO,
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
)
logger = logging.getLogger(__name__)
app = FastAPI()
@app.on_event("startup")
def startup_event():
init_model()
app.include_router(material_project_router)
app.include_router(oqmd_router)
app.include_router(fairchem_router)