update

app.py

@@ -75,19 +75,15 @@ def driver_fun(audio) :
   print(f"lower_limit : upper_limit = {lower_limit} : {upper_limit}")  
   dataset_subset = dataset['Joke'][lower_limit : upper_limit]
   data = query({"inputs": {"source_sentence": "That is a happy person","sentences": dataset_subset} } )
+  if 'error' in data:
+    return 'Error in model inference - Run Again Please', 'Error in model inference - Run Again Please', None
   print(f"type(data) : {type(data)}")
   print(f"data : {data} ")
   max_match_score = max(data)
   indx_score = data.index(max_match_score)
   joke = data[indx_score]
-  #if translation 
-  #For now only taking in English text for Bloom prompting as inference model is not high spec
-  #text_generated = lang_model_response(transcribe, lang)
-  #text_generated_en = lang_model_response(translation, 'en')
+  print(f"Joke is : {joke}")
   
-  #if lang in ['es', 'fr']:
-  #  speech = tts(transcribe, lang)
-  #else:
   speech = tts(joke, 'en') #'en' # translation
   return translation, joke, speech #transcribe, 
 
@@ -123,9 +119,10 @@ def whisper_stt(audio):
 # Coqui - Text-to-Speech
 def tts(text, language):
   print(f"Inside tts - language is : {language}")
-  coqui_langs = ['en' ,'es' ,'fr' ,'de' ,'pl' ,'uk' ,'ro' ,'hu' ,'bg' ,'nl' ,'fi' ,'sl' ,'lv' ,'ga']
-  if language not in coqui_langs:
-    language = 'en'
+  #coqui_langs = ['en' ,'es' ,'fr' ,'de' ,'pl' ,'uk' ,'ro' ,'hu' ,'bg' ,'nl' ,'fi' ,'sl' ,'lv' ,'ga']
+  #if language not in coqui_langs:
+  #  language = 'en'
+  print(f"Text is : {text}")
   with tempfile.NamedTemporaryFile(suffix=".wav", delete=False) as fp:
       coquiTTS.get_tts(text, fp, speaker = {"language" : language})
       return fp.name