From 46f855331110b4a7c850c339c35e17eb9bbffb7b Mon Sep 17 00:00:00 2001
From: Anish Sarkar <104695310+AnishSarkar22@users.noreply.github.com>
Date: Mon, 30 Mar 2026 02:22:35 +0530
Subject: [PATCH] feat: update MobileCitationDrawer layout and enhance
 DrawerHeader for improved UI presentation

---
 .../components/assistant-ui/assistant-message.tsx    | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/surfsense_web/components/assistant-ui/assistant-message.tsx b/surfsense_web/components/assistant-ui/assistant-message.tsx
index 1418cde98..62407bbf7 100644
--- a/surfsense_web/components/assistant-ui/assistant-message.tsx
+++ b/surfsense_web/components/assistant-ui/assistant-message.tsx
@@ -77,7 +77,7 @@ import {
 } from "@/components/tool-ui/shared/media";
 import { RecallMemoryToolUI, SaveMemoryToolUI } from "@/components/tool-ui/user-memory";
 import { GenerateVideoPresentationToolUI } from "@/components/tool-ui/video-presentation";
-import { Drawer, DrawerContent, DrawerHandle, DrawerTitle } from "@/components/ui/drawer";
+import { Drawer, DrawerContent, DrawerHandle, DrawerHeader, DrawerTitle } from "@/components/ui/drawer";
 import { useComments } from "@/hooks/use-comments";
 import { useMediaQuery } from "@/hooks/use-media-query";
 import { cn } from "@/lib/utils";
@@ -180,12 +180,12 @@ const MobileCitationDrawer: FC = () => {
 			</button>
 
 			<Drawer open={open} onOpenChange={setOpen}>
-				<DrawerContent className="max-h-[70dvh]">
+				<DrawerContent className="max-h-[85vh] flex flex-col">
 					<DrawerHandle />
-					<div className="flex items-center px-4 py-2">
-						<DrawerTitle className="text-sm font-medium">Sources</DrawerTitle>
-					</div>
-					<div className="overflow-y-auto px-1 pb-6">
+					<DrawerHeader className="text-left">
+						<DrawerTitle className="text-base font-semibold">Sources</DrawerTitle>
+					</DrawerHeader>
+					<div className="overflow-y-auto flex-1 min-h-0 px-1 pb-6">
 						{citations.map((citation) => (
 							<button
 								key={citation.id}