Update app.py

app.py

@@ -22,10 +22,31 @@ def translate_text(text, source_lang, target_lang):
     """
     try:
         client = get_client()
-        # 프롬프트를 명확하게 설정
-        prompt = f"Translate the following text from {source_lang} to {target_lang}: '{text}'. Do not add any extra text or explanations, just provide the translation."
-        response = client.text_generation(prompt, max_new_tokens=50)  # 토큰 수를 적절히 제한
-        return response.strip()  # 불필요한 공백 제거
+
+        # 프롬프트 규칙 설정
+        if source_lang == "한국어" and target_lang == "영어":
+            prompt = f"""
+            You are a 30-year veteran English translator.
+            You are renowned for your accurate and meticulous expressions.
+            Translate the following text from Korean to English:
+            '{text}'
+            - Translate accurately according to the context and flow of the content.
+            - Do not add any extra text or explanations, just provide the translation.
+            """
+        elif source_lang == "영어" and target_lang == "한국어":
+            prompt = f"""
+            너는 30년차 한국어 전문 번역가이다.
+            정확하고 섬세한 표현력으로 유명한 한국어 전문 번역가이다.
+            다음 텍스트를 영어에서 한국어로 번역하라:
+            '{text}'
+            - 문맥과 내용의 흐름에 맞춰 정확한 번역을 한다.
+            - 추가적인 설명 없이 번역 결과만 제공하라.
+            """
+        else:
+            return "지원하지 않는 언어 조합입니다."
+
+        response = client.text_generation(prompt, max_new_tokens=200, temperature=0.7, top_p=0.95)
+        return response.strip()
     except Exception as e:
         return f"오류가 발생했습니다: {str(e)}"
 
@@ -54,6 +75,12 @@ with gr.Blocks() as demo:
         # 번역 결과
         translation_output = gr.Textbox(label="번역 결과", lines=5, interactive=False)
 
+        # 고급 설정
+        with gr.Accordion("고급 설정", open=False):
+            max_tokens = gr.Slider(minimum=0, maximum=2000, value=500, step=100, label="Max Tokens")
+            temperature = gr.Slider(minimum=0.1, maximum=2.0, value=0.7, step=0.05, label="Temperature")
+            top_p = gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p")
+
         # 번역 버튼
         translate_button = gr.Button("번역")