feat: transcribe immediately after download from URL

thewh1teagle · Oct 15, 2024 · 276a6a2 · 276a6a2
1 parent cb0eba9
commit 276a6a2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/desktop/src/pages/home/Page.tsx b/desktop/src/pages/home/Page.tsx
@@ -106,7 +106,7 @@ export default function Home() {
  )}
  {vm.audio && !vm.loading && (
  <>
- <button onMouseDown={vm.transcribe} className="btn btn-primary mt-3">
+ <button onMouseDown={() => vm.transcribe(vm.files[0].path)} className="btn btn-primary mt-3">
  {t('common.transcribe')}
  </button>
  <ModelOptions options={vm.preference.modelOptions} setOptions={vm.preference.setModelOptions} />

diff --git a/desktop/src/pages/home/viewModel.ts b/desktop/src/pages/home/viewModel.ts
@@ -258,15 +258,15 @@ export function viewModel() {
  emit('stop_record')
  }
 
- async function transcribe() {
+ async function transcribe(path: string) {
  setSegments(null)
  setLoading(true)
  abortRef.current = false
 
  try {
  await invoke('load_model', { modelPath: preference.modelPath, gpuDevice: preference.gpuDevice })
  const options = {
- path: files[0].path,
+ path,
  ...preference.modelOptions,
  }
  const startTime = performance.now()
@@ -324,6 +324,7 @@ export function viewModel() {
  const outPath = await ytDlp.downloadAudio(audioUrl, preference.storeRecordInDocuments)
  preference.setHomeTabIndex(1)
  setFiles([{ name: 'audio.m4a', path: outPath }])
+ transcribe(outPath)
  } catch (e) {
  console.error(e)
  setErrorModal?.({ log: String(e), open: true })