diff --git a/src/timelines/MediaAttachmentGrid.tsx b/src/timelines/MediaAttachmentGrid.tsx index a88d7a6..5748f0d 100644 --- a/src/timelines/MediaAttachmentGrid.tsx +++ b/src/timelines/MediaAttachmentGrid.tsx @@ -51,9 +51,19 @@ const MediaAttachmentGrid: Component<{ setViewerIndex(index); }; + const columnCount = () => { + if (props.attachments.length === 1) { + return 1; + } else if (props.attachments.length % 2 === 0) { + return 2; + } else { + return 3; + } + }; + css` .attachments { - column-count: ${(props.attachments.length === 1 ? 1 : 3).toString()}; + column-count: ${columnCount.toString()}; } `; return ( @@ -104,17 +114,17 @@ const MediaAttachmentGrid: Component<{ /> ); case "gifv": // Later we can handle the preview - return ( -