from utils import *
import gradio as gr


def main(file, ext_threshold, article_type):
    paper = read_text_to_json(file.name)
    if not is_valid_format(paper):
        return "invalid_format"
    sentJson = convert_to_sentence_json(paper)
    sentFeat = extract_sentence_features(sentJson)

    ExtModel = load_ExtModel('model/LGB_model_F10_S.pkl')
    ext = extractive_method(sentJson, sentFeat, ExtModel, TGB=False)
    
    return ext
    


# 定義Gradio介面
iface = gr.Interface(
    fn=main,
    inputs=[
        gr.inputs.File(),
        gr.inputs.Slider(minimum=0.5, maximum=1, default=0.5, step=0.01, label="Extractive - Threshold"),
        gr.inputs.Dropdown(["non-specialized field", "biomedical field"],default="non-specialized field", label="Abstractive - Field")
    ],
    outputs=gr.outputs.Textbox(label="Output - Structured Abstract"),
    title="Ext-Abs-StructuredSum",
    description="please upload a .txt file formatted in the form of the example.",
    # examples=[['text.txt']],
    allow_flagging='never'
)

# 啟動Gradio介面
iface.launch(share=False)  # share=False 用於停用分享模式