Update app.py

app.py

@@ -76,6 +76,16 @@ def tts_fn(text):
         x_tst_lengths = torch.LongTensor([stn_tst.size(0)])
         audio = net_g.infer(x_tst, x_tst_lengths, noise_scale=.667, noise_scale_w=0.8, length_scale=1)[0][0,0].data.float().numpy()
     return "Success", (hps.data.sampling_rate, audio)
+    
+def jtts_fn(text):
+    if len(text) > 150:
+        return "Error: Text is too long", None
+    stn_tst = get_text(japanese_phrase_cleaners(text), hps)
+    with torch.no_grad():
+        x_tst = stn_tst.unsqueeze(0)
+        x_tst_lengths = torch.LongTensor([stn_tst.size(0)])
+        audio = net_g.infer(x_tst, x_tst_lengths, noise_scale=.667, noise_scale_w=0.8, length_scale=1)[0][0,0].data.float().numpy()
+    return "Success", (hps.data.sampling_rate, audio)
 
 
 if __name__ == '__main__':
@@ -84,11 +94,20 @@ if __name__ == '__main__':
     app = gr.Blocks()
 
     with app:
-        with gr.Column():
-            tts_input1 = gr.TextArea(label="Text (150 words limitation)", value="こんにちは。")
-            tts_submit = gr.Button("Generate", variant="primary")
-            tts_output1 = gr.Textbox(label="Output Message")
-            tts_output2 = gr.Audio(label="Output Audio")
+        with gr.Tabs():
+            with gr.TabItem("TTS"):
+                with gr.Column():
+                    tts_input1 = gr.TextArea(label="Text (150 words limitation)", value="mutekino teioodeNsetsu, kokokara sUtaatodacl!")
+                    tts_submit = gr.Button("Generate", variant="primary")
+                    tts_output1 = gr.Textbox(label="Output Message")
+                    tts_output2 = gr.Audio(label="Output Audio")
+            with gr.TabItem("JTTS"):
+                with gr.Column():
+                    jtts_input1 = gr.TextArea(label="Text (150 words limitation)", value="無敵のテイオー伝説、ここからスタートだっ！")
+                    jtts_submit = gr.Button("Generate", variant="primary")
+                    jtts_output1 = gr.Textbox(label="Output Message")
+                    jtts_output2 = gr.Audio(label="Output Audio")
         tts_submit.click(tts_fn, tts_input1, [tts_output1, tts_output2])
-
+        jtts_submit.click(jtts_fn, jtts_input1, [jtts_output1, jtts_output2])
+    
     app.launch()