test

app.py

@@ -42,17 +42,70 @@ def translate(source, target, text):
 
     return translated_text
 
+def conversion(source, text):
+    formats = {
+                "formal":"문어체",
+                "informal":"구어체",
+                "android":"안드로이드",
+                "azae":"아재",
+                "chat":"채팅",
+                "choding":"초등학생",
+                "emoticon":"이모티콘",
+                "enfp":"enfp",
+                "gentle":"신사",
+                "halbae":"할아버지",
+                "halmae":"할머니",
+                "joongding":"중학생",
+                "king":"왕",
+                "naruto":"나루토",
+                "seonbi":"선비",
+                "sosim":"소심한",
+                "translator":"번역기",
+            }
+    style = formats[source]
+
+    input_text = f"{style} 형식으로 변환:" + text
+
+    converter = pipeline(
+            'text2text-generation',
+            model=style_model,
+            tokenizer=style_tokenizer,
+        )
+
+    output = converter(input_text)
+    generated_text = output[0]['generated_text']
+
+    return generated_text
+
 title = 'KoTAN Translator & Speech-style converter'
 lang = ['English','Korean']
+style = ['formal', 'informal', 'android', 'azae', 'chat', 'choding', 'emoticon', 'enfp', \
+         'gentle', 'halbae', 'halmae', 'joongding', 'king', 'naruto', 'seonbi', 'sosim', 'translator']
+
+with gr.Blocks() as demo:
+    gr.Markdown("KoTAN: Korean Translation and Augmentation with fine-tuned NLLB. You can exercise translation tasks, and speech-style conversion. \
+                If you want to download as pip package, please visit our github. (https://github.com/KoJLabs/KoTAN)")
+    
+    with gr.Tab("Translation"):
+        translator_app = gr.Interface(
+            fn=translate,
+            inputs=[gr.inputs.Dropdown(choices=lang, label='Source Language'), gr.inputs.Dropdown(choices=lang, label='Target Language'), gr.inputs.Textbox(lines=5, label='Text to Translate')],
+            outputs=[gr.outputs.Textbox(label='Translated Text')],
+            title=title,
+            # description = 'KoTAN: Korean Translation and Augmentation with fine-tuned NLLB. If you want to download as pip package, please visit our github. (https://github.com/KoJLabs/KoTAN)',
+            # article='Jisu, Kim. Juhwan, Lee',
+            enable_queue=True,
+        )
+    
+    with gr.Tab("Speech-style conversion"):
+        translator_app = gr.Interface(
+            fn=conversion,
+            inputs=[gr.inputs.Dropdown(choices=style, label='Speech Style'), gr.inputs.Textbox(lines=5, label='Text to style conversion')],
+            outputs=[gr.outputs.Textbox(label='Converted Text')],
+            title=title,
+            # description = 'KoTAN: Korean Translation and Augmentation with fine-tuned NLLB. If you want to download as pip package, please visit our github. (https://github.com/KoJLabs/KoTAN)',
+            # article='Jisu, Kim. Juhwan, Lee',
+            enable_queue=True,
+        )
 
-translator_app = gr.Interface(
-    fn=translate,
-    inputs=[gr.inputs.Dropdown(choices=lang, label='Source Language'), gr.inputs.Dropdown(choices=lang, label='Target Language'), gr.inputs.Textbox(lines=5, label='Text to Translate')],
-    outputs=[gr.outputs.Textbox(label='Translated Text')],
-    title=title,
-    description = 'KoTAN: Korean Translation and Augmentation with fine-tuned NLLB. If you want to download as pip package, please visit our github. (https://github.com/KoJLabs/KoTAN)',
-    article='Jisu, Kim. Juhwan, Lee',
-    enable_queue=True,
-)
-
-translator_app.launch()
+demo.launch()