fix: Minor UI tweaks and fixing addition of first bookmark

2 files changed, 4 insertions, 6 deletions

diff --git a/packages/web/app/dashboard/bookmarks/components/Bookmarks.tsx b/packages/web/app/dashboard/bookmarks/components/Bookmarks.tsx
index 44495f1f..d704b9d6 100644
--- a/packages/web/app/dashboard/bookmarks/components/Bookmarks.tsx
+++ b/packages/web/app/dashboard/bookmarks/components/Bookmarks.tsx
@@ -21,16 +21,11 @@ export default async function Bookmarks({
 
   const bookmarks = await api.bookmarks.getBookmarks(query);
 
-  // TODO: This needs to be polished
   return (
     <>
       <div className="container pb-4 text-2xl">{title}</div>
       <div className="container">
-        {bookmarks.bookmarks.length == 0 ? (
-          "No bookmarks"
-        ) : (
-          <BookmarksGrid query={query} bookmarks={bookmarks.bookmarks} />
-        )}
+        <BookmarksGrid query={query} bookmarks={bookmarks.bookmarks} />
       </div>
     </>
   );
diff --git a/packages/web/app/dashboard/bookmarks/components/BookmarksGrid.tsx b/packages/web/app/dashboard/bookmarks/components/BookmarksGrid.tsx
index dc98472e..e07d48b6 100644
--- a/packages/web/app/dashboard/bookmarks/components/BookmarksGrid.tsx
+++ b/packages/web/app/dashboard/bookmarks/components/BookmarksGrid.tsx
@@ -21,6 +21,9 @@ export default function BookmarksGrid({
   const { data } = api.bookmarks.getBookmarks.useQuery(query, {
     initialData: { bookmarks: initialBookmarks },
   });
+  if (data.bookmarks.length == 0) {
+    return <p>No bookmarks</p>;
+  }
   return (
     <div className="container grid grid-cols-1 gap-4 sm:grid-cols-1 md:grid-cols-2 lg:grid-cols-3">
       {data.bookmarks.map((b) => renderBookmark(b))}