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 (
-
- );
+ return (
+
+ );
case "audio":
case "unknown":