TxtTranslator.h

#ifndef TXTTRANSLATOR_H
#define TXTTRANSLATOR_H

#include "FileTranslator.h"
#include "OpenaiManager.h"
#include "TxtParser.h"
#include "TxtPrompt.h"
#include <QPlainTextEdit>
#include <QString>
#include <QTime>
#include <QVector>

class TxtTranslator : public FileTranslator
{
public:
    TxtTranslator(int maxLen, int minLen);
    ~TxtTranslator();

    // 实现父类的虚函数
    void setLog(QPlainTextEdit *logOutput) override
    {
        m_logOutput = logOutput;
    }
    bool translate(const QString &inputFilePath, const QString &outputFilePath, const QString &url,
                   const QString &apiKey, const QString &model, bool keepHistory = false) override;

private:
    // 实现父类的虚函数
    void outlog(const QString &log) override;

    // 构建完整的翻译提示
    QString buildPrompt(const QString &content, const PromptInfo &promptInfo);

    TxtParser m_txtParser;
    OpenaiManager *m_openaiManager;
    TxtPrompt *m_translatePrompt;

    // 用于保存最近的翻译历史
    struct TranslateHistory
    {
        QString original;   // 原文
        QString translated; // 译文
    };
    QVector<TranslateHistory> m_history;

    // 段落长度限制
    int m_maxLen = 3072; // 最大段落长度
    int m_minLen = 1024; // 最小段落长度
};

#endif // TXTTRANSLATOR_H