geminiAPI.cpp

#include "geminiAPI.hpp"

using json = nlohmann::json;

// initializer list
GeminiClient::GeminiClient(std::string apiKey) :
    GEMINI_KEY(apiKey),
    state(State::IDLE) {}

void GeminiClient::reset() {
    state = State::IDLE;
    errorFeedback = "";
}

bool GeminiClient::isClientDoingSomething() {
    return state != GeminiClient::IDLE;
}

bool GeminiClient::callAPI(std::string prompt, std::string clipboardText) {
    if (state != State::IDLE) {
        return false;
    }
    state = State::RUNNING;

    std::string promptWithText = prompt + " Try your best to give me 10 suggestions. Here is the text:\n\n" + clipboardText;

    // FAILED quote test "this should fail" end text
    // FAILED slash test \this might fail\
    // semi quote test 'this might fail' end text
    // test ; , " \ , end text

    std::string escapedPrompt;
    for (char c : promptWithText) {
        if (c == '"' || c == '\\') { 
            escapedPrompt += '\\'; 
        }
        escapedPrompt += c;
    }

    cpr::PostCallback(
        // c++ lambda
        [this](cpr::Response response) {
            processResponse(response);
        },

        cpr::Url{ "https://generativelanguage.googleapis.com/v1beta/models/gemini-2.0-flash:generateContent?key=" + GEMINI_KEY },
        cpr::Body{ "{ \"contents\": [ { \"role\": \"user\", \"parts\": [ { \"text\": \"" + escapedPrompt + "\" }, ] }, ], \"generationConfig\": { \"responseMimeType\": \"application/json\", \"responseSchema\": { \"type\": \"object\", \"properties\": { \"suggestions\": { \"type\": \"array\", \"items\": { \"type\": \"string\" } } }, \"required\": [ \"suggestions\" ] }, }, \"safetySettings\": [ { \"category\": \"HARM_CATEGORY_CIVIC_INTEGRITY\", \"threshold\": \"BLOCK_LOW_AND_ABOVE\" }, ], }" },
        cpr::Header{ {"Content-Type", "application/json"} }
    );

    return true;
}

void GeminiClient::processResponse(cpr::Response response) {
    //std::cout << "!!!!!!!!!!! RAW RESPONSE" << response.text << std::endl;

    if (response.status_code != 200) {
        state = State::FAILED;
        errorFeedback = "There was an error trying to get suggestions from Gemini. (HTTP " + std::to_string(response.status_code) + ")";
        errorFeedback += "\n(CPR[" + std::to_string(static_cast<int>(response.error.code)) + "] " + response.error.message + ")";

        try {
            json jsonData = json::parse(response.text);
            errorFeedback += "\n" + jsonData["error"]["message"].template get<std::string>();
        }
        catch (json::exception e) {
            // double error
        }

        return;
    }

    try {
        json data = json::parse(response.text);
        std::string suggestionsRaw = data["candidates"][0]["content"]["parts"][0]["text"];

        json suggestionsJson = json::parse(suggestionsRaw);
        suggestions = suggestionsJson["suggestions"].template get<std::vector<std::string>>();
        state = State::FINISHED;
    }
    catch (json::exception e) {
        state = State::FAILED;
        std::string errorString(e.what());
        errorFeedback = "There was an error trying to parse Gemini's reponse.\n\n" + errorString;
    }
}

std::string GeminiClient::getPrompt(PromptType type) {
    switch (type) {
    case PromptType::SYNONYMS:
        return "If the following text snippet is a single word or a two word phrase, give me 10 synonyms for it. Otherwise give me 10 ways to say something similar. If suggestions cannot be expressed in 10 words due to needing more context or other reasons, give a variety of suggestions that span multiple possible contexts.";
        break;
    case PromptType::REPHRASE:
        return "Give me 10 other ways to rephrase the following text snippet.";
        break;
    case PromptType::FORMALIZE:
        return "Make the following text snippet more formal.";
        break;
    case PromptType::ANTONYMS:
        return "If the following text snippet is a single word or a two word phrase, give me 10 antonyms for it. Otherwise give me 10 ways to say the opposite thing while still keeping the same intent. If suggestions cannot be expressed in 10 words due to needing more context or other reasons, give a variety of suggestions that span multiple possible contexts.";
        break;
    case PromptType::UNGARBLE:
        return "This snippet of text doesn't sound quite right, please rewrite it while keeping the same tone, intent, and meaning.";
        break;
    case PromptType::SHORTEN:
        return "Make the following text snippet shorter.";
        break;
        // -----
    case PromptType::HEADLINE:
        return "Turn the following text snippet into a headline, like, for example, the title for a blog post or web page.";
        break;
    case PromptType::TAGLINE:
        return "Rewrite the following text snippet into a tag line.";
        break;
    case PromptType::ONEWORD:
        return "Rewrite the following text snippet into a one word phrase that encapsulates the same meaning.";
        break;
    case PromptType::TWOWORD:
        return "Rewrite the following text snippet into two word phrase that encapsulates the same meaning.";
        break;
    default:
        return "";
        break;
    }
}