Function STTScreen

example/src/STTScreen.tsx:18–241

()

import * as RNFS from '@dr.pogodin/react-native-fs';

const STTScreen = () => {
  const cactusSTT = useCactusSTT({ model: 'whisper-small' });
  const [audioFile, setAudioFile] = useState<string | null>(null);
  const [audioFileName, setAudioFileName] = useState<string>('');
  const [result, setResult] = useState<CactusSTTTranscribeResult | null>(null);
  const [embeddingResult, setEmbeddingResult] =
    useState<CactusSTTAudioEmbedResult | null>(null);

  useEffect(() => {
    if (!cactusSTT.isDownloaded) {
      cactusSTT.download();
    }
    // eslint-disable-next-line react-hooks/exhaustive-deps
  }, [cactusSTT.isDownloaded]);

  const handleInit = () => {
    cactusSTT.init();
  };

  const handleSelectAudio = async () => {
    try {
      const res = await DocumentPicker.pick({
        type: [DocumentPicker.types.audio],
      });
      if (res && res.length > 0) {
        const fileName = `audio_${Date.now()}.wav`;
        const destPath = `${RNFS.CachesDirectoryPath}/${fileName}`;
        await RNFS.copyFile(res[0].uri, destPath);
        setAudioFile(destPath);
        setAudioFileName(res[0].name || 'Unknown');
      }
    } catch (err) {
      console.error(err);
    }
  };

  const handleTranscribe = async () => {
    if (!audioFile) {
      return;
    }
    const transcribeResult = await cactusSTT.transcribe({
      audio: audioFile,
    });
    setResult(transcribeResult);
  };

  const handleAudioEmbed = async () => {
    if (!audioFile) {
      return;
    }
    const embedResult = await cactusSTT.audioEmbed({
      audioPath: audioFile,
    });
    setEmbeddingResult(embedResult);
  };

  const handleStop = () => {
    cactusSTT.stop();