import gradio as gr
from transformers import pipeline

context = ""
def SetContext(textfile):
             with open(textfile, 'r') as file:
                     context = file.read()
SetContext('WritingCarePlans.txt')
question = "What should be documented in a care plan?"
title = "Transformers - Sentence to Paragraph - For Mindfulness"
examples = [
    ["Break the cycle of stress and anxiety"],
    ["Feel calm in stressful situations"],
    ["Deal with work pressure"],
    ["Learn to reduce feelings of overwhelmed"]
]
model1 = gr.Interface.load("huggingface/deepset/roberta-base-squad2")

def f1(x,y):
    return model1(x,y)


demo = gr.Blocks()
with demo:
    inputs1=[gr.inputs.Textbox(lines=40, default=context, label="Context paragraph"),gr.inputs.Textbox(lines=10, default=question, label="Question")]
    out1=gr.Textbox()
    b1 = gr.Button("roberta")

    inp = gr.Textbox(placeholder="What is a care plan for?")

    inp.change(fn=f1, 
               inputs=inputs1,
               outputs=out1)
               
demo.launch()