import gradio as gr
import pickle
import numpy as np

class QModel():
    def __init__(self):
        self.qtable = qtab
        
    #returns the top n states
    def predict(self,state_n, n):
        return np.argsort(self.qtable[state_n][:])[-n:]

learner = pickle.load(open("q-learn-multiply-game.pkl", "rb"))

def predict_next_questions(state,n_question):
    return learner.predict(state, n_questions)

intf = gr.Interface(fn=predict_next_questions, inputs=["number", "number"], outputs="number")
intf.launch