import gradio as gr
from transformers import pipeline

# Hugging Face의 Helsinki-NLP Ko->Ru 번역기 사용 
translator = pipeline("translation_ko_to_ru", model="Helsinki-NLP/opus-mt-ko-ru")

def chat_translate(user_input):
    if not user_input.strip():
        return "입력을 해주세요!"
    
    # 번역
    result = translator(user_input)
    translated_text = result[0]['translation_text']
    
    return translated_text

# 간단한 Gradio 챗 인터페이스
iface = gr.ChatInterface(
    fn=chat_translate, 
    chatbot_name="한국어->러시아어 번역 챗봇",
    title="한국어 입력시 러시아어 번역 챗봇",
    description="한국어로 질문하면 러시아어로 번역해드립니다."
)

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