import subprocess
subprocess.run(["pip", "install", "./textgen.zip"])

import gradio as gr
from textgen import TextGenerator

textgen = TextGenerator()
textgen.load("textgen.bin")

def generate_examples():
    return textgen.generate(length=100).split("\n")

def generate(text, length=50):
    return textgen.generate(text, length=length)

def main():
    with gr.Blocks(title="🦊 textgen") as demo:
        gr.Markdown("# 🦊 textgen")
        with gr.Group():
            text = gr.TextArea(label="Text")
        length = gr.Number(label="Length", value=50)
        with gr.Row():
            btn = gr.Button("Generate", variant="primary")
            clear_btn = gr.Button("Clear")
        examples = gr.Examples(
            inputs=[text]
        )
        demo.load(generate_examples, [], examples)
        btn.click(generate, [text, length], [text])
        clear_btn.click(lambda: gr.update(value=""), [], [text])
    demo.queue().launch(debug=True)

if __name__ == "__main__":
    # Launch interface
    main()