import gradio as gr

from transformers import AutoModelForQuestionAnswering, AutoTokenizer, pipeline

model_name = "deepset/roberta-base-squad2"

nlp = pipeline('question-answering', model=model_name, tokenizer=model_name)

def chat(context, question):
  QA_input = {
      "question" : question,
      "context" : context
  }
  res = nlp(QA_input)

  return res['answer']

screen = gr.Interface(
    fn = chat,
    inputs = [gr.Textbox(lines = 8, placeholder = "Enter your context here 👉"), gr.Textbox(lines = 2, placeholder = "Enter your question here 👉")],
    outputs = gr.Textbox(lines = 10, placeholder = "Your answer will be here soon 🚀"),
    title="Facilitating the QnA with roberta-base-squad2 👩🏻‍💻📓✍🏻💡",
    description="This app aims to facilitate the simple QnA with the provided context💡",
    theme="soft",
    article = """### Disclaimer : This model is purely used for QnA. User is expected to paste the text from which they want the answer in **context** section. <br> Then paste the question in the **question** section. <br> User will get the answer in the answer section."""
    
)

screen.launch()