import gradio as gr
from load_data import *

def chitti(query):
    re_table = search(query)
    return [re_table[0][0],re_table[0][3],re_table[1][0],re_table[1][3]]

demo = gr.Interface(
    fn=chitti,
    inputs=["text"],
    outputs=["text","text","text","text"],
)
demo.launch()