/pronunciation api 추가

app.py

@@ -28,20 +28,23 @@ pipe = pipeline(
 # 3. 요청 본문(Request Body)의 데이터 형식을 지정
 #    프론트엔드에서 "text"라는 키에 번역할 문장을 담아서 보내야 한다는 규칙.
 class TranslationRequest(BaseModel):
-    text: str
-    target_lang: str # 'english' 또는 'japanese'
+    korean_text: str
+    target_lang: str # 'en' 또는 'ja'
 
-# 4. API 엔드포인트(Endpoint) 생성
-#    "/translate" 라는 주소로 POST 요청이 들어왔을 때 이 함수가 실행됨.
+# 발음 변환 API를 위한 주문서 양식
+class PronunciationRequest(BaseModel):
+    japanese_text: str
+
+# 4. "/translate" API 엔드포인트 생성
 @app.post("/translate")
 async def translate(request: TranslationRequest):
     korean_text = request.text
     target_lang = request.target_lang
 
     # 프롬프트 엔지니어링: 모델에게 원하는 결과물을 명확하게 지시
-    if target_lang == 'english':
+    if target_lang == 'en':
         prompt = f"Translate the following Korean sentence into natural, everyday English. Provide only the translated sentence, without any additional explanations or quotation marks.\n\nKorean: \"{korean_text}\"\n\nEnglish:"
-    elif target_lang == 'japanese':
+    elif target_lang == 'ja':
         prompt = f"Translate the following Korean sentence into natural, everyday Japanese. Provide only the translated sentence, without any additional explanations or quotation marks.\n\nKorean: \"{korean_text}\"\n\nJapanese:"
     else:
         return {"error": "Invalid target language"}
@@ -65,9 +68,34 @@ async def translate(request: TranslationRequest):
     # 'assistant\n' 다음에 오는 실제 번역 결과만 깔끔하게 잘라내기
     translated_text = generated_text.split("assistant\n")[-1].strip()
 
-    # 5. 프론트엔드에 번역된 텍스트를 JSON 형태로 반환
+    # 프론트엔드에 번역된 텍스트를 JSON 형태로 반환
     return {"translated_text": translated_text}
 
+# 5. "/pronunciation" API 엔드포인트 생성
+@app.post("/pronunciation")
+async def get_pronunciation(request: PronunciationRequest):
+    japanese_text = request.japanese_text
+
+    # 발음 변환을 위한 새로운 프롬프트
+    prompt = f"Provide the Korean pronunciation (Hangul) for the following Japanese sentence. Provide only the Hangul pronunciation, without any other text, labels, or quotation marks.\n\nJapanese: \"{japanese_text}\"\n\nKorean Pronunciation:"
+    
+    messages = [
+        {"role": "user", "content": prompt}
+    ]
+
+    outputs = pipe(
+        messages,
+        max_new_tokens=150,
+        do_sample=True,
+        temperature=0.7,
+        top_k=50,
+    )
+
+    generated_text = outputs[0]["generated_text"]
+    pronunciation_text = generated_text.split("assistant\n")[-1].strip()
+
+    return {"pronunciation_text": pronunciation_text}
+
 # 서버가 잘 작동하는지 확인하기 위한 기본 주소
 @app.get("/")
 def read_root():