NG_2024_Stanislav_Mykhailenko/Lesson_8/Task_1/llmmathquiz.cpp

#include "llmmathquiz.h"
#include "ui_llmmathquiz.h"

LlmMathQuiz::LlmMathQuiz(QWidget *parent)
    : QMainWindow(parent)
    , ui(new Ui::LlmMathQuiz)
{
    ui->setupUi(this);
    connect(ui->b_submit, &QPushButton::clicked, this, &LlmMathQuiz::submit);
    connect(ui->b_generate, &QPushButton::clicked, this, &LlmMathQuiz::generate);
    connect (m_manager, &QNetworkAccessManager::finished, this, &LlmMathQuiz::newAnswer);
}

LlmMathQuiz::~LlmMathQuiz()
{
    delete ui;
}

void LlmMathQuiz::submit()
{
    if (ui->e_user->text().isEmpty())
        return;

    ui->e_llm->setText(m_llmAnswer);
    ui->b_submit->setEnabled(false);
    ui->e_user->setReadOnly(true);
    ui->l_opinion->setText("Generating…");

    sendPrompt("The two following lines will contain two answers for a mathematical task. The first value is the expected answer, and the second one is the answer provided. Respond with \"true\" if the provided answer matches the expected answer, and \"false\" if it is not. Always use lowercase."
               + QString("\n") + m_llmAnswer
               + QString("\n") + ui->e_user->text());
}

void LlmMathQuiz::generate()
{
    if (m_taskReceived) {
        ui->l_number->setText(QString::number(ui->l_number->text().toInt() + 1));
        m_taskReceived = false;
    }
    m_taskReceived = false;
    ui->b_generate->setEnabled(false);
    ui->e_task->setText("Generating…");
    ui->e_user->clear();
    ui->e_llm->clear();
    ui->l_opinion->setText("Not checked");

    sendPrompt("Generate a math task and a correct answer for it. The task can be anything from simple equations, to logical tasks. Both the task and the answer need to take only one line. Provide the task on the first line, and the answer on the second line. Never use equation 2x + 5 = 11.");
}

void LlmMathQuiz::newAnswer(QNetworkReply *reply)
{
    if (!m_taskReceived) {
        if (reply->error() == QNetworkReply::NoError) {
            QByteArray textReply = reply->readAll();
            QStringList data = QJsonDocument::fromJson(textReply).object().value("response").toString().split('\n');
            ui->e_task->setText(data[0]);
            m_llmAnswer = data[1].isEmpty() ? data[2] : data[1];
            m_taskReceived = true;
            ui->b_submit->setEnabled(true);
            ui->e_user->setReadOnly(false);
        } else {
            ui->e_task->setText("Error!");
            ui->b_generate->setEnabled(true);
        }
    } else {
        if (reply->error() == QNetworkReply::NoError) {
            QByteArray textReply = reply->readAll();
            QString data = QJsonDocument::fromJson(textReply).object().value("response").toString();
            if (data == "true") {
                ui->l_matched->setText(QString::number(ui->l_matched->text().toInt() + 1));
                ui->l_opinion->setText("Match");
            } else if (data == "false")
                ui->l_opinion->setText("Mismatch");
            else
                ui->l_opinion->setText("Not match nor mismatch");
            ui->b_generate->setEnabled(true);
        } else {
            ui->e_llm->setText("Error!");
            ui->b_submit->setEnabled(true);
        }
    }
}

void LlmMathQuiz::sendPrompt(QString prompt)
{
    QNetworkRequest request(QUrl("http://127.0.0.1:11434/api/generate"));
    request.setHeader(QNetworkRequest::ContentTypeHeader, "application/json");

    QJsonObject json;
    json["model"] = "llama3.1:8b";
    json["prompt"] = prompt;
    json["stream"] = false;

    m_manager->post(request, QJsonDocument(json).toJson());
}