From 83217a3905f0c9782c4e68c6947eb89cfd4cdc74 Mon Sep 17 00:00:00 2001
From: Sam Becker <sam@sambecker.com>
Date: Tue, 19 Mar 2024 23:47:07 -0500
Subject: [PATCH] Add descriptions of different lengths

---
 src/photo/ai/index.ts        | 12 ++++++++++--
 src/photo/form/PhotoForm.tsx | 36 +++++++++++++++++++++++++++++-------
 2 files changed, 39 insertions(+), 9 deletions(-)

diff --git a/src/photo/ai/index.ts b/src/photo/ai/index.ts
index b6fa7d07..a2ea7b8e 100644
--- a/src/photo/ai/index.ts
+++ b/src/photo/ai/index.ts
@@ -1,12 +1,20 @@
 import { streamOpenAiImageQuery } from '@/services/openai';
 
-export type ImageQuery = 'title' | 'caption' | 'tags' | 'description';
+export type ImageQuery =
+  'title' |
+  'caption' |
+  'tags' |
+  'descriptionSmall' |
+  'descriptionMedium' |
+  'descriptionLarge';
 
 export const IMAGE_QUERIES: Record<ImageQuery, string> = {
   title: 'Provide a short title for this image',
   caption: 'What is the caption of this image?',
   tags: 'Describe this image three or less comma-separated keywords',
-  description: 'Describe this image in detail',
+  descriptionSmall: 'Describe this image succinctly',
+  descriptionMedium: 'Describe this image',
+  descriptionLarge: 'Describe this image in detail',
 };
 
 export const streamImageQuery = (imageBase64: string, query: ImageQuery) =>
diff --git a/src/photo/form/PhotoForm.tsx b/src/photo/form/PhotoForm.tsx
index 6b5b9ba5..5faac6b7 100644
--- a/src/photo/form/PhotoForm.tsx
+++ b/src/photo/form/PhotoForm.tsx
@@ -131,10 +131,16 @@ export default function PhotoForm({
   ] = useImageQuery(imageData, 'tags');
 
   const [
-    requestDescription,
-    description,
-    isLoadingDescription,
-  ] = useImageQuery(imageData, 'description');
+    requestDescriptionSmall,
+    descriptionSmall,
+    isLoadingDescriptionSmall,
+  ] = useImageQuery(imageData, 'descriptionSmall');
+
+  const [
+    requestDescriptionLarge,
+    descriptionLarge,
+    isLoadingDescriptionLarge,
+  ] = useImageQuery(imageData, 'descriptionLarge');
 
   return (
     <div className="space-y-8 max-w-[38rem]">
@@ -145,8 +151,17 @@ export default function PhotoForm({
         <button onClick={requestTags} disabled={isLoadingTags}>
           Tags ✨
         </button>
-        <button onClick={requestDescription} disabled={isLoadingDescription}>
-          Description ✨
+        <button
+          onClick={requestDescriptionSmall}
+          disabled={isLoadingDescriptionSmall}
+        >
+          Description (S) ✨
+        </button>
+        <button
+          onClick={requestDescriptionLarge}
+          disabled={isLoadingDescriptionLarge}
+        >
+          Description (L) ✨
         </button>
       </div>
       <div className="flex gap-2">
@@ -195,7 +210,14 @@ export default function PhotoForm({
         </>}
       </p>
       <p>
-        ✨ DESCRIPTION: {description} {isLoadingDescription && <>
+        ✨ DESCRIPTION (S): {descriptionSmall} {isLoadingDescriptionSmall && <>
+          <span className="inline-flex translate-y-[1.5px]">
+            <Spinner />
+          </span>
+        </>}
+      </p>
+      <p>
+        ✨ DESCRIPTION (L): {descriptionLarge} {isLoadingDescriptionLarge && <>
           <span className="inline-flex translate-y-[1.5px]">
             <Spinner />
           </span>