from db import get_sqlite_engine, get_prompt_tinydb, get_all_dataset
from tools import scan_docs_directory

_prompt_store = None
_sql_engine = None
_docs = None
_datasets = None

def get_prompt_store():
    global _prompt_store
    if _prompt_store is None:
        _prompt_store = get_prompt_tinydb("workdir")
    return _prompt_store

def set_prompt_store(new_prompt_store):
    global _prompt_store
    _prompt_store = new_prompt_store

def get_sql_engine():
    global _sql_engine
    if _sql_engine is None:
        _sql_engine = get_sqlite_engine("workdir")
    return _sql_engine

def set_sql_engine(new_sql_engine):
    global _sql_engine
    _sql_engine = new_sql_engine

def get_docs():
    global _docs
    if _docs is None:
        _docs = scan_docs_directory("workdir")
    return _docs

def set_docs(new_docs):
    global _docs
    _docs = new_docs

def get_datasets():
    global _datasets
    if _datasets is None:
        _datasets = get_all_dataset("workdir")
    return _datasets

def set_datasets(new_datasets):
    global _datasets
    _datasets = new_datasets

_model = None
_tokenizer = None

def get_model():
    return _model

def set_model(new_model):
    global _model
    _model = new_model

def get_tokenizer():
    return _tokenizer

def set_tokenizer(new_tokenizer):
    global _tokenizer
    _tokenizer = new_tokenizer