from gradio_client import Client
import gradio as gr

# Hugging Face Space 이름
SPACE_NAME = "Kims12/English_translator"

# Hugging Face 토큰이 필요한 경우 아래에 입력하세요. (비공개 Space일 경우)
# TOKEN = "your_huggingface_token"

# Client 초기화 (토큰이 필요한 경우 Client(SPACE_NAME, token=TOKEN))
client = Client(SPACE_NAME)

def translate_to_korean(english_text):
    """
    영어 텍스트를 한국어로 번역하는 함수
    """
    try:
        # API 엔드포인트 호출
        result = client.predict(
            english_text=english_text,
            api_name="/translate_to_korean"
        )
        return result
    except Exception as e:
        return f"번역 중 오류가 발생했습니다: {e}"

# Gradio 인터페이스 구성
with gr.Blocks() as demo:
    gr.Markdown("## 영어 텍스트를 한국어로 번역하기")

    with gr.Row():
        with gr.Column():
            english_input = gr.Textbox(
                label="영어 텍스트",
                placeholder="번역할 영어 문장을 입력하세요."
            )
            translate_button = gr.Button("번역하기")
        with gr.Column():
            korean_output = gr.Textbox(
                label="한국어 번역",
                interactive=False
            )
    
    # 버튼 클릭 시 번역 함수 실행
    translate_button.click(
        fn=translate_to_korean,
        inputs=english_input,
        outputs=korean_output
    )

# 애플리케이션 실행
if __name__ == "__main__":
    demo.launch()