import gradio as gr

#gr.Interface.load("models/bigscience/bloom").launch()
api = gr.Interface.load("models/bigscience/bloom")

def change_textbox(choice, textin):
    if choice == "Expository":
        return textin[:-50] + api(textin[-50:])
    elif choice == "Narrative":
        return textin[:-50] + api(textin[-50:])
    else:
        return textin[:-50] + api(textin[-50:])

with gr.Blocks() as block:
    radio = gr.Radio(
        ["Expository", "Narrative", "Opinion"], label="What kind of essay would you like to write?"
    )
    textin = gr.Textbox(lines=2, interactive=True)
    textin.update(lines=8, visible=True)
    
    textout = gr.Textbox(lines=2, interactive=True)
    textout.update(lines=8, visible=True)

    radio.change(fn=change_textbox, inputs=[radio, textin], outputs=textout)
    block.launch()