from moviepy.editor import *
import sys
# print('argv[1]:' + str(sys.argv[1]))
import os
from pydub import AudioSegment


def mp4_to_mp3(mp4_uri):
	# Load the mp4 file
	video = VideoFileClip(mp4_uri)

	# Extract audio from video
	video.audio.write_audiofile(mp4_uri + ".mp3")


def audio_to_wav(wav_uri, out_sample_rate=16000):
    src = wav_uri
    dst = wav_uri + ".wav"
    if os.path.exists(dst): os.remove(dst)

    # convert wav to mp3
    sound = AudioSegment.from_file(src)
    sound = sound.set_frame_rate(out_sample_rate)
    sound.export(dst, format="wav")

    print(' audio_to_wav: ', str(dst))
    return dst


def wav_to_mp3(wav_uri):
	AudioSegment.from_wav(wav_uri).export(wav_uri + '.mp3', format='mp3')


# sys.argv
if len(sys.argv) > 1:
	src_uri = sys.argv[1]
	if src_uri.endswith('.mp4'):
		res = mp4_to_mp3(src_uri)
	else:
		res = wav_to_mp3(src_uri)
else:
	print(' usage: python mp4_to_mp3.py /path/to/mp4_file ')