Spaces:
Runtime error
Runtime error
| import os | |
| import sys | |
| import time | |
| import shutil | |
| import requests | |
| import zipfile | |
| def insert_new_line(file_name, line_to_find, text_to_insert): | |
| lines = [] | |
| with open(file_name, 'r', encoding='utf-8') as read_obj: | |
| lines = read_obj.readlines() | |
| already_exists = False | |
| with open(file_name + '.tmp', 'w', encoding='utf-8') as write_obj: | |
| for i in range(len(lines)): | |
| write_obj.write(lines[i]) | |
| if lines[i].strip() == line_to_find: | |
| # If next line exists and starts with sys.path.append, skip | |
| if i+1 < len(lines) and lines[i+1].strip().startswith("sys.path.append"): | |
| print('It was already fixed! Skip adding a line...') | |
| already_exists = True | |
| break | |
| else: | |
| write_obj.write(text_to_insert + '\n') | |
| # If no existing sys.path.append line was found, replace the original file | |
| if not already_exists: | |
| os.replace(file_name + '.tmp', file_name) | |
| return True | |
| else: | |
| # If existing line was found, delete temporary file | |
| os.remove(file_name + '.tmp') | |
| return False | |
| def replace_in_file(file_name, old_text, new_text): | |
| with open(file_name, 'r', encoding='utf-8') as file: | |
| file_contents = file.read() | |
| if old_text in file_contents: | |
| file_contents = file_contents.replace(old_text, new_text) | |
| with open(file_name, 'w', encoding='utf-8') as file: | |
| file.write(file_contents) | |
| return True | |
| return False | |
| if __name__ == "__main__": | |
| current_path = os.getcwd() | |
| file_name = os.path.join(current_path, "infer", "modules", "train", "extract", "extract_f0_print.py") | |
| line_to_find = 'import numpy as np, logging' | |
| text_to_insert = "sys.path.append(r'" + current_path + "')" | |
| success_1 = insert_new_line(file_name, line_to_find, text_to_insert) | |
| if success_1: | |
| print('The first operation was successful!') | |
| else: | |
| print('He skipped the first operation because it was already fixed!') | |
| file_name = 'infer-web.py' | |
| old_text = 'with gr.Blocks(theme=gr.themes.Soft()) as app:' | |
| new_text = 'with gr.Blocks() as app:' | |
| success_2 = replace_in_file(file_name, old_text, new_text) | |
| if success_2: | |
| print('The second operation was successful!') | |
| else: | |
| print('The second operation was omitted because it was already fixed!') | |
| print('Local corrections successful! You should now be able to infer and train locally in Applio RVC Fork.') | |
| time.sleep(5) | |
| def find_torchcrepe_directory(directory): | |
| """ | |
| Recursively searches for the topmost folder named 'torchcrepe' within a directory. | |
| Returns the path of the directory found or None if none is found. | |
| """ | |
| for root, dirs, files in os.walk(directory): | |
| if 'torchcrepe' in dirs: | |
| return os.path.join(root, 'torchcrepe') | |
| return None | |
| def download_and_extract_torchcrepe(): | |
| url = 'https://github.com/maxrmorrison/torchcrepe/archive/refs/heads/master.zip' | |
| temp_dir = 'temp_torchcrepe' | |
| destination_dir = os.getcwd() | |
| try: | |
| torchcrepe_dir_path = os.path.join(destination_dir, 'torchcrepe') | |
| if os.path.exists(torchcrepe_dir_path): | |
| print("Skipping the torchcrepe download. The folder already exists.") | |
| return | |
| # Download the file | |
| print("Starting torchcrepe download...") | |
| response = requests.get(url) | |
| # Raise an error if the GET request was unsuccessful | |
| response.raise_for_status() | |
| print("Download completed.") | |
| # Save the downloaded file | |
| zip_file_path = os.path.join(temp_dir, 'master.zip') | |
| os.makedirs(temp_dir, exist_ok=True) | |
| with open(zip_file_path, 'wb') as file: | |
| file.write(response.content) | |
| print(f"Zip file saved to {zip_file_path}") | |
| # Extract the zip file | |
| print("Extracting content...") | |
| with zipfile.ZipFile(zip_file_path, 'r') as zip_file: | |
| zip_file.extractall(temp_dir) | |
| print("Extraction completed.") | |
| # Locate the torchcrepe folder and move it to the destination directory | |
| torchcrepe_dir = find_torchcrepe_directory(temp_dir) | |
| if torchcrepe_dir: | |
| shutil.move(torchcrepe_dir, destination_dir) | |
| print(f"Moved the torchcrepe directory to {destination_dir}!") | |
| else: | |
| print("The torchcrepe directory could not be located.") | |
| except Exception as e: | |
| print("Torchcrepe not successfully downloaded", e) | |
| # Clean up temporary directory | |
| if os.path.exists(temp_dir): | |
| shutil.rmtree(temp_dir) | |
| # Run the function | |
| download_and_extract_torchcrepe() | |
| temp_dir = 'temp_torchcrepe' | |
| if os.path.exists(temp_dir): | |
| shutil.rmtree(temp_dir) | |