Italian-Brainrot / clean.py
NeuralFalcon's picture
Upload clean.py
123cdd2 verified
raw
history blame
1.49 kB
import os
assets_dir = './assets'
character = {}
# Step 1: Load character dictionary
for i in os.listdir(assets_dir):
if i.endswith('.wav'):
audio_path = f"{assets_dir}/{i}"
image_path= f"{assets_dir}/{i.replace('.wav', '.png')}"
sound_name = i.replace('.wav', '').replace('_', ' ').title()
character[sound_name] = {
'image_path': image_path,
'audio_path': audio_path
}
# Step 2: Define custom priority order
priority_names = [
'Bombardiro Crocodilo',
'Tralalero Tralala',
'Bombombini Gusini',
'Tung Tung Tung Tung Tung Tung Tung Tung Tung Sahur',
'Brr Brr Patapim',
'Cappuccino Assassino',
'Lirili Larila',
'Trulimero Trulichina',
'Boneca Ambalabu',
'Chimpanzini Bananini' # Note: Fix spacing issue from original
]
# Normalize keys to fix any mismatch
fixed_character = {}
for name, data in character.items():
fixed_name = ' '.join(name.split()) # Removes extra spaces
fixed_character[fixed_name] = data
# Step 3: Sort with custom priority
sorted_keys = (
priority_names +
sorted([k for k in fixed_character if k not in priority_names])
)
# Step 4: Create the final sorted dictionary
sorted_character = {k: fixed_character[k] for k in sorted_keys}
import json
json_path = 'character.json'
with open(json_path, 'w', encoding='utf-8') as f:
json.dump(sorted_character, f, ensure_ascii=False, indent=4)