# env Daily usage directory/py
# -*- coding: UTF-8 -*-
'''
@Project: Daily usage directory
@File:
@IDE: PyCharm
@Author: Mr data Yang
@Date: 2024/3/27 16:24
'''
from pydub import AudioSegment
import shutil
import os
from gradio_client import Client, handle_file
# API URL
client = Client("http://localhost:9888/")
base_dir = "H:/MyScriptPublic/AIGC/GPT-SoVITS-Gradio【bak】/"
change_sovits_weights = os.path.join(base_dir, "SoVITS_weights", "MrDataYang_e8_s1400.pth")
result_sovits = client.predict(
change_sovits_weights,
api_name="/change_sovits_weights"
)
change_sovits_weights = os.path.join(base_dir, "GPT_weights", "")
result_gpt = client.predict(
"GPT_SoVITS/pretrained_models/s1bert25hz-2kh-longer-epoch=68e-step=",
api_name="/change_gpt_weights"
)
# Merge audio files
def merge_wav_files(directory, result_path):
# Get all .wav files and sort them by file name
files = sorted([f for f in os.listdir(directory) if f.endswith('.wav')], key=lambda x:int(x.split('.')[0]))
print(directory, result_path)
print(files)
# Initialize an empty audio segment
combined = AudioSegment.empty()
# Load each file in turn and append it to the combined variable
for file in files:
path = os.path.join(directory, file)
audio = AudioSegment.from_wav(path)
combined += audio
# Export the merged file
combined.export(result_path, format="wav")
def move_files_and_directories(src_dir, matching_directory):
# Make sure the target directory exists, and if it does not exist, create it
os.makedirs(matching_directory, exist_ok=True)
wav_matching_directory = os.path.join(matching_directory, 'wav')
os.makedirs(wav_matching_directory, exist_ok=True)
# traverse all files and subdirectories in the source directory
for root, dirs, files in os.walk(src_dir):
# Calculate the subdirectory path of the target path
relative_path = os.path.relpath(root, src_dir)
dest_path = os.path.join(wav_matching_directory, relative_path)
# Ensure that the subdirectory of the target path exists
os.makedirs(dest_path, exist_ok=True)
# Move files
for filename in files:
src_file = os.path.join(root, filename)
dest_file = os.path.join(dest_path, filename)
shutil.move(src_file, dest_file)
print(f"Moved: {src_file} to {dest_file}")
# After the movement is completed, delete the empty source directory
shutil.rmtree(src_dir)
print(f"Deleted empty source directory: {src_dir}")
def find_matching_directory(dest_dir, dir_name):
# Get all folder names in the target directory
for root, dirs, files in os.walk(dest_dir):
for folder in dirs:
# Check if dir_name is included in a folder name
if dir_name in folder:
matching_dir = os.path.join(root, folder)
print(f"Found matching directory: {matching_dir}")
return matching_dir # Return the matching directory path
# If no matching folder is found, return None
print(f"No matching directory found for {dir_name} in {dest_dir}.")
return None
# Specify directory path
base_dir = 'wav'
temp = 'temp_wav'
dest_dir = 'E:/Document and Video/ Programming Basics'
for subdir, dirs, files in os.walk(base_dir):
# Initialize the list of files that meet the criteria
valid_files = []
# Check whether each file meets the criteria
for file in files:
if file.endswith('.wav') and not 'txt' in file and file[0].isdigit():
valid_files.append(os.path.join(subdir, file))
# List of relative paths to all files
# ['wav\\Python Learning Notes - Explore 5 Data Types\\temp\\', 'wav\\Python Learning Notes - Explore 5 Data Types\\temp\\', 'wav\\Python Learning Notes - Explore 5 Data Types\\temp\\', 'wav\\Python Learning Notes - Explore 5 Data Types\\temp\\',
# 'wav\\Python Learning Notes - Explore 5 Data Types\\temp\\', 'wav\\Python Learning Notes - Explore 5 Data Types\\temp\\', 'wav\\Python Learning Notes - Explore 5 Data Types\\temp\\', 'wav\\Python Learning Notes - Explore 5 Data Types\\temp\\',
# 'wav\\Python Learning Notes - Explore 5 Data Types\\temp\\', 'wav\\Python Learning Notes - Explore 5 Data Types\\temp\\', 'wav\\Python Learning Notes - Explore 5 Data Types\\temp\\', 'wav\\Python Learning Notes - Explore 5 Data Types\\temp\\',
# 'wav\\Python Learning Notes - Explore 5 Data Types\\temp\\', 'wav\\Python Learning Notes - Explore 5 Data Types\\temp\\', 'wav\\Python Learning Notes - Explore 5 Data Types\\temp\\', 'wav\\Python Learning Notes - Explore 5 Data Types\\temp\\',
# 'wav\\Python Learning Notes - Explore 5 Data Types\\temp\\', 'wav\\Python Learning Notes - Explore 5 Data Types\\temp\\', 'wav\\Python Learning Notes - Explore 5 Data Types\\temp\\', 'wav\\Python Learning Notes - Explore 5 Data Types\\temp\\',
# 'wav\\Python Learning Notes - Explore 5 Data Types\\temp\\', 'wav\\Python Learning Notes - Explore 5 Data Types\\temp\\', 'wav\\Python Learning Notes - Explore 5 Data Types\\temp\\', 'wav\\Python Learning Notes - Explore 5 Data Types\\temp\\',
# 'wav\\Python Learning Notes - Explore 5 Data Types\\temp\\', 'wav\\Python Learning Notes - Explore 5 Data Types\\temp\\']
# If there are files that meet the criteria, print the path to those files
if valid_files:
menu_name = os.path.abspath(subdir).replace('\\temp', '')
dir_name = menu_name.split("\\")[-1]
# 0120-Chapter 120 Zhen Shiyin talks about Taixu's love in detail, Jia Yucun sums up the dream of the Red Chamber
print("menu_name", menu_name) # H:\MyScriptPublic\AIGC\so-vits-svc_2023-08-18\Clone API\wav\Python Learning Notes - Explore 5 Data Types
print("dir_name", dir_name) # Python Learning Notes - Explore 5 Data Types
# Clear the temp directory
shutil.rmtree(temp)
os.makedirs(temp)
# Merge audio files
result_path = os.path.join(menu_name, "clone_" + dir_name + '.wav')
print("result_path", result_path)
# Check if the file already exists
if os.path.exists(result_path):
print(f" file{result_path}Already exists, skip merge. ")
# continue # skip the remaining part of the current loop
print(f"Start the folder processing:{os.path.basename(subdir)}The path to the .wav file:")
for wav_file in valid_files:
abs_path = os.path.abspath(wav_file)
wav_name = os.path.basename(abs_path)
print('Start the file processing:', wav_name, abs_path)
# H:\Daily Use Catalog\6.Article Reading\"Dream of Red Mansions"\0118-Chapter 118 Recording Wei's Sneak My uncle bullies the weak daughter, shocking riddles, wives and concubines advise fools\
# 3-10 seconds sample file
sample_file = "H:/MyScriptPublic/AIGC/GPT-SoVITS-Gradio【bak】/logs/MrDataYang/5-wav32k/MrDataYang_1_FormatFactoryPart1.wav_0000000000_0000240960.wav"
source_path = client.predict(
handle_file(abs_path), # filepath in 'source audio' Audio component
"Hello!!", # str in 'Source audio corresponding text' Textbox component
"Chinese-English Mix", # Literal['Chinese', 'English', 'Japanese', 'Chinese-English mix', 'Japanese-English mix', 'Multi-lingual mix'] in 'Text language' Dropdown component
handle_file(sample_file), # filepath in 'Please upload the audio within 3~10 seconds, and it will be alarmed if it exceeds the price! ' Audio component
api_name="/vc_main"
)
# Create the target directory path
target_path = os.path.join(temp, wav_name)
# Copy and rename the file
shutil.copy(source_path, target_path)
# Merge file to save results
merge_wav_files(temp, result_path)
src_dir = menu_name.replace("\\", '/')
print("src_dir", src_dir)
matching_directory = find_matching_directory(dest_dir, dir_name).replace("\\", '/')
print("matching_directory", matching_directory)
dest_dir = os.path.join(dest_dir, dir_name).replace("\\", '/')
print("src_dir", src_dir)
print("dest_dir", dest_dir)
move_files_and_directories(src_dir, matching_directory)
