import gradio as gr
import subprocess


def generate_text(length, nsamples, prefix, temperature):
    # 构建命令行参数
    my_prefix = "--prefix=" + prefix + ","
    args = ["python", "generate.py", f"--length={length}", f"--nsamples={nsamples}", my_prefix, f"--temperature={temperature}", "--model_path=model/model_epoch10"]
    
    # 执行命令并获取输出
    process = subprocess.Popen(args, stdout=subprocess.PIPE)
    output, error = process.communicate()
    output = output.decode("utf-8")
    
    return output

input_length = gr.Number(label="生成文本长度", min_value=1, max_value=10000, default=1000)
input_nsamples = gr.Number(label="生成文本数量", min_value=1, max_value=10, default=3)
input_prefix = gr.Textbox(label="起始文本")
input_temperature = gr.Slider(label="生成温度", minimum=0.1, maximum=2.0, default=0.7, step=0.1)
output_text = gr.Textbox(label="生成的文本")

title = "GPT2中文文本生成器"
description = "输入起始文本、生成文本长度、生成文本数量和生成温度，生成中文文本。"

gr.Interface(fn=generate_text, inputs=[input_length, input_nsamples, input_prefix, input_temperature], outputs=output_text, title=title, description=description).launch()