marquesafonso commited on
Commit
21b498c
·
1 Parent(s): 3fc2a44

enhancement: bump audio format to m4a (mp3 deprecated)

Browse files
cli.py CHANGED
@@ -2,7 +2,7 @@ from argparse import ArgumentParser
2
  from utils.download_video import download_video
3
  from utils.transcriber import transcriber
4
  from utils.subtitler import subtitler
5
- from utils.convert_mp4_to_mp3 import convert_mp4_to_mp3
6
  import logging, os
7
  from tqdm import tqdm
8
 
@@ -28,8 +28,8 @@ def main(video_url:str,
28
  if video_url != None:
29
  INVIDEO_PATH = download_video(video_url, INVIDEO_DIR, invideo_filename)
30
  pbar.update(25)
31
- INAUDIO_PATH = os.path.join(INVIDEO_DIR, f"{invideo_filename}.mp3")
32
- convert_mp4_to_mp3(INVIDEO_PATH,INAUDIO_PATH)
33
  pbar.update(25)
34
  if not os.path.exists(SRT_PATH):
35
  transcriber(INAUDIO_PATH, SRT_PATH, max_words_per_line)
@@ -38,9 +38,9 @@ def main(video_url:str,
38
  pbar.update(25)
39
  return
40
  INVIDEO_PATH = os.path.join(INVIDEO_DIR, f"{invideo_filename}.mp4")
41
- INAUDIO_PATH = os.path.join(INVIDEO_DIR, f"{invideo_filename}.mp3")
42
  if not os.path.exists(INAUDIO_PATH):
43
- convert_mp4_to_mp3(INVIDEO_PATH,INAUDIO_PATH)
44
  pbar.update(50)
45
  if not os.path.exists(SRT_PATH):
46
  transcriber(INAUDIO_PATH, SRT_PATH, max_words_per_line)
 
2
  from utils.download_video import download_video
3
  from utils.transcriber import transcriber
4
  from utils.subtitler import subtitler
5
+ from utils.convert_video_to_audio import convert_video_to_audio
6
  import logging, os
7
  from tqdm import tqdm
8
 
 
28
  if video_url != None:
29
  INVIDEO_PATH = download_video(video_url, INVIDEO_DIR, invideo_filename)
30
  pbar.update(25)
31
+ INAUDIO_PATH = os.path.join(INVIDEO_DIR, f"{invideo_filename}.m4a")
32
+ convert_video_to_audio(INVIDEO_PATH,INAUDIO_PATH)
33
  pbar.update(25)
34
  if not os.path.exists(SRT_PATH):
35
  transcriber(INAUDIO_PATH, SRT_PATH, max_words_per_line)
 
38
  pbar.update(25)
39
  return
40
  INVIDEO_PATH = os.path.join(INVIDEO_DIR, f"{invideo_filename}.mp4")
41
+ INAUDIO_PATH = os.path.join(INVIDEO_DIR, f"{invideo_filename}.m4a")
42
  if not os.path.exists(INAUDIO_PATH):
43
+ convert_video_to_audio(INVIDEO_PATH,INAUDIO_PATH)
44
  pbar.update(50)
45
  if not os.path.exists(SRT_PATH):
46
  transcriber(INAUDIO_PATH, SRT_PATH, max_words_per_line)
utils/{convert_mp4_to_mp3.py → convert_video_to_audio.py} RENAMED
@@ -1,16 +1,16 @@
1
  from moviepy.editor import VideoFileClip
2
 
3
- def convert_mp4_to_mp3(mp4_file_path, mp3_file_path):
4
  # Load the video file
5
  video_clip = VideoFileClip(mp4_file_path)
6
 
7
  # Extract the audio from the video clip
8
  audio_clip = video_clip.audio
9
 
10
- # Save the audio clip as an MP3 file
11
- audio_clip.write_audiofile(mp3_file_path)
12
 
13
  # Close the clips
14
  audio_clip.close()
15
  video_clip.close()
16
- return mp3_file_path
 
1
  from moviepy.editor import VideoFileClip
2
 
3
+ def convert_video_to_audio(mp4_file_path, m4a_file_path):
4
  # Load the video file
5
  video_clip = VideoFileClip(mp4_file_path)
6
 
7
  # Extract the audio from the video clip
8
  audio_clip = video_clip.audio
9
 
10
+ # Save the audio clip as an m4a file
11
+ audio_clip.write_audiofile(m4a_file_path, codec='aac')
12
 
13
  # Close the clips
14
  audio_clip.close()
15
  video_clip.close()
16
+ return m4a_file_path
utils/process_video.py CHANGED
@@ -26,7 +26,7 @@ def process_video(invideo_filename:str,
26
  subtitler(invideo_filename, srt_path, OUTVIDEO_PATH, fontsize, font, bg_color, text_color)
27
  return OUTVIDEO_PATH
28
  logging.info("Converting Video to Audio")
29
- INAUDIO_PATH = os.path.abspath(f"{invideo_filename.split('.')[0]}.mp3")
30
  if not os.path.exists(INAUDIO_PATH):
31
  convert_mp4_to_mp3(invideo_filename, INAUDIO_PATH)
32
  SRT_PATH = os.path.abspath(f"{invideo_filename.split('.')[0]}.srt")
 
26
  subtitler(invideo_filename, srt_path, OUTVIDEO_PATH, fontsize, font, bg_color, text_color)
27
  return OUTVIDEO_PATH
28
  logging.info("Converting Video to Audio")
29
+ INAUDIO_PATH = os.path.abspath(f"{invideo_filename.split('.')[0]}.m4a")
30
  if not os.path.exists(INAUDIO_PATH):
31
  convert_mp4_to_mp3(invideo_filename, INAUDIO_PATH)
32
  SRT_PATH = os.path.abspath(f"{invideo_filename.split('.')[0]}.srt")