# app.py

import gradio as gr
from gradio_client import Client

# Hugging Face Space의 클라이언트 설정
client = Client("Kims12/English_translator")

def translate_to_korean(english_text):
    """
    영어 텍스트를 한국어로 번역하는 함수.
    
    Parameters:
        english_text (str): 번역할 영어 텍스트
    
    Returns:
        str: 번역된 한국어 텍스트
    """
    try:
        result = client.predict(
            english_text=english_text,
            api_name="/translate_to_korean"
        )
        return result
    except Exception as e:
        return f"번역 중 오류가 발생했습니다: {str(e)}"

# Gradio 인터페이스 구성
with gr.Blocks() as demo:
    gr.Markdown("# 영어-한국어 번역기")
    with gr.Row():
        with gr.Column():
            english_input = gr.Textbox(
                label="영어 텍스트",
                placeholder="여기에 영어 문장을 입력하세요...",
                lines=4
            )
            translate_button = gr.Button("번역하기")
        with gr.Column():
            korean_output = gr.Textbox(
                label="한국어 번역",
                placeholder="번역 결과가 여기에 표시됩니다...",
                lines=4
            )
    
    # 버튼 클릭 시 번역 함수 호출
    translate_button.click(
        fn=translate_to_korean,
        inputs=english_input,
        outputs=korean_output
    )

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