Update main.py

main.py

@@ -30,31 +30,37 @@ def main(youtube_url):
     if not os.path.exists('./translated'):
         os.makedirs('./translated')
 
-    # Step 1: Download the video
+    # Step 1: Download video
+    ## yt_download.py
     downloaded_video_path = download_video(youtube_url)
 
-    # Step 2: Transcribe the video's audio
+    # Step 2: Transcribe video's audio
+    ## video_to_text.py
     transcribed_text_path = get_transcription_filename(downloaded_video_path)
-    model_type = 'base'  # You can specify the Whisper model type
+    model_type = 'base' # Whisper model type
     convert_video_to_text(downloaded_video_path, model_type)
 
 
-    # Step 3: Translate the transcribed text to Spanish
+    # Step 3: Translate transcribed text to Spanish
+    ## opus.py
     translated_text_path = './translated/translated_text.txt'
     translate_file(transcribed_text_path, translated_text_path)
 
     # Step 4: Perform diarization
+    ## diarization.py
     audio_path = get_audio_filename(downloaded_video_path)
     diarized_audio_dir = './audio/diarization'
     perform_diarization(audio_path, translated_text_path)
 
     # Step 5: Generate speech for translated text
+    ## tts.py
     speaker_directory = './audio/diarization'
-    aligned_text_file = './audio/diarization/aligned_text.txt'  # Ensure this is the correct path
+    aligned_text_file = './audio/diarization/aligned_text.txt'
     output_audio_file = './translated/final_audio.wav'
     tts_main(speaker_directory, aligned_text_file, output_audio_file)
 
-    # Step 6: Create the final translated video
+    # Step 6: Create final translated video
+    ## translated_video.py
     video_name = get_video_name(downloaded_video_path)
     final_video_path = create_translated_video(downloaded_video_path, output_audio_file, translated_text_path, video_name)
 
@@ -67,4 +73,4 @@ if __name__ == "__main__":
     parser.add_argument("youtube_url", help="YouTube video URL")
     args = parser.parse_args()
 
-    main(args.youtube_url)
+    main(args.youtube_url)