import gradio as gr
import surprise
from surprise import dump

file_name = 'modelo_entrenado.sav'
model = dump.load(file_name)[1]


def predict_rating(userId, id):
    prediction = model.predict(userId, id)

    if prediction.est >= 3.5:
        mensaje = "Es hora de verla!!", prediction.est
    elif prediction.est >= 2.5 and prediction.est < 3.5:
        mensaje = "Puedes verla luego!", prediction.est
    else:
        mensaje = "No es la película que esta buscando!", prediction.est

    return mensaje

iface = gr.Interface(fn=predict_rating, 
                     inputs= [gr.inputs.Textbox(lines=1,placeholder="ingrese su número de usuario aquí"), 
                             gr.inputs.Textbox(lines=1,placeholder="ingrese su número de usuario aquí")],
                     outputs="text")
iface.launch()