76 lines
2.0 KiB
C#
76 lines
2.0 KiB
C#
using System;
|
|
using System.IO;
|
|
using Vosk;
|
|
using System.Text.Json;
|
|
using NAudio.Wave;
|
|
using Avalonia.Controls;
|
|
using Avalonia.Threading;
|
|
|
|
namespace VisionAsist.Models;
|
|
|
|
public class TrigerCore
|
|
{
|
|
|
|
private WaveInEvent? _waveIn;
|
|
private Model? _model;
|
|
private VoskRecognizer? _rec;
|
|
private readonly object _voskLock = new();
|
|
public string RecognizedText { get; private set; } = "";
|
|
|
|
|
|
public TrigerCore()
|
|
{
|
|
string VoskPath = Path.Combine(AppContext.BaseDirectory, "models/Vosk/");
|
|
if (!Directory.Exists(VoskPath))
|
|
throw new DirectoryNotFoundException($"Модель не найдена по пути: {VoskPath}");
|
|
|
|
_model = new Model(VoskPath);
|
|
_rec = new VoskRecognizer(_model, 16000.0f);
|
|
|
|
}
|
|
|
|
public void StartRecording()
|
|
{
|
|
if (_waveIn != null || _rec == null) return;
|
|
|
|
_waveIn = new WaveInEvent { WaveFormat = new WaveFormat(16000, 1) };
|
|
_waveIn.DataAvailable += OnDataAvailable;
|
|
_waveIn.StartRecording();
|
|
}
|
|
|
|
|
|
public void StopRecording()
|
|
{
|
|
_waveIn?.StopRecording();
|
|
_waveIn?.Dispose();
|
|
_waveIn = null;
|
|
}
|
|
public event Action<string>? OnRecognized;
|
|
|
|
private void OnDataAvailable(object? sender, WaveInEventArgs e)
|
|
{
|
|
lock (_voskLock)
|
|
{
|
|
if (_rec != null && _rec.AcceptWaveform(e.Buffer, e.BytesRecorded))
|
|
{
|
|
var json = _rec.Result();
|
|
using var doc = JsonDocument.Parse(json);
|
|
var result = doc.RootElement.GetProperty("text").GetString();
|
|
|
|
if (!string.IsNullOrWhiteSpace(result))
|
|
{
|
|
RecognizedText += result + " ";
|
|
OnRecognized?.Invoke(result); // уведомляем подписчиков
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
protected void Stop()
|
|
{
|
|
StopRecording();
|
|
_rec?.Dispose();
|
|
_model?.Dispose();
|
|
|
|
}
|
|
} |