transcriptionworker.h

Go to the documentation of this file.

#pragma once
 
#include "audio/recording.h"
#include "audio/recordingnormalizer.h"
#include "transcription/transcriptionengine.h"
 
#include <QObject>
#include <memory>
 
class TranscriptionWorker final : public QObject
{
    Q_OBJECT
 
public:
    explicit TranscriptionWorker(std::shared_ptr<const TranscriptionEngine> engine, QObject *parent = nullptr);
    explicit TranscriptionWorker(std::unique_ptr<TranscriptionSession> transcriber, QObject *parent = nullptr);
    ~TranscriptionWorker() override = default;
 
    Q_DISABLE_COPY_MOVE(TranscriptionWorker)
 
    
    [[nodiscard]] QString backendName() const;
 
    [[nodiscard]] BackendCapabilities capabilities() const;
 
    [[nodiscard]] RuntimeDiagnostics runtimeDiagnostics() const;
 
    bool warmup(RuntimeError *error = nullptr);
 
    void transcribeRecordingCompat(const Recording &recording);
 
signals:
    void transcriptionReady(const QString &text);
 
    void transcriptionFailed(const RuntimeError &error);
 
private:
    static bool shouldDiscardSession(const RuntimeError &error);
 
    bool ensureModel(RuntimeError *error = nullptr);
    bool ensureSession(RuntimeError *error = nullptr);
 
    std::shared_ptr<const TranscriptionEngine> m_engine;
    std::shared_ptr<const TranscriptionModelHandle> m_model;
    BackendCapabilities m_capabilities;
    RuntimeDiagnostics m_runtimeDiagnostics;
    RecordingNormalizer m_normalizer;
    std::unique_ptr<TranscriptionSession> m_transcriber;
};