Update app.py

app.py

@@ -1,51 +1,55 @@
 import gradio as gr
-from transformers import MarianMTModel, MarianTokenizer
+import requests
+import json
 
-# 한국어 -> 영어 모델 로드
-model_name_ko_en = "Helsinki-NLP/opus-mt-ko-en"
-tokenizer_ko_en = MarianTokenizer.from_pretrained(model_name_ko_en)
-model_ko_en = MarianMTModel.from_pretrained(model_name_ko_en)
+# Naver Papago API를 사용하여 번역하는 함수
+def translate_text(text, source_lang, target_lang):
+    # 네이버 개발자 센터에서 발급받은 Client ID와 Client Secret
+    client_id = "YOUR_NAVER_CLIENT_ID"  # 여기에 네이버에서 발급받은 Client ID를 입력하세요.
+    client_secret = "YOUR_NAVER_CLIENT_SECRET"  # 여기에 네이버에서 발급받은 Client Secret을 입력하세요.
 
-# 영어 -> 한국어 모델 로드
-model_name_en_ko = "Helsinki-NLP/opus-mt-en-ko"
-tokenizer_en_ko = MarianTokenizer.from_pretrained(model_name_en_ko)
-model_en_ko = MarianMTModel.from_pretrained(model_name_en_ko)
+    # Papago API 주소
+    url = "https://openapi.naver.com/v1/papago/n2mt"
+
+    # 요청 헤더 (네이버 API 키를 포함)
+    headers = {
+        "X-Naver-Client-Id": client_id,
+        "X-Naver-Client-Secret": client_secret,
+    }
+
+    # 요청 데이터 (원본 언어, 목표 언어, 번역할 텍스트)
+    data = {
+        "source": source_lang,
+        "target": target_lang,
+        "text": text,
+    }
 
-def translate_text(text, source_lang, target_lang):
-    """
-    텍스트를 번역하는 함수.
-    """
     try:
-        if source_lang == "한국어" and target_lang == "영어":
-            # 한국어 -> 영어 번역
-            tokenized = tokenizer_ko_en(text, return_tensors="pt", padding=True)
-            translated = model_ko_en.generate(**tokenized)
-            result = tokenizer_ko_en.decode(translated[0], skip_special_tokens=True)
-        elif source_lang == "영어" and target_lang == "한국어":
-            # 영어 -> 한국어 번역
-            tokenized = tokenizer_en_ko(text, return_tensors="pt", padding=True)
-            translated = model_en_ko.generate(**tokenized)
-            result = tokenizer_en_ko.decode(translated[0], skip_special_tokens=True)
-        else:
-            return "지원하지 않는 언어 조합입니다."
-        return result
+        # Papago API에 요청 보내기
+        response = requests.post(url, headers=headers, data=data)
+        # 응답을 JSON 형식으로 변환
+        result = response.json()
+        # 번역된 텍스트 추출
+        translated_text = result["message"]["result"]["translatedText"]
+        return translated_text
     except Exception as e:
+        # 오류 발생 시 메시지 반환
         return f"오류가 발생했습니다: {str(e)}"
 
-# Gradio UI 구성
+# Gradio를 사용하여 웹 인터페이스 만들기
 with gr.Blocks() as demo:
-    gr.Markdown("# 번역기")
+    gr.Markdown("# 한국어 ↔ 영어 번역기")
 
     # 언어 선택
     source_lang = gr.Radio(
-        choices=["한국어", "영어"],
+        choices=["ko", "en"],  # 원본 언어 선택 (한국어: ko, 영어: en)
         label="원본 언어",
-        value="한국어"
+        value="ko"
     )
     target_lang = gr.Radio(
-        choices=["한국어", "영어"],
+        choices=["ko", "en"],  # 목표 언어 선택 (한국어: ko, 영어: en)
         label="목표 언어",
-        value="영어"
+        value="en"
     )
 
     # 입력 텍스트
@@ -59,9 +63,9 @@ with gr.Blocks() as demo:
 
     # 번역 함수 연결
     translate_button.click(
-        fn=translate_text,
-        inputs=[input_text, source_lang, target_lang],
-        outputs=translation_output
+        fn=translate_text,  # 번역 함수
+        inputs=[input_text, source_lang, target_lang],  # 입력값
+        outputs=translation_output  # 출력값
     )
 
 # 메인 실행부