06_PREDICCION.py

# 06_PREDICCION.py

import os
import joblib
import pandas as pd
from dotenv import load_dotenv

# --- Carga de configuración desde .env ---
load_dotenv()
ruta_base = os.getenv("PROYECTO_RUTA_BASE")

# Validar que la ruta base se haya cargado
if not ruta_base:
    raise ValueError("La variable 'PROYECTO_RUTA_BASE' no se encontró en el archivo .env.")

# --- Carga del modelo ---

# Construir la ruta al modelo de forma dinámica
ruta_modelo = os.path.join(ruta_base, "PROCESADO", "modelo_prioridad_pipeline.joblib")

# Cargar el pipeline desde la ruta construida
pipeline = joblib.load(ruta_modelo)

def predecir_texto(texto):
    # Crear DataFrame con columnas que espera el pipeline
    df = pd.DataFrame([{
        "Categoria": "Desconocido",
        "Asunto": texto,
        "Grupo_Asignado": "Desconocido",
        "Area_Solicitante": "Desconocido"
    }])
    
    pred = pipeline.predict(df)[0]
    proba = pipeline.predict_proba(df)[0]
    
    print(f"📝 Texto: {texto}")
    print(f"🎯 Predicción: {pred}")
    print("📊 Probabilidades:")
    for clase, p in zip(pipeline.classes_, proba):
        print(f"   {clase}: {p:.2f}")
    print()

# Ejemplo de uso
predecir_texto("El sistema está caído en toda la planta, no funciona nada.")
predecir_texto("El usuario olvidó su contraseña de acceso al correo.")