import os
from dotenv import load_dotenv

load_dotenv(override=True)

ROOT_DATA_PATH = "data/customer_data"

JENKINS_URL: str = os.getenv("JENKINS_URL")
JENKINS_USERNAME: str = os.getenv("JENKINS_USERNAME")
JENKINS_API_TOKEN: str = os.getenv("JENKINS_API_TOKEN")

AI_MODEL_DTLS_URL: str = os.getenv("AI_MODEL_DTLS_URL")
AI_MODEL_TRAINING_URL: str = os.getenv("AI_MODEL_TRAINING_URL")
AI_MODEL_TRAINING_HSTRY_URL: str = os.getenv("AI_MODEL_TRAINING_HSTRY_URL")
AI_MODEL_RETRAINING_URL: str = os.getenv("AI_MODEL_RETRAINING_URL")
AI_MODEL_ID_GENERATION_URL: str = os.getenv("AI_MODEL_ID_GENERATION_URL")

LLM_PRIVATE_FLAN_URL: str = os.getenv("LLM_PRIVATE_FLAN_URL")
YOLO_DETECTION_URL: str = os.getenv("YOLO_DETECTION_URL")
YOLO_CLASSIFICATION_URL: str = os.getenv("YOLO_CLASSIFICATION_URL")
LLM_PRIVATE_NAS_URL: str = os.getenv("LLM_PRIVATE_NAS_URL")
TIME_SPACE_URL: str = os.getenv("TIME_SPACE_URL")
EVENT_GRAPH_MODEL_URL: str = os.getenv("EVENT_GRAPH_MODEL_URL")

S3_ACCESS_KEY: str = os.getenv("S3_ACCESS_KEY")
S3_SECRET_KEY: str = os.getenv("S3_SECRET_KEY")

BUCKET_NAME: str = os.getenv("BUCKET_NAME", "eizen-dev")
EIZEN_DMS_STORAGE_SERVER_URL: str = os.getenv("EIZEN_DMS_STORAGE_SERVER_URL", 'dms-storage.eizen.ai')

MODEL_UPDATE_URL: str = os.getenv("MODEL_UPDATE_URL")
HA_PROCESS_UPDATE_URL: str = os.getenv("HA_PROCESS_UPDATE_URL")

SEND_TO_DVC: bool = os.getenv("SEND_TO_DVC", True)

LLM_PRIVATE_FLAN_MODEL_NAME: str = 'google/flan-t5-small'
LLM_PRIVATE_FLAN_BOTTLENECK_SIZE: int = 512
LLM_PRIVATE_FLAN_BATCH_SIZE: int = 8
LLM_PRIVATE_FLAN_LEARNING_RATE: float = 1e-4

LLM_PRIVATE_NAS_MODEL_NAME: str = 'google/flan-t5-small'
LLM_PRIVATE_NAS_LEARNING_RATE: float = 0.0001
LLM_PRIVATE_NAS_WEIGHT_DECAY: float = 0.01
LLM_PRIVATE_NAS_BATCH_SIZE: int = 8
LLM_PRIVATE_NAS_GRADIENT_CLIPPING: float = 1.0

YOLO_DETECTION_DATA_FILE_FORMAT: str = 'labelstudio'