Update Milestone5API.py

Milestone5API.py

@@ -19,27 +19,21 @@ def replace_video_audio(video_file_path, new_audio_file_path, output_video_file_
     video = video.set_audio(new_audio)
 
     video.write_videofile(output_video_file_path, codec="libx264", audio_codec="aac", temp_audiofile='temp_audio.mp3', remove_temp=True)
-def text_to_speech(translated_text, video_path):
+def text_to_speech(translated_text_file_path, video_path):
     device = "cuda" if torch.cuda.is_available() else "cpu"
     tts = TTS("tts_models/fr/mai/tacotron2-DDC").to(device)
-    text_files_dir = f'CS370_Milestone5/captions/{translated_text}'
     output_audio_dir = 'CS370_Milestone5/audio'
     os.makedirs(output_audio_dir, exist_ok=True)
-    
-    for file_name in os.listdir(text_files_dir):
-        if file_name.endswith('.txt'):
-            file_path = os.path.join(text_files_dir, file_name)
-            file_name_only = os.path.splitext(file_name)[0]
-            
-            with open(file_path, 'r', encoding='utf-8') as file:
-                translated_text = file.read()
-                
-            output_file_path = os.path.join(output_audio_dir, f"{file_name_only}.wav")
-            
-            
-            tts.tts_to_file(text=translated_text, file_path=output_file_path)
 
-    output_video_path = f"CS370_Milestone5/videos/{translated_text.split('.')[0]}_new.mp4"
+    file_name_only = os.path.splitext(os.path.basename(translated_text_file_path))[0]
+    output_file_path = os.path.join(output_audio_dir, f"{file_name_only}.wav")
+
+    with open(translated_text_file_path, 'r', encoding='utf-8') as file:
+        translated_text = file.read()
+
+    tts.tts_to_file(text=translated_text, file_path=output_file_path)
+
+    output_video_path = f"CS370_Milestone5/videos/{file_name_only}_new.mp4"
 
     replace_video_audio(video_path, output_file_path, output_video_path)
     return output_video_path