import os

from consts.consts import GPT_4, LANGUAGE_CN, MODE_TYPE_ASK_BILLING, MODE_TYPE_LENGJING, MODE_TYPE_SPIDER, MODE_TYPE_TIINSIGHT
from models.base import with_session
from models.user_preference import UserPreference, Preference


env = os.getenv("ENV", "dev")


def add_user_preference_dev():
    with with_session() as s:
        # tiinsight_dev = UserPreference.load(0)
        # tiinsight_dev.ai_model = GPT_4
        # tiinsight_dev.mode_type = MODE_TYPE_TIINSIGHT
        # Preference.upsert(s, 0, tiinsight_dev.to_json())

        # spider_dev = UserPreference.load(1)
        # spider_dev.ai_model = GPT_4
        # spider_dev.mode_type = MODE_TYPE_SPIDER
        # Preference.upsert(s, 1, spider_dev.to_json())

        # ask_billing_dev = UserPreference.load(2)
        # ask_billing_dev.ai_model = GPT_4
        # ask_billing_dev.mode_type = MODE_TYPE_ASK_BILLING
        # Preference.upsert(s, 2, ask_billing_dev.to_json())

        # O0T0DMS0
        lengjing_dev = UserPreference.load(1470738)
        lengjing_dev.llm_model = GPT_4
        lengjing_dev.mode_type = MODE_TYPE_LENGJING
        lengjing_dev.language_code = LANGUAGE_CN
        Preference.upsert(s, 1470738, lengjing_dev.to_json())


def add_user_preference_prod():
    with with_session() as s:
        # 371137 lengjing 40R9P3M0
        lengjing_prod = UserPreference.load(371137)
        lengjing_prod.llm_model = GPT_4
        lengjing_prod.mode_type = MODE_TYPE_LENGJING
        lengjing_prod.language_code = LANGUAGE_CN
        Preference.upsert(s, 371137, lengjing_prod.to_json())

        # 372819 tiinsight 80T0RFD0
        tiinsight_prod = UserPreference.load(372819)
        tiinsight_prod.llm_model = GPT_4
        tiinsight_prod.mode_type = MODE_TYPE_TIINSIGHT
        Preference.upsert(s, 372819, tiinsight_prod.to_json())

        # 372820 ask_billing S0T0RH50
        ask_billing_prod = UserPreference.load(372820)
        ask_billing_prod.llm_model = GPT_4
        ask_billing_prod.mode_type = MODE_TYPE_ASK_BILLING
        Preference.upsert(s, 372820, ask_billing_prod.to_json())

        # 372821 spider 40T0RJ40
        spider_prod = UserPreference.load(372821)
        spider_prod.llm_model = GPT_4
        spider_prod.mode_type = MODE_TYPE_SPIDER
        Preference.upsert(s, 372821, spider_prod.to_json())


def add_user_preference():
    if env == "prod":
        add_user_preference_prod()
    elif env == "staging":
        pass
    elif env == "dev":
        add_user_preference_dev()


if __name__ == "__main__":
    add_user_preference()