From 3aaf8979a0658bc69d71dd209f0f7b9baef547d8 Mon Sep 17 00:00:00 2001
From: Shotaro Nakamura <syoutarou1103@outlook.jp>
Date: Sat, 9 Nov 2024 07:19:48 +0000
Subject: [PATCH] feat

---
 .../note/components/notes/TextNote/AI.tsx     | 244 +++++++++++++-----
 1 file changed, 182 insertions(+), 62 deletions(-)

diff --git a/src/islands/note/components/notes/TextNote/AI.tsx b/src/islands/note/components/notes/TextNote/AI.tsx
index fc41180..3cc9ce5 100644
--- a/src/islands/note/components/notes/TextNote/AI.tsx
+++ b/src/islands/note/components/notes/TextNote/AI.tsx
@@ -1,10 +1,15 @@
-import type {
-  GenerateContentStreamResult,
-  GoogleGenerativeAI,
+import {
+  type GenerateContentStreamResult,
+  type GoogleGenerativeAI,
+  GoogleGenerativeAIResponseError,
+  type GenerateContentResponse,
+  GoogleGenerativeAIFetchError,
 } from '@google/generative-ai'
 import markdownIt from 'markdown-it'
 import {
+  Match,
   Show,
+  Switch,
   createComputed,
   createEffect,
   createMemo,
@@ -20,6 +25,9 @@ import { Spinner } from '../../utils/Spinner'
 const markdownParser = markdownIt()
 
 type Mode = 'text' | 'image'
+type Generate = {
+  generate: () => Promise<GenerateContentStreamResult>
+}
 
 const renderMarkdown = (markdown: string) =>
   markdownParser
@@ -85,7 +93,7 @@ const getGemini = (): GoogleGenerativeAI => {
   return ai
 }
 export const FromText = (props: {
-  setStream(stream: GenerateContentStreamResult): void
+  setStream(stream: Generate): void
   initPrompt?: string
 }) => {
   const [getCurrentPlaceholder, setCurrentPlaceholder] = createSignal({
@@ -157,20 +165,21 @@ export const FromText = (props: {
     const model = gemini.getGenerativeModel({
       model: 'gemini-1.5-flash',
     })
-    props.setStream(
-      await model
-        .startChat({
-          systemInstruction: {
-            role: 'model',
-            parts: [
-              {
-                text: 'ユーザーの指示に基づき、暗記の手助けになる赤シート用文章を生成しなさい。赤シートで隠すべき単語は、Markdownの太字機能で表現しなさい。隠す必要がない場所には太字は使わないでください。太字の場所はユーザーに見えません。また、単語の一覧を箇条書きにすることはしないでください。',
-              },
-            ],
-          },
-        })
-        .sendMessageStream(getPrompt()),
-    )
+    props.setStream({
+      generate: () =>
+        model
+          .startChat({
+            systemInstruction: {
+              role: 'model',
+              parts: [
+                {
+                  text: 'ユーザーの指示に基づき、暗記の手助けになる赤シート用文章を生成しなさい。赤シートで隠すべき単語は、Markdownの太字機能で表現しなさい。隠す必要がない場所には太字は使わないでください。太字の場所はユーザーに見えません。また、単語の一覧を箇条書きにすることはしないでください。',
+                },
+              ],
+            },
+          })
+          .sendMessageStream(getPrompt()),
+    })
     setIsPreprocessing(false)
   }
 
@@ -192,7 +201,7 @@ export const FromText = (props: {
           disabled={!getPrompt() || getIsPreprocessing()}
           type="button"
         >
-          <Show when={getIsPreprocessing()} fallback='生成'>
+          <Show when={getIsPreprocessing()} fallback="生成">
             <Spinner />
             準備中...
           </Show>
@@ -202,7 +211,7 @@ export const FromText = (props: {
   )
 }
 export const FromImage = (props: {
-  setStream(stream: GenerateContentStreamResult): void
+  setStream(stream: Generate): void
 }) => {
   const [getImageFile, setImageFile] = createSignal<File>()
   const [getScanedImageURL, setScanedImageURL] = createSignal<string>()
@@ -236,29 +245,30 @@ export const FromImage = (props: {
       reader.readAsDataURL(file)
     })
 
-    const stream = await model
-      .startChat({
-        systemInstruction: {
-          role: 'model',
-          parts: [
-            {
-              text: '画像を抽出し、そっくりそのまま書き出しなさい。省略せずに画像の文字全てを書き出すこと。画像に書いていないことは書かないこと。また、ユーザーからの指示があれば、条件を満たす場所をMarkdownの太字で表現しなさい。',
+    const generate = () =>
+      model
+        .startChat({
+          systemInstruction: {
+            role: 'model',
+            parts: [
+              {
+                text: '画像を抽出し、そっくりそのまま書き出しなさい。省略せずに画像の文字全てを書き出すこと。画像に書いていないことは書かないこと。また、ユーザーからの指示があれば、条件を満たす場所をMarkdownの太字で表現しなさい。',
+              },
+            ],
+          },
+        })
+        .sendMessageStream([
+          {
+            text: '',
+          },
+          {
+            inlineData: {
+              mimeType: file.type,
+              data: b64,
             },
-          ],
-        },
-      })
-      .sendMessageStream([
-        {
-          text: '',
-        },
-        {
-          inlineData: {
-            mimeType: file.type,
-            data: b64,
           },
-        },
-      ])
-    props.setStream(stream)
+        ])
+    props.setStream({ generate })
     setIsPreprocessing(false)
   }
 
@@ -284,7 +294,12 @@ export const FromImage = (props: {
             <label class="h-full flex flex-col">
               <div>どのように処理するかのカスタムプロンプト(任意)</div>
               <textarea class="w-full m-1 p-1 border grow" placeholder="" />
-              <button onClick={generate} type="button" class="filled-button disabled:opacity-80 flex items-center gap-1 justify-center" disabled={getIsPreprocessing()}>
+              <button
+                onClick={generate}
+                type="button"
+                class="filled-button disabled:opacity-80 flex items-center gap-1 justify-center"
+                disabled={getIsPreprocessing()}
+              >
                 <Show when={getIsPreprocessing()} fallback={'生成'}>
                   <Spinner />
                   準備中...
@@ -339,27 +354,81 @@ export const AIDialogCore = (props: {
   initPrompt?: string
 }) => {
   const [getGenerateMode, setGenerateMode] = createSignal<Mode>('text')
-  const [getStream, setStream] = createSignal<GenerateContentStreamResult>()
+  const [getStream, setStream] = createSignal<Generate>()
   const [getGenerated, setGenerated] = createSignal('')
   const [getIsGenerating, setIsGenerating] = createSignal(false)
+  const [getGenerateError, setGenerateError] = createSignal<{
+    type: 'SERVICE_UNAVAILABLE' | 'UNKNOWN' | 'SAFETY' | 'RATE_LIMIT'
+    isFetchError: boolean
+  }>()
   let outputRef!: HTMLDivElement
 
-  createEffect(() => {
-    const stream = getStream()
-    if (!stream) {
+  const generate = async () => {
+    setIsGenerating(true)
+    setGenerated('')
+    setGenerateError()
+    const streamer = getStream()
+    if (!streamer) {
+      return
+    }
+    let stream: GenerateContentStreamResult
+    try {
+      stream = await streamer.generate()
+    } catch (error) {
+      if (error instanceof GoogleGenerativeAIFetchError) {
+        switch (error.status) {
+          case 429:
+            setGenerateError({ type: 'RATE_LIMIT', isFetchError: true })
+            break
+          case 500:
+          case 503:
+            setGenerateError({
+              type: 'SERVICE_UNAVAILABLE',
+              isFetchError: true,
+            })
+            break
+          default:
+            setGenerateError({ type: 'UNKNOWN', isFetchError: true })
+            break
+        }
+        setIsGenerating(false)
+        return
+      }
+      setIsGenerating(false)
       return
     }
-    ;(async () => {
-      setIsGenerating(true)
+    try {
       for await (const chunk of stream.stream) {
         setGenerated(`${getGenerated()}${chunk.text()}`)
         outputRef.scrollTo({
           behavior: 'smooth',
-          top: 100 + outputRef.scrollHeight
+          top: 100 + outputRef.scrollHeight,
         })
       }
+    } catch (error) {
+      if (error instanceof GoogleGenerativeAIResponseError) {
+        const response = (
+          error as GoogleGenerativeAIResponseError<GenerateContentResponse>
+        ).response
+        if (response?.candidates?.[0]?.finishReason === 'SAFETY') {
+          setGenerateError({
+            type: 'SAFETY',
+            isFetchError: false,
+          })
+          return
+        }
+      }
+      setGenerateError({
+        type: 'UNKNOWN',
+        isFetchError: false,
+      })
+    } finally {
       setIsGenerating(false)
-    })()
+    }
+  }
+  createEffect(() => {
+    getStream()
+    generate()
   })
   return (
     <div>
@@ -383,19 +452,70 @@ export const AIDialogCore = (props: {
             innerHTML={renderMarkdown(getGenerated())}
             ref={outputRef}
           />
-          <button
-            onClick={() => props.close(renderMarkdown(getGenerated()))}
-            class="filled-button disabled:opacity-80"
-            disabled={getIsGenerating()}
-            type="button"
+          <div
+            classList={{
+              'grid grid-cols-1 md:grid-cols-2': !!getGenerateError(),
+            }}
           >
-            <Show when={getIsGenerating()} fallback='結果を挿入する'>
-              <div class="flex gap-2 justify-start items-center">
-                <Spinner />
-                生成中...
-              </div>
-            </Show>
-          </button>
+            <div class="text-error">
+              <Switch>
+                <Match when={getGenerateError()?.type === 'SAFETY'}>
+                  安全上の理由により停止されました
+                </Match>
+                <Match when={getGenerateError()?.type === 'RATE_LIMIT'}>
+                  API
+                  呼び出しの上限に達しました。時間をおいて試す、またはカスタム
+                  Gemini API Key
+                  を設定するか、更新することにより解決するかもしれません。
+                </Match>
+                <Match
+                  when={getGenerateError()?.type === 'SERVICE_UNAVAILABLE'}
+                >
+                  Google
+                  のサーバーにエラーが発生している可能性があります。時間をおいて試してください。
+                </Match>
+                <Match when={getGenerateError()?.type === 'UNKNOWN'}>
+                  Gemini の使用において、不明なエラーが発生しました。
+                </Match>
+              </Switch>
+            </div>
+            <div class="flex items-center">
+              <Show when={!getIsGenerating() && getGenerateError()}>
+                <button class="filled-button" type="button" onClick={generate}>
+                  再生成
+                </button>
+              </Show>
+              <Show when={!getGenerateError()?.isFetchError}>
+                <button
+                  onClick={() => props.close(renderMarkdown(getGenerated()))}
+                  classList={{
+                    'text-button': !!getGenerateError(),
+                    'filled-button': !getGenerateError(),
+                  }}
+                  class="disabled:opacity-80"
+                  disabled={getIsGenerating()}
+                  type="button"
+                >
+                  <Show
+                    when={getIsGenerating()}
+                    fallback={
+                      getGenerateError() ? 'このまま挿入' : '結果を挿入する'
+                    }
+                  >
+                    <div class="flex gap-2 justify-start items-center">
+                      <Spinner />
+                      生成中...
+                    </div>
+                  </Show>
+                </button>
+              </Show>
+              <Show when={!getIsGenerating() && !getGenerateError()}>
+                <button class="text-button" type="button" onClick={generate}>
+                  再生成
+                </button>
+              </Show>
+            </div>
+          </div>
         </div>
       </Show>
     </div>