config.py

import os
import torch
from huggingface_hub import hf_hub_download

class Config: 
    DB_PATH = '/tmp/dataset.db'
    TABLE_NAME = 'mojica_Ventas'
    EXMAPLES_JSON = './examples.json'
    MODEL_NAME = "ibm-granite/granite-3b-code-instruct-128k"  
    CSV_PATH = "/kaggle/input/mojica-hoja-1/mojica_hoja_1.csv"
    MAX_HISTORY = 3  # Mantener las últimas 3 interacciones (memoria)
    MAX_TOKENS = 4096  
    MAX_NEW_TOKENS = 256  
    DEVICE = "cuda" if torch.cuda.is_available() else "cpu"

if not os.path.exists(Config.DB_PATH):
    Config.DB_PATH = hf_hub_download(
        repo_id="TaquitoTomatoe/MojicaData",
        filename="dataset.db",
        repo_type="dataset",
        token=os.getenv("HF_TOKEN")  
    )