Update app.py

app.py

@@ -6,62 +6,21 @@ from neon_tts_plugin_coqui import CoquiTTS
 
 # load the model and set up constants
 LANGUAGES = list(CoquiTTS.langs.keys())
-default_lang = "en"
-title = "🐸💬 - NeonAI Coqui AI TTS Plugin"
-description = "🐸💬 - a deep learning toolkit for Text-to-Speech, battle-tested in research and production"
-info = "more info at [Neon Coqui TTS Plugin](https://github.com/NeonGeckoCom/neon-tts-plugin-coqui), [Coqui TTS](https://github.com/coqui-ai/TTS)"
-badge = "https://visitor-badge-reloaded.herokuapp.com/badge?page_id=neongeckocom.neon-tts-plugin-coqui"
 coquiTTS = CoquiTTS()
 
 # define core fn 
 def tts(text: str, language: str):
-    print(text, language)
-    # return output
     with tempfile.NamedTemporaryFile(suffix=".wav", delete=False) as fp:
         coquiTTS.get_tts(text, fp, speaker = {"language" : language})
         return fp.name
 
+# define inputs and outputs
+inputs = [gr.Textbox(label="Input", value=CoquiTTS.langs["en"]["sentence"], max_lines=3), 
+            gr.Radio(label="Language", choices=LANGUAGES, value="en")]
+outputs = gr.Audio(label="Output")
 
-# start a block 
-with gr.Blocks() as blocks:
-    # define text on page
-    gr.Markdown("<h1 style='text-align: center; margin-bottom: 1rem'>"
-                + title
-                + "</h1>")
-    gr.Markdown(description)
-    # define layout 
-    with gr.Row():
-        with gr.Column():
-            # define inputs
-            textbox = gr.Textbox(
-                label="Input",
-                value=CoquiTTS.langs[default_lang]["sentence"],
-                max_lines=3,
-            )
-            radio = gr.Radio(
-                label="Language",
-                choices=LANGUAGES,
-                value=default_lang
-            )
-            with gr.Row():
-                # define button 
-                submit = gr.Button("Submit", variant="primary")
-        # define output 
-        audio = gr.Audio(label="Output", interactive=False)
-    
-    gr.Markdown(info)
-    gr.Markdown("<center>"
-                +f'<img src={badge} alt="visitors badge"/>'
-                +"</center>")
+# define interface
+demo = gr.Interface(fn=tts, inputs=inputs, outputs=outputs)
 
-    # define what will run when submit is clicked
-    submit.click(
-        tts,
-        [textbox, radio],
-        [audio],
-    )
-    # define what will run when the radio input is changed 
-    radio.change(lambda lang: CoquiTTS.langs[lang]["sentence"], radio, textbox)
-
-# launch
-blocks.launch()
+# launch 
+demo.launch()