src/predict.py

import joblib
from preprocess import clean_text
import random

# Load model & responses
model = joblib.load("models/lms_chatbot.joblib")
responses = joblib.load("models/responses.joblib")

def chatbot_response(user_input: str) -> str:
    cleaned = clean_text(user_input)
    tag = model.predict([cleaned])[0]
    return random.choice(responses.get(tag, ["Sorry, I don't understand."]))