109 lines
3.8 KiB
Python
109 lines
3.8 KiB
Python
|
import subprocess
|
|||
|
|
import time
|
|||
|
|
import os
|
|||
|
|
os.environ['PATH'] += os.pathsep + os.path.join(os.getcwd(), "test", "ovr_lipsync", "ffmpeg", "bin")
|
|||
|
|
from pydub import AudioSegment
|
|||
|
|
import json
|
|||
|
|
|
|||
|
|
def list_files(dir_path):
|
|||
|
|
for root, dirs, files in os.walk(dir_path):
|
|||
|
|
for file in files:
|
|||
|
|
print(os.path.join(root, file))
|
|||
|
|
|
|||
|
|
class LipSyncGenerator:
|
|||
|
|
def __init__(self):
|
|||
|
|
self.viseme_em = [
|
|||
|
|
"sil", "PP", "FF", "TH", "DD",
|
|||
|
|
"kk", "CH", "SS", "nn", "RR",
|
|||
|
|
"aa", "E", "ih", "oh", "ou"]
|
|||
|
|
self.viseme = []
|
|||
|
|
self.exe_path = os.path.join(os.getcwd(), "test", "ovr_lipsync", "ovr_lipsync_exe", "ProcessWAV.exe")
|
|||
|
|
|
|||
|
|
def convert_mp3_to_wav(self, mp3_filepath):
|
|||
|
|
audio = AudioSegment.from_mp3(mp3_filepath)
|
|||
|
|
# 使用 set_frame_rate 方法设置采样率
|
|||
|
|
audio = audio.set_frame_rate(44100)
|
|||
|
|
wav_filepath = mp3_filepath.rsplit(".", 1)[0] + ".wav"
|
|||
|
|
audio.export(wav_filepath, format="wav")
|
|||
|
|
return wav_filepath
|
|||
|
|
|
|||
|
|
def run_exe_and_get_output(self, arguments):
|
|||
|
|
process = subprocess.Popen([self.exe_path] + arguments, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
|
|||
|
|
|
|||
|
|
while True:
|
|||
|
|
output = process.stdout.readline()
|
|||
|
|
if output == b'' and process.poll() is not None:
|
|||
|
|
break
|
|||
|
|
if output:
|
|||
|
|
self.viseme.append(output.strip().decode())
|
|||
|
|
rc = process.poll()
|
|||
|
|
return rc
|
|||
|
|
|
|||
|
|
def filter(self, viseme):
|
|||
|
|
new_viseme = []
|
|||
|
|
for v in self.viseme:
|
|||
|
|
if v in self.viseme_em:
|
|||
|
|
new_viseme.append(v)
|
|||
|
|
return new_viseme
|
|||
|
|
|
|||
|
|
def generate_visemes(self, mp3_filepath):
|
|||
|
|
|
|||
|
|
|
|||
|
|
wav_filepath = self.convert_mp3_to_wav(mp3_filepath)
|
|||
|
|
arguments = ["--print-viseme-name", wav_filepath]
|
|||
|
|
self.run_exe_and_get_output(arguments)
|
|||
|
|
|
|||
|
|
return self.filter(self.viseme)
|
|||
|
|
|
|||
|
|
def consolidate_visemes(self, viseme_list):
|
|||
|
|
if not viseme_list:
|
|||
|
|
return []
|
|||
|
|
|
|||
|
|
result = []
|
|||
|
|
current_viseme = viseme_list[0]
|
|||
|
|
count = 1
|
|||
|
|
|
|||
|
|
for viseme in viseme_list[1:]:
|
|||
|
|
if viseme == current_viseme:
|
|||
|
|
count += 1
|
|||
|
|
else:
|
|||
|
|
result.append({"Lip": current_viseme, "Time": count*33}) # Multiply by 10 for duration in ms
|
|||
|
|
current_viseme = viseme
|
|||
|
|
count = 1
|
|||
|
|
|
|||
|
|
# Add the last viseme to the result
|
|||
|
|
result.append({"Lip": current_viseme, "Time": count*33}) # Multiply by 10 for duration in ms
|
|||
|
|
|
|||
|
|
new_data = []
|
|||
|
|
for i in range(len(result)):
|
|||
|
|
if result[i]['Time'] < 30:
|
|||
|
|
if len(new_data) > 0:
|
|||
|
|
new_data[-1]['Time'] += result[i]['Time']
|
|||
|
|
else:
|
|||
|
|
new_data.append(result[i])
|
|||
|
|
return new_data
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
start_time = time.time()
|
|||
|
|
lip_sync_generator = LipSyncGenerator()
|
|||
|
|
viseme_list = lip_sync_generator.generate_visemes(R"E:\\github\\Fay assistant\\samples\\sample-1701224060795.mp3")
|
|||
|
|
print(viseme_list)
|
|||
|
|
consolidated_visemes = lip_sync_generator.consolidate_visemes(viseme_list)
|
|||
|
|
print(json.dumps(consolidated_visemes))
|
|||
|
|
print(time.time() - start_time)
|
|||
|
|
|
|||
|
|
audio = AudioSegment.from_file("E:\\github\\Fay assistant\\samples\\sample-1701224060795.wav")
|
|||
|
|
audio_length = len(audio) # 持续时间，单位为毫秒
|
|||
|
|
|
|||
|
|
interface_text = f"""{{
|
|||
|
|
"Topic": "Unreal",
|
|||
|
|
"Data": {{
|
|||
|
|
"Key": "audio",
|
|||
|
|
"Value": "E:\\github\\Fay assistant\\samples\\sample-1701224060795.mp3",
|
|||
|
|
"Text" : "我叫Fay,我今年18岁，很年青。"
|
|||
|
|
"Lips":{consolidated_visemes},
|
|||
|
|
"Time": {audio_length / 1000},
|
|||
|
|
"Type": "interact"
|
|||
|
|
}}
|
|||
|
|
}}"""
|
|||
|
|
with open("notepad.txt", "w") as f:
|
|||
|
|
f.write(interface_text)
|