import gradio as gr
from handler import handler


def predict(action, query):
    try:
        return handler(action, query)
    except Exception as e:
        return {"error": e}


with gr.Blocks() as app:
    action_type = gr.Dropdown(label="Type", value="Search", choices=[
                              "Search", "AutoSuggest"])
    query = gr.Textbox(label="Query")
    output = gr.Textbox(label="Output Box")
    greet_btn = gr.Button("Action!")
    greet_btn.click(
        fn=predict,
        inputs=[action_type, query], outputs=output, api_name="api")

app.launch()