import joblib
import pandas as pd
from huggingface_hub import hf_hub_download

model_repo = "Pratik26Dec/TourismModel"

model_file = hf_hub_download(repo_id=model_repo, filename="model.pkl")
encoders_file = hf_hub_download(repo_id=model_repo, filename="label_encoders.pkl")

model = joblib.load(model_file)
label_encoders = joblib.load(encoders_file)

def preprocess(df):
    for col in df.columns:
        if col in label_encoders:
            df[col] = label_encoders[col].transform(df[col].astype(str))
    return df

def predict(input_data: dict):
    df = pd.DataFrame([input_data])
    df = preprocess(df)
    pred = model.predict(df)[0]
    return int(pred)