Upload app.py

app.py

@@ -10,6 +10,19 @@ import sys
 import os
 import subprocess
 from pydub import AudioSegment
+import yaml
+import wave
+
+
+
+def get_wav_duration(file_path):
+    with wave.open(file_path, 'rb') as wav_file:
+        frames = wav_file.getnframes()
+        rate = wav_file.getframerate()
+        duration = frames / float(rate)
+        return duration
+
+
 
 def install_fairseq():
     try:
@@ -44,20 +57,29 @@ def run_my_code(input_text, language):
     audio=convert_audio_to_16k_wav(input_text)
     hi_wav = audio
 
+
     data_root=""
     model_checkpoint=""
     d_r=""
+    yam=""
 
     if(language=="Hindi"):
         model_checkpoint = "./models/hindi_model.pt"
         data_root="./MUSTC_ROOT_hindi/en-hi/"
         d_r="MUSTC_ROOT_hindi/"
+        yam="./MUSTC_ROOT_hindi/en-hi/data/tst-COMMON/txt/tst-COMMON.yaml"
     if(language=="French"):
         model_checkpoint = "./models/french_model.pt"
         data_root="./MUSTC_ROOT_french/en-fr/"
         d_r="MUSTC_ROOT_french/"
-
-
+        yam="./MUSTC_ROOT_french/en-fr/data/tst-COMMON/txt/tst-COMMON.yaml"
+
+    #code to change the duration of the yaml file accordign to the audio input
+    with open(yam, 'r') as yaml_file:
+        data = yaml.safe_load(yaml_file)
+    data[0]['duration']=get_wav_duration(hi_wav)
+    with open(yam, 'w') as yaml_file:
+        yaml.dump(data, yaml_file)
 
     os.system(f"cp {hi_wav} {data_root}data/tst-COMMON/wav/test.wav")