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 init_global_var(workdir="workdir"):
    """Initialize all global variables"""
    global _prompt_store, _sql_engine, _docs, _datasets
    _prompt_store = get_prompt_tinydb(workdir)
    _sql_engine = get_sqlite_engine(workdir)
    _docs = scan_docs_directory(workdir)
    _datasets = get_all_dataset(workdir)

def get_prompt_store():
    return _prompt_store

def set_prompt_store(new_prompt_store):
    global _prompt_store
    _prompt_store = new_prompt_store

def get_sql_engine():
    return _sql_engine

def set_sql_engine(new_sql_engine):
    global _sql_engine
    _sql_engine = new_sql_engine

def get_docs():
    return _docs

def set_docs(new_docs):
    global _docs
    _docs = new_docs

def get_datasets():
    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