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 = joblib.load("./src/dup_ques/pipeline.pkl")

def predict(X):
	return pipeline.predict_proba(X).round(3).tolist()