So, I just wanted to batch delete old models if a new version of the model was present in the said folder, but that is not working. If anyone can solve the issue, please do so

[midnight-god-01]

import json

# Set the directory path where the .safetensors files are located
directory = r"C:\AI\stable-diffusion-webui"

# Create a list to store the information about .safetensors files
file_list = []

# Scan the directory and subdirectories for .safetensors files and collect their information
for root, dirs, files in os.walk(directory):
    for file in files:
        if file.endswith(".safetensors"):
            file_path = os.path.join(root, file)
            with open(file_path, "r") as f:
                data = json.load(f)
                file_info = {
                    "file": file_path,
                    "versions": list(data.keys())
                }
                file_list.append(file_info)

# Display the list of .safetensors files and their versions
for i, file_info in enumerate(file_list, start=1):
    print(f"{i}. {file_info['file']}")
    for version in file_info["versions"]:
        print(f"   - {version}")
    print()

# Prompt user to select a file and version to delete
file_index = int(input("Enter the number of the file to delete a version from: ")) - 1
version_to_delete = input("Enter the version to delete: ")

# Delete the specified version from the selected .safetensors file
selected_file = file_list[file_index]["file"]
with open(selected_file, "r") as f:
    data = json.load(f)
    if version_to_delete in data:
        del data[version_to_delete]
        with open(selected_file, "w") as f:
            json.dump(data, f)
        print(f"Deleted version '{version_to_delete}' from file '{selected_file}'.")
    else:
        print(f"Version '{version_to_delete}' does not exist in file '{selected_file}'.")

input("Press Enter to exit...")