website refactor

apps/website/components/media/MediaGrid.tsx

@@ -1,16 +1,16 @@
-import { Grid } from '@/ui/primitives/Grid';
+import { Grid } from '@/ui/Grid';
 import React from 'react';
 
 export interface MediaGridProps {
   children: React.ReactNode;
-  columns?: {
+  columns?: number | {
     base?: number;
     sm?: number;
     md?: number;
     lg?: number;
     xl?: number;
   };
-  gap?: 0 | 1 | 2 | 3 | 4 | 6 | 8 | 12 | 16;
+  gap?: number;
 }
 
 export function MediaGrid({
@@ -20,10 +20,8 @@ export function MediaGrid({
 }: MediaGridProps) {
   return (
     <Grid
-      cols={(columns.base ?? 1) as any}
-      mdCols={columns.md as any}
-      lgCols={columns.lg as any}
-      gap={gap as any}
+      cols={columns}
+      gap={gap}
     >
       {children}
     </Grid>