import gradio as gr
from transformers import AutoModelForSeq2SeqLM, AutoTokenizer

# 영어-한국어 번역 모델 로드
model_name = "t5-base"
model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
tokenizer = AutoTokenizer.from_pretrained(model_name)

def translate_text(input_text):
    # 입력 텍스트 토큰화
    input_ids = tokenizer.encode(input_text, return_tensors="pt")
    # 번역 모델 실행
    output = model.generate(input_ids, max_length=50)
    # 출력 텍스트 생성
    output_text = tokenizer.decode(output[0], skip_special_tokens=True)
    return output_text

# Gradio 애플리케이션 구축
iface = gr.Interface(
    fn=translate_text,
    inputs="text",
    outputs="text",
    title="English-Korean Translator",
    description="Enter English text to translate to Korean."
)

iface.launch()