import gradio as gr
import pandas as pd
from retrieval import *

df = pd.read_csv("Responses.csv")
text = list(df["text"].values)


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

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