import logging
from llama_index.core.indices import VectorStoreIndex
from app.engine.vectordb import get_vector_store
from app.engine.loaders import get_document_Types

logger = logging.getLogger("uvicorn")

indexs = {}

def get_index(params=None):
    global indexs
    if len(index) <= 0:
        logger.info("Connecting vector store...")
        docTypes = get_document_Types()
        for docType in docTypes:
            store = get_vector_store(docType)
            # Load the index from the vector store
            # If you are using a vector store that doesn't store text,
            # you must load the index from both the vector store and the document store
            index = VectorStoreIndex.from_vector_store(store)
            logger.info("Finished load index from vector store.")
            indexs[docType] = index
    return indexs