import gradio as gr
from transformers import pipeline

# 문법 체크를 위한 'fill-mask' 파이프라인 로드
grammar_checker = pipeline('fill-mask', model='bert-base-uncased')

def check_grammar(sentence):
    # 입력 문장에서 문법 체크
    result = grammar_checker(sentence)
    
    # 반환된 결과 중 가장 가능성이 높은 단어 선택
    suggested_word = result[0]["token_str"]
    
    return suggested_word

# Gradio 애플리케이션 정의
gr.Interface(check_grammar, 
             inputs=gr.inputs.Textbox(lines=7, label="Enter a sentence to check grammar"), 
             outputs=gr.outputs.Textbox(label="Corrected sentence"), 
             title="Grammar Checker").launch()