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, placeholder="한국어로 질문을 입력하세요"),
    outputs=gr.Textbox(label="이탈리아어 번역"),
    title="한국어 → 이탈리아어 번역 챗봇",
    description="한국어로 입력하면 이탈리아어로 번역해드립니다. Hugging Face의 Helsinki-NLP 모델 사용."
)

# 앱 실행
if __name__ == "__main__":
    interface.launch()