app.py

import gradio as gr
from transformers import MarianMTModel, MarianTokenizer

# 모델과 토크나이저 로드
model_name = "Helsinki-NLP/opus-mt-ko-it"
tokenizer = MarianTokenizer.from_pretrained(model_name)
model = MarianMTModel.from_pretrained(model_name)

# 번역 함수
def translate_korean_to_italian(text):
    if not text.strip():
        return "⚠️ 번역할 문장을 입력하세요."
    inputs = tokenizer(text, return_tensors="pt", padding=True)
    translated = model.generate(**inputs)
    result = tokenizer.decode(translated[0], skip_special_tokens=True)
    return result

# Gradio 인터페이스 구성
interface = gr.Interface(
    fn=translate_korean_to_italian,
    inputs=gr.Textbox(lines=2, label="한국어 입력"),
    outputs=gr.Textbox(label="이탈리아어 번역"),
    title="한국어 → 이탈리아어 번역기",
    description="한국어 문장을 입력하면 이탈리아어로 번역합니다. 모델: Helsinki-NLP/opus-mt-ko-it"
)

if __name__ == "__main__":
    interface.launch()