import os
import base64

def decode_filename(filename):
    name, ext = os.path.splitext(filename)
    try:
        decoded = base64.b64decode(name.encode('utf-8')).decode('utf-8')
        return f"{decoded}{ext}"
    except Exception as e:
        print(f"Skip: {filename} (cannot decode)")
        return None

def main():
    for fname in os.listdir('.'):
        if os.path.isfile(fname) and fname.endswith(".safetensors"):
            new_name = decode_filename(fname)
            if new_name:
                os.rename(fname, new_name)
                print(f"Restored: {fname} -> {new_name}")

if __name__ == "__main__":
    main()