import os
import base64


def encode_filename(filename):
    name, ext = os.path.splitext(filename)
    encoded = base64.b64encode(name.encode('utf-8')).decode('utf-8')
    return f"{encoded}{ext}"

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

if __name__ == "__main__":
    main()