from django.apps import AppConfig
import os
import joblib
from django.conf import settings

class MaxhunterapiConfig(AppConfig):
    default_auto_field = 'django.db.models.BigAutoField'
    name = 'MaxHunterAPI'

class ApiConfig(AppConfig):
    name = 'api'
    
    MODEL_FILE1 = os.path.join(settings.MODELS, "LogitModel_Lead_Qualificacao.joblib")
    model1 = joblib.load(MODEL_FILE1)
    
    MODEL_FILE2 = os.path.join(settings.MODELS, "LogitModel_Qualificacao_1Reuniao.joblib")
    model2 = joblib.load(MODEL_FILE2)

    MODEL_FILE3 = os.path.join(settings.MODELS, "LogitModel_1Reuniao_2Reuniao.joblib")
    model3 = joblib.load(MODEL_FILE3)

    MODEL_FILE4 = os.path.join(settings.MODELS, "LogitModel_2Reuniao_Assinatura.joblib")
    model4 = joblib.load(MODEL_FILE4)

    MODEL_FILE5 = os.path.join(settings.MODELS, "LogitModel_Assinatura_Recrutado.joblib")
    model5 = joblib.load(MODEL_FILE5)