refactor(SW-337): fixes for lightbox

components/Lightbox/Gallery.tsx

@@ -123,7 +123,12 @@ export default function Gallery({
 
       {/* Mobile Gallery */}
       <div className={styles.mobileGallery}>
-        <Button intent="text" size="small" onClick={onClose}>
+        <Button
+          intent="text"
+          size="small"
+          className={styles.mobileGalleryCloseButton}
+          onClick={onClose}
+        >
           <ChevronRightIcon
             color="black"
             width={32}
@@ -137,7 +142,10 @@ export default function Gallery({
               <motion.div
                 key={image.url}
                 className={`${styles.thumbnailContainer} ${index % 3 === 0 ? styles.fullWidthImage : ""}`}
-                onClick={() => onImageClick()}
+                onClick={() => {
+                  onSelectImage(image)
+                  onImageClick()
+                }}
                 initial={{ opacity: 0, y: 20 }}
                 animate={{ opacity: 1, y: 0 }}
                 transition={{ duration: 0.3, delay: index * 0.05 }}