@plteam/chat-ui/dist/views/message/attachments/useGalleryItemSize.js

CUI Kit is a free and open-source library for creating AI assistant chat interfaces, built with React, Material UI, and TypeScript

import React from 'react';
const useGalleryItemSize = (itemsCount) => {
    const size = React.useMemo(() => {
        if (itemsCount === 1)
            return 256;
        if (itemsCount >= 2 && itemsCount <= 6)
            return 128;
        return 80;
    }, [itemsCount]);
    return size;
};
export default useGalleryItemSize;