envConfig.py

import os
import tempfile

from dotenv import load_dotenv

#******************************************************************************************************************************************************************

class envConfiguration:

    ready = False;
    
    def __init__ (self):
        load_dotenv()

    #sets the values into local env
    def set_config(self, openai_key, openai_model, openai_temp, pinecone_key, pinecone_env, pinecone_index):
        os.environ["OPEN_AI_KEY"] = openai_key
        os.environ["OPEN_AI_MODEL"] = openai_model
        os.environ["OPEN_AI_TEMP"] = str(openai_temp)
        os.environ["PINECONE_KEY"] = pinecone_key
        os.environ["PINECONE_ENV"] = pinecone_env
        os.environ["PINECONE_IND"] = pinecone_index
        os.environ["INPUT_DIR"] = os.path.join(tempfile.gettempdir(),
                                                'rkk-document-gpt',
                                                'input'
                                                )
        os.environ["TEMPLATE_DIR"] = os.path.join(tempfile.gettempdir(),
                                                'rkk-document-gpt',
                                                'template'
                                                )
        os.environ["OUTPUT_DIR"] = os.path.join(tempfile.gettempdir(),
                                                'rkk-document-gpt',
                                                'output'
                                                )

        os.makedirs(os.environ.get("INPUT_DIR"), exist_ok = True)
        os.makedirs(os.environ.get("TEMPLATE_DIR"), exist_ok = True)
        os.makedirs(os.environ.get("OUTPUT_DIR"), exist_ok = True)

        self.ready = True

        return None
    
    def openai_key(self):
        return os.environ.get("OPEN_AI_KEY")

    def openai_model(self):
        return os.environ.get("OPEN_AI_MODEL")

    def openai_temp(self):
        return os.environ.get("OPEN_AI_TEMP")

    def pinecone_key(self):
        return os.environ.get("PINECONE_KEY")

    def pinecone_env(self):
        return os.environ.get("PINECONE_ENV")

    def pinecone_index(self):
        return os.environ.get("PINECONE_IND")

    def input_dir(self):
        return os.environ.get("INPUT_DIR")    

    def template_dir(self):
        return os.environ.get("TEMPLATE_DIR")

    def output_dir(self):
        return os.environ.get("OUTPUT_DIR")

    def is_ready(self):
        return self.ready