How can I maintain 1:1 aspect ratio of my image container responsively on any screen?

Question

I am designing a gallery, which shows a list of all the images on the page. Now I am keeping images inside a container such that all the images maintain a specific size on the screen. So the CSS for the same is

.gallery-image{
    border-radius: 2rem !important;
    overflow: hidden;
    position: relative;
    background-color: black;
    width: 100%;
    height: 100%;
    z-index: 99;
  }
  
  .image-box{
    opacity: 0.6 !important;
    height: 150px;
    width: 100%;
    object-fit: cover;
    overflow: hidden;
    transition:all 0.3s ease-in-out !important;
  }
  .image-box:hover{
    opacity: 1 !important;
    transform: scale(1.2);
    overflow: hidden;
  }


 
 <div class="col-md-2 col-sm-4 col-4 my-2 align-self-center">
          <div class='gallery-image'>
               <img src={baseUrl + image.path} class="image-box" />
          </div>
 </div>

gallery-image is my image container and the image-box is the style for my CSS. Where can I add a property such that my image container looks 1:1 (perfect square) in all the screens?

Here I am attaching images for the reference

ViewPort on Lappy

Viewport on Mobile