#3026 ui: implement AI enable checkbox

pbek · May 15, 2024 · b943032 · b943032
1 parent c14d71d
commit b943032
Show file tree

Hide file tree

Showing 4 changed files with 24 additions and 0 deletions.
diff --git a/src/mainwindow.cpp b/src/mainwindow.cpp
@@ -11892,6 +11892,9 @@ void MainWindow::on_actionMove_down_in_tag_list_triggered() {
 void MainWindow::buildAiToolbarAndActions() {
     _aiToolbar->clear();
     _aiToolbar->addAction(ui->actionEnable_AI);
+    ui->actionEnable_AI->blockSignals(true);
+    ui->actionEnable_AI->setChecked(OpenAiService::getEnabled());
+    ui->actionEnable_AI->blockSignals(false);
 
     _aiBackendComboBox = new QComboBox(this);
     connect(_aiBackendComboBox,
@@ -11923,3 +11926,8 @@ void MainWindow::buildAiToolbarAndActions() {
     aiModelWidgetAction->setText(tr("AI model selector"));
     _aiToolbar->addAction(aiModelWidgetAction);
 }
+
+void MainWindow::on_actionEnable_AI_toggled(bool arg1) {
+    OpenAiService::setEnabled(arg1);
+    qDebug() << __func__ << " - 'checked': " << arg1;
+}
diff --git a/src/mainwindow.h b/src/mainwindow.h
@@ -706,6 +706,8 @@ class MainWindow : public QMainWindow {
 
     void on_actionMove_down_in_tag_list_triggered();
 
+    void on_actionEnable_AI_toggled(bool arg1);
+
    public:
     /** Settings access **/
     static bool isInDistractionFreeMode();

diff --git a/src/services/openaiservice.cpp b/src/services/openaiservice.cpp
@@ -154,6 +154,18 @@ QString OpenAiService::getCurrentModelSettingsKey() {
     return QStringLiteral("ai/") + getBackendId() + QStringLiteral("/") + QStringLiteral("currentModel");
 }
 
+bool OpenAiService::setEnabled(bool enabled) {
+    QSettings settings;
+    settings.setValue(QStringLiteral("ai/enabled"), enabled);
+
+    return true;
+ }
+
+bool OpenAiService::getEnabled() {
+    QSettings settings;
+    return settings.value(QStringLiteral("ai/enabled")).toBool();
+}
+
 OpenAiCompleter::OpenAiCompleter(QString apiKey, QString modelId, QString apiBaseUrl,
                                  QObject* parent)
     : QObject(parent),

diff --git a/src/services/openaiservice.h b/src/services/openaiservice.h
@@ -30,6 +30,8 @@ class OpenAiService : public QObject {
     QString getBackendId();
     bool setModelId(const QString& id);
     QString getModelId();
+    static bool setEnabled(bool enabled);
+    static bool getEnabled();
 
    private:
     QMap<QString, QStringList> backendModels;