feat: new chat page (#31)

<img width="1904" alt="image"
src="https://github.com/landing-ai/vision-agent-ui/assets/5669963/45268e97-758d-4c4c-8c05-84f71f7a2fbd">

app/chat/page.tsx

@@ -7,13 +7,20 @@ import { fetcher } from '@/lib/utils';
 import Image from 'next/image';
 import { useRouter } from 'next/navigation';
 
-type Example = {
-  url: string;
-  initMessages: MessageBase[];
-};
+import {
+  Tooltip,
+  TooltipContent,
+  TooltipTrigger,
+} from '@/components/ui/Tooltip';
+import { IconDiscord, IconGitHub } from '@/components/ui/Icons';
+import Link from 'next/link';
+import { Button } from '@/components/ui/Button';
+import Img from '@/components/ui/Img';
 
-const examples: Example[] = [
+const exampleMessages = [
   {
+    heading: 'Counting',
+    subheading: 'number of cereals in an image',
     url: 'https://landing-lens-support.s3.us-east-2.amazonaws.com/vision-agent-examples/cereal-example.jpg',
     initMessages: [
       {
@@ -28,52 +35,83 @@ const examples: Example[] = [
       },
     ],
   },
-  // 'https://landing-lens-support.s3.us-east-2.amazonaws.com/vision-agent-examples/people-example.jpeg',
-  // 'https://landing-lens-support.s3.us-east-2.amazonaws.com/vision-agent-examples/house-exmaple.jpg',
-  // 'https://landing-lens-support.s3.us-east-2.amazonaws.com/vision-agent-examples/safari-example.png',
+  // {
+  //   heading: 'Detecting',
+  //   url: 'https://landing-lens-support.s3.us-east-2.amazonaws.com/vision-agent-examples/cereal-example.jpg',
+  //   subheading: 'number of cereals in an image',
+  //   message: `How many cereals are there in the image?`,
+  // },
 ];
 
 export default function Page() {
   const router = useRouter();
   return (
     <div className="mx-auto max-w-2xl px-4 mt-8">
-      <div className="rounded-lg border bg-background p-8">
+      <div className="rounded-lg border bg-background p-8 mb-6">
         <h1 className="mb-2 text-lg font-semibold">Welcome to Vision Agent</h1>
         <p>
           Vision Agent is a library that helps you utilize agent frameworks for
           your vision tasks. Vision Agent aims to provide an in-seconds
           experience by allowing users to describe their problem in text and
-          utilizing agent frameworks to solve the task for them. Check out our
-          discord for updates and roadmap!
+          utilizing agent frameworks to solve the task for them.
         </p>
-        <ImageSelector />
-        <p className="mt-4 mb-2">
-          You can also choose from below examples we provided
-        </p>
-        <div className="flex">
-          {examples.map(({ url, initMessages }, index) => (
-            <Image
-              src={url}
-              key={index}
-              width={120}
-              height={120}
-              alt="example images"
-              className="object-cover rounded mr-3 shadow-md hover:scale-105 cursor-pointer transition-transform"
-              onClick={async () => {
-                const resp = await fetcher<ChatEntity>('/api/upload', {
-                  method: 'POST',
-                  headers: {
-                    'Content-Type': 'application/json',
-                  },
-                  body: JSON.stringify({ url, initMessages }),
-                });
-                if (resp) {
-                  router.push(`/chat/${resp.id}`);
-                }
-              }}
-            />
-          ))}
+        <div className="my-2">
+          <Tooltip>
+            <TooltipTrigger asChild>
+              <Button variant="link" size="icon" asChild className="mr-2">
+                <Link
+                  href="https://github.com/landing-ai/vision-agent"
+                  target="_blank"
+                >
+                  <IconGitHub className="size-6" />
+                </Link>
+              </Button>
+            </TooltipTrigger>
+            <TooltipContent>Github</TooltipContent>
+          </Tooltip>
+          <Tooltip>
+            <TooltipTrigger asChild>
+              <Button variant="link" size="icon" asChild className="mr-2">
+                <Link href="https://discord.gg/wZ2A7J69" target="_blank">
+                  <IconDiscord className="size-6" />
+                </Link>
+              </Button>
+            </TooltipTrigger>
+            <TooltipContent>Discord</TooltipContent>
+          </Tooltip>
         </div>
+        <ImageSelector />
+      </div>
+      <div className="mb-4 grid grid-cols-2 gap-2 px-4 sm:px-0">
+        {exampleMessages.map((example, index) => (
+          <div
+            key={index}
+            className={`cursor-pointer rounded-lg border bg-white p-4 hover:bg-zinc-50 dark:bg-zinc-950 dark:hover:bg-zinc-900 flex items-center size-full ${
+              index > 1 && 'hidden md:block'
+            }`}
+            onClick={async () => {
+              const resp = await fetcher<ChatEntity>('/api/upload', {
+                method: 'POST',
+                headers: {
+                  'Content-Type': 'application/json',
+                },
+                body: JSON.stringify({
+                  url: example.url,
+                  initMessages: example.initMessages,
+                }),
+              });
+              if (resp) {
+                router.push(`/chat/${resp.id}`);
+              }
+            }}
+          >
+            <Img src={example.url} alt="example images" className="w-1/4" />
+            <div className="flex items-start flex-col h-full ml-3 w-3/4">
+              <div className="text-sm font-semibold">{example.heading}</div>
+              <div className="text-sm text-zinc-600">{example.subheading}</div>
+            </div>
+          </div>
+        ))}
       </div>
     </div>
   );

components/chat-sidebar/ChatCard.tsx

@@ -48,7 +48,7 @@ const ChatCard: React.FC<ChatCardProps> = ({ chat }) => {
       classNames={chatIdFromParam === id && 'border-gray-500'}
     >
       <div className="overflow-hidden flex items-center size-full">
-        <Img src={url} alt={`chat-${id}-card-image`} className="w-1/4 " />
+        <Img src={url} alt={`chat-${id}-card-image`} className="w-1/4" />
         <div className="flex items-start flex-col h-full ml-3 w-3/4">
           <p className="text-sm mb-1">{title}</p>
           <p className="text-xs text-gray-500">

components/chat/ImageSelector.tsx

@@ -8,7 +8,7 @@ import { useRouter } from 'next/navigation';
 import Loading from '../ui/Loading';
 import toast from 'react-hot-toast';
 
-export interface ImageSelectorProps { }
+export interface ImageSelectorProps {}
 
 type Example = {
   url: string;
@@ -29,23 +29,24 @@ const ImageSelector: React.FC<ImageSelectorProps> = () => {
       const reader = new FileReader();
       reader.readAsDataURL(files[0]);
       reader.onload = async () => {
-        const { id, signedUrl, publicUrl, fields } = await fetcher<SignedPayload>('/api/sign', {
-          method: 'POST',
-          body: JSON.stringify({
-            fileType: files[0].type,
-            fileName: files[0].name,
-          }),
-        });
+        const { id, signedUrl, publicUrl, fields } =
+          await fetcher<SignedPayload>('/api/sign', {
+            method: 'POST',
+            body: JSON.stringify({
+              fileType: files[0].type,
+              fileName: files[0].name,
+            }),
+          });
         const formData = new FormData();
         Object.entries(fields).forEach(([key, value]) => {
-          formData.append(key, value as string)
-        })
+          formData.append(key, value as string);
+        });
         formData.append('file', files[0]);
 
         const uploadResponse = await fetch(signedUrl, {
           method: 'POST',
           body: formData,
-        })
+        });
         if (!uploadResponse.ok) {
           toast.error(uploadResponse.statusText);
           return;
@@ -64,8 +65,9 @@ const ImageSelector: React.FC<ImageSelectorProps> = () => {
         if (resp) {
           router.push(`/chat/${resp.id}`);
         }
-      }
-    });
+      };
+    },
+  );
   return (
     <div
       {...getRootProps()}
@@ -75,11 +77,11 @@ const ImageSelector: React.FC<ImageSelectorProps> = () => {
       )}
     >
       <input {...getInputProps()} />
-      <div className="text-gray-400 text-lg">
+      <div className="text-gray-400 text-md">
         {isUploading ? (
           <Loading />
         ) : (
-          'Drag or drop image here, or click to select images'
+          'Start using Vision Agent by selecting an image'
         )}
       </div>
     </div>

components/ui/Icons.tsx

@@ -383,6 +383,29 @@ function IconCopy({ className, ...props }: React.ComponentProps<'svg'>) {
   );
 }
 
+function IconDiscord({ className, ...props }: React.ComponentProps<'svg'>) {
+  return (
+    <svg
+      width="800px"
+      height="800px"
+      viewBox="0 -28.5 256 256"
+      version="1.1"
+      xmlns="http://www.w3.org/2000/svg"
+      preserveAspectRatio="xMidYMid"
+      className={cn('size-4', className)}
+      {...props}
+    >
+      <g>
+        <path
+          d="M216.856339,16.5966031 C200.285002,8.84328665 182.566144,3.2084988 164.041564,0 C161.766523,4.11318106 159.108624,9.64549908 157.276099,14.0464379 C137.583995,11.0849896 118.072967,11.0849896 98.7430163,14.0464379 C96.9108417,9.64549908 94.1925838,4.11318106 91.8971895,0 C73.3526068,3.2084988 55.6133949,8.86399117 39.0420583,16.6376612 C5.61752293,67.146514 -3.4433191,116.400813 1.08711069,164.955721 C23.2560196,181.510915 44.7403634,191.567697 65.8621325,198.148576 C71.0772151,190.971126 75.7283628,183.341335 79.7352139,175.300261 C72.104019,172.400575 64.7949724,168.822202 57.8887866,164.667963 C59.7209612,163.310589 61.5131304,161.891452 63.2445898,160.431257 C105.36741,180.133187 151.134928,180.133187 192.754523,160.431257 C194.506336,161.891452 196.298154,163.310589 198.110326,164.667963 C191.183787,168.842556 183.854737,172.420929 176.223542,175.320965 C180.230393,183.341335 184.861538,190.991831 190.096624,198.16893 C211.238746,191.588051 232.743023,181.531619 254.911949,164.955721 C260.227747,108.668201 245.831087,59.8662432 216.856339,16.5966031 Z M85.4738752,135.09489 C72.8290281,135.09489 62.4592217,123.290155 62.4592217,108.914901 C62.4592217,94.5396472 72.607595,82.7145587 85.4738752,82.7145587 C98.3405064,82.7145587 108.709962,94.5189427 108.488529,108.914901 C108.508531,123.290155 98.3405064,135.09489 85.4738752,135.09489 Z M170.525237,135.09489 C157.88039,135.09489 147.510584,123.290155 147.510584,108.914901 C147.510584,94.5396472 157.658606,82.7145587 170.525237,82.7145587 C183.391518,82.7145587 193.761324,94.5189427 193.539891,108.914901 C193.539891,123.290155 183.391518,135.09489 170.525237,135.09489 Z"
+          fill="#5865F2"
+          fillRule="nonzero"
+        ></path>
+      </g>
+    </svg>
+  );
+}
+
 function IconCheck({ className, ...props }: React.ComponentProps<'svg'>) {
   return (
     <svg
@@ -558,4 +581,5 @@ export {
   IconChevronUpDown,
   IconGoogle,
   IconLoading,
+  IconDiscord,
 };