J.A.R.V.I.S-rust/resources/commands/math/math.lua

local lang = jarvis.context.language
local phrase = (jarvis.context.phrase or ""):lower()

local triggers = {
    "сколько будет", "сколько это", "посчитай", "вычисли", "реши",
    "how much is", "calculate", "compute", "solve", "what is",
    "скільки буде", "порахуй", "обчисли",
}

local query = phrase
for _, t in ipairs(triggers) do
    local s, e = string.find(query, t, 1, true)
    if s == 1 then
        query = query:sub(e + 1)
        break
    end
end
query = query:gsub("^%s+", ""):gsub("%s+$", "")

if query == "" then
    jarvis.system.notify("Math", lang == "ru" and "Что посчитать?" or "What to compute?")
    jarvis.audio.play_error()
    return { chain = false }
end

local token = jarvis.system.env("GROQ_TOKEN")
if not token or token == "" then
    jarvis.system.notify("Math", "GROQ_TOKEN не задан")
    jarvis.audio.play_error()
    return { chain = false }
end

local base  = jarvis.system.env("GROQ_BASE_URL"); if not base or base == "" then base = "https://api.groq.com/openai/v1" end
local model = jarvis.system.env("GROQ_MODEL");    if not model or model == "" then model = "llama-3.3-70b-versatile" end

local sys = "Ты калькулятор. Решай арифметику, простые уравнения, конверсии единиц, проценты. "
    .. "Отвечай ТОЛЬКО результатом числом или короткой строкой (для уравнений — значения корней). "
    .. "Если запрос — не математика, ответь одним словом «нет»."

local payload = {
    model = model,
    messages = {
        { role = "system", content = sys },
        { role = "user",   content = query },
    },
    max_tokens = 64,
    temperature = 0.0,
}

jarvis.log("info", "math query: " .. query)
local res = jarvis.http.post_json(base .. "/chat/completions", payload, { Authorization = "Bearer " .. token })

if not res.ok then
    jarvis.log("error", "math api failed: " .. tostring(res.status))
    jarvis.system.notify("Math", "Ошибка API")
    jarvis.audio.play_error()
    return { chain = false }
end

local body = res.body or ""
local content = body:match('"content"%s*:%s*"(.-[^\\])"')
if not content then
    jarvis.system.notify("Math", "Не смог распарсить ответ")
    jarvis.audio.play_error()
    return { chain = false }
end
content = content:gsub('\\n', '\n'):gsub('\\"', '"'):gsub('\\\\', '\\'):gsub('\\t', '\t')
content = content:gsub("^%s+", ""):gsub("%s+$", "")

if content:lower() == "нет" or content == "" then
    jarvis.system.notify("Math", lang == "ru" and "Это не математика" or "Not math")
    jarvis.audio.play_not_found()
    return { chain = false }
end

jarvis.system.notify(query, content)
jarvis.log("info", "math result: " .. content)

local sapi_text = ((lang == "ru" and "Получилось " or "Result: ") .. content):gsub("'", "''"):gsub("\r?\n", " ")
local ps = string.format(
    [[Add-Type -AssemblyName System.Speech; $s = New-Object System.Speech.Synthesis.SpeechSynthesizer; foreach ($v in $s.GetInstalledVoices()) { if ($v.VoiceInfo.Culture.TwoLetterISOLanguageName -eq 'ru') { try { $s.SelectVoice($v.VoiceInfo.Name); break } catch {} } } $s.Speak('%s')]],
    sapi_text
)
jarvis.system.exec(string.format('powershell -NoProfile -ExecutionPolicy Bypass -Command "%s"', ps:gsub('"', '\\"')))

jarvis.audio.play_ok()
return { chain = false }