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 == "short":
        #return gr.Textbox.update(lines=2, visible=True)
    elif choice == "long":
        textin[:-50] + api(textin[-50:])
        #return gr.Textbox.update(lines=8, visible=True)
    else:
        return gr.Textbox.update(visible=False)

with gr.Blocks() as block:
    radio = gr.Radio(
        ["short", "long", "none"], label="What kind of essay would you like to write?"
    )
    textin = gr.Textbox(lines=2, interactive=True)
    textout = gr.Textbox(lines=2, interactive=True)
    radio.change(fn=change_textbox, inputs=radio, textin, outputs=textout)
    block.launch()