UI Updates

Updated some pages pages to be more consistent.

src/features/staff/components/add-stamp-button.tsx

@@ -6,14 +6,19 @@ import { useState, useTransition } from "react";
 import { Button } from "@/components/ui/button";
 import { Modal } from "@/components/ui/modal";
 import { Input } from "@/components/ui/input";
+import { cn } from "@/lib/utils/cn";
 import { addLoyaltyStampAction } from "../actions/loyalty-actions";
 
 export function AddStampButton({
   customerId,
   appointmentId,
+  className,
+  size = "sm",
 }: {
   customerId: string;
   appointmentId?: string;
+  className?: string;
+  size?: "sm" | "md" | "lg";
 }) {
   const router = useRouter();
   const [open, setOpen] = useState(false);
@@ -57,7 +62,12 @@ export function AddStampButton({
 
   return (
     <>
-      <Button variant="secondary" size="sm" onClick={() => setOpen(true)}>
+      <Button
+        variant="secondary"
+        size={size}
+        className={cn("shrink-0", className)}
+        onClick={() => setOpen(true)}
+      >
         <Stamp className="h-4 w-4" aria-hidden="true" />
         Add Loyalty Stamp
       </Button>