Update app.py

app.py

@@ -133,6 +133,18 @@ def tts(text: str, model_name: str):
     with tempfile.NamedTemporaryFile(suffix=".wav", delete=False) as fp:
         synthesizer.save_wav(wavs, fp)
         return fp.name
+        
+#TTStory
+def ttstory(text: str, model_name: str):
+    print(text, model_name)
+    gen = GENERATORS.get(model_name, None)
+    if gen is None:
+        raise NameError("model not found")
+    #wavs = gen.tts(text)
+    #with tempfile.NamedTemporaryFile(suffix=".wav", delete=False) as fp:
+    #    synthesizer.save_wav(wavs, fp)
+    #    return fp.name
+
 
 #Stories
 def Story(text: input, model_name: str):
@@ -167,7 +179,7 @@ with demo:
     b3.click(upsert, inputs=text, outputs=saved)
     b4.click(selectall, inputs=text, outputs=savedAll)
     b5.click(tts,  inputs=[text,TTSchoice], outputs=audio)
-    b6.click(tts,  inputs=[text,Storychoice], outputs=text)
+    b6.click(ttstory,  inputs=[text,Storychoice], outputs=text)
 
 # Lets Do It
 demo.launch(share=True)