Update main.py

main.py

@@ -15,6 +15,10 @@ def get_audio_filename(video_path):
     base_name = os.path.splitext(os.path.basename(video_path))[0]
     return f'./audio/{base_name}.wav'
 
+def get_video_name(video_path):
+    base_name = os.path.splitext(os.path.basename(video_path))[0]
+    return base_name
+
 def main(youtube_url):
     # Ensure necessary directories exist
     if not os.path.exists('./downloads'):
@@ -51,7 +55,8 @@ def main(youtube_url):
     tts_main(speaker_directory, aligned_text_file, output_audio_file)
 
     # Step 6: Create the final translated video
-    final_video_path = create_translated_video(downloaded_video_path, output_audio_file, translated_text_path)
+    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)
 
     print(f"Final translated video created at {final_video_path}")