import joblib
from pydantic import BaseModel
from src.dup_ques.preprocess import get_x

# SCHEMA
class Schema(BaseModel):
	ques1: str
	ques2: str

# Request Handler
def dup_ques(req):
	ques1 = req.ques1
	ques2 = req.ques2

	X = get_x(ques1, ques2)
	y = predict(X)

	return y

# PIPELINE
PIPELINE_PATH = "./src/dup_ques/pipeline.pkl"

def predict(X):
	pipeline = joblib.load(PIPELINE_PATH)
	return pipeline.predict_proba(X).round(3).tolist()