Merge pull request #91 from eric2788/develop

updates v2.0.3.1028
eric2788 · Oct 27, 2024 · da308fb · da308fb
2 parents daa1838 + 5f18349
commit da308fb
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 4 deletions.
diff --git a/src/hooks/form.ts b/src/hooks/form.ts
@@ -5,6 +5,12 @@ export function useFileInput(onFileChange: (files: FileList) => Promise<void>, o
     const inputRef = useRef<HTMLInputElement>()
     const selectFiles = useCallback(function (): Promise<void> {
         return new Promise((resolve, reject) => {
+            const finallize = () => {
+                inputRef.current.removeEventListener('change', listener)
+                inputRef.current.removeEventListener('cancel', finallize)
+                inputRef.current.files = null
+                resolve()
+            }
             const listener = async (e: Event) => {
                 try {
                     const files = (e.target as HTMLInputElement).files
@@ -15,12 +21,11 @@ export function useFileInput(onFileChange: (files: FileList) => Promise<void>, o
                     onError?.(e)
                     reject(e)
                 } finally {
-                    inputRef.current.removeEventListener('change', listener)
-                    inputRef.current.files = null
-                    resolve()
+                    finallize()
                 }
             }
             inputRef.current.addEventListener('change', listener)
+            inputRef.current.addEventListener('cancel', finallize)
             inputRef.current.click()
         })
     }, deps)

diff --git a/src/tabs/summarizer.tsx b/src/tabs/summarizer.tsx
@@ -52,7 +52,7 @@ function App() {
             const llmSettings = await getSettingStorage('settings.llm')
             const llm = createLLMProvider(llmSettings)
             llm.on('progress', (p, t) => setProgressText(() => p < 1 ? t : ''))
-            const summaryStream = llm.promptStream(`这位是一名在b站直播间直播的日本vtuber说过的话,请根据下文对话猜测与观众的互动内容,并用中文总结一下他们的对话:\n\n${danmakus.join('\n')}`)
+            const summaryStream = llm.promptStream(`这位是一名在b站直播间直播的日本vtuber说过的话${roomTitle ? `,其标题为"${roomTitle}"` : ''},请根据下文对话猜测与观众的互动内容,并用中文总结一下他们的对话:\n\n${danmakus.join('\n')}`)
             for await (const words of summaryStream) {
                 if (llm.cumulative) {
                     setSummary(summary => summary === loadingText ? words : summary + words)