.back-button {
    margin: 10px;
    padding: 8px 12px;
    background: #333;
    color: white;
    border-radius: 8px;
    cursor: pointer;
}

.albums-wrapper {
    grid-template-columns: repeat(auto-fit, minmax(150px, 1fr));
}

.albums-wrapper,
.gallery {
    display: grid;
    gap: 10px;
    padding: 15px;
}

.album {
    position: relative;
    cursor: pointer;
    border-radius: 12px;
    overflow: hidden;
    box-shadow: 0 4px 8px rgba(0, 0, 0, 0.1);
}

.album img {
    width: 100%;
    height: 100%;
    object-fit: cover;
}

.album span {
    position: absolute;
    bottom: 8px;
    left: 8px;
    background: rgba(0, 0, 0, 0.6);
    color: white;
    padding: 4px 8px;
    border-radius: 6px;
    font-size: 14px;
}

.gallery {
    column-count: 3;
    column-gap: 10px;
    padding: 20px;
}

.gallery img {
    width: 100%;
    margin-bottom: 10px;
    display: block;
    break-inside: avoid;
}

.gallery {
    grid-template-columns: repeat(auto-fill, minmax(120px, 1fr));
}

.gallery img {
    width: 100%;
    height: 100%;
    border-radius: 8px;
    object-fit: cover;
}

@media (max-width: 900px) {
    .gallery {
        column-count: 2;
    }
}

@media (max-width: 600px) {
    .gallery {
        column-count: 1;
    }
}