You've already forked M5Module-LLM
mirror of
https://github.com/m5stack/M5Module-LLM.git
synced 2026-05-20 11:31:45 -07:00
lovyan03
87 lines
2.3 KiB
Arduino
87 lines
2.3 KiB
Arduino
/*
|
|
* SPDX-FileCopyrightText: 2024 M5Stack Technology CO LTD
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*/
|
|
#include <Arduino.h>
|
|
#include <M5Unified.h>
|
|
#include <M5ModuleLLM.h>
|
|
|
|
M5ModuleLLM module_llm;
|
|
M5ModuleLLM_VoiceAssistant voice_assistant(&module_llm);
|
|
|
|
/* On ASR data callback */
|
|
void on_asr_data_input(String data, bool isFinish, int index)
|
|
{
|
|
M5.Display.setTextColor(TFT_GREEN, TFT_BLACK);
|
|
// M5.Display.setFont(&fonts::efontCN_12); // Support Chinese display
|
|
M5.Display.printf(">> %s\n", data.c_str());
|
|
|
|
/* If ASR data is finish */
|
|
if (isFinish) {
|
|
M5.Display.setTextColor(TFT_YELLOW, TFT_BLACK);
|
|
M5.Display.print(">> ");
|
|
}
|
|
};
|
|
|
|
/* On LLM data callback */
|
|
void on_llm_data_input(String data, bool isFinish, int index)
|
|
{
|
|
M5.Display.print(data);
|
|
|
|
/* If LLM data is finish */
|
|
if (isFinish) {
|
|
M5.Display.print("\n");
|
|
}
|
|
};
|
|
|
|
void setup()
|
|
{
|
|
M5.begin();
|
|
M5.Display.setTextSize(2);
|
|
M5.Display.setTextScroll(true);
|
|
|
|
/* Init module serial port */
|
|
// int rxd = 16, txd = 17; // Basic
|
|
// int rxd = 13, txd = 14; // Core2
|
|
// int rxd = 18, txd = 17; // CoreS3
|
|
int rxd = M5.getPin(m5::pin_name_t::port_c_rxd);
|
|
int txd = M5.getPin(m5::pin_name_t::port_c_txd);
|
|
Serial2.begin(115200, SERIAL_8N1, rxd, txd);
|
|
|
|
/* Init module */
|
|
module_llm.begin(&Serial2);
|
|
|
|
/* Make sure module is connected */
|
|
M5.Display.printf(">> Check ModuleLLM connection..\n");
|
|
while (1) {
|
|
if (module_llm.checkConnection()) {
|
|
break;
|
|
}
|
|
}
|
|
|
|
/* Begin voice assistant preset */
|
|
M5.Display.printf(">> Begin voice assistant..\n");
|
|
int ret = voice_assistant.begin("HELLO");
|
|
// int ret = voice_assistant.begin("你好你好", "", "zh_CN"); // Chinese kws and asr
|
|
if (ret != MODULE_LLM_OK) {
|
|
while (1) {
|
|
M5.Display.setTextColor(TFT_RED);
|
|
M5.Display.printf(">> Begin voice assistant failed\n");
|
|
}
|
|
}
|
|
|
|
/* Register on ASR data callback function */
|
|
voice_assistant.onAsrDataInput(on_asr_data_input);
|
|
|
|
/* Register on LLM data callback function */
|
|
voice_assistant.onLlmDataInput(on_llm_data_input);
|
|
|
|
M5.Display.printf(">> Voice assistant ready\n");
|
|
}
|
|
|
|
void loop()
|
|
{
|
|
/* Keep voice assistant preset update */
|
|
voice_assistant.update();
|
|
} |