카카오톡 챗봇(메신저봇) + 구글 시트 + Gemini 로 자동 CS봇 만들기 테스트입니다. (2부)

// CS봇 2024-10-27  1.1.7
// gemini-1.5-flash_api_key 사용
// 구글시트(SHEET_ID, SHEET_API_KEY)-카카오톡(메신저봇) 연동
// 답변에 Markdown 중 ## 와 ** 제거

// 구글 시트 주소 : https://docs.google.com/spreadsheets...
// 구글 시트 공유 : 뷰어로 변경하기 (링크가 있는 인터넷상의 모든 사용자가 볼 수 있음 )
// 호출 : "CS봇", "cs봇", "!질문",  "!"

var SHEET_ID = "SHEET_ID";
var SHEET_API_KEY = "SHEET_API_KEY";
var GEMINI_API_KEY = 'GEMINI_API_KEY';
var GEMINI_ENDPOINT = 'https://generativelanguage.googleapis.com/v1/models/gemini-1.5-flash:generateContent?key=';

function customLog(message) {
    var logMessage = new Date().toLocaleString() + ": " + message;
    Log.i(logMessage);
}

function extractJSON(text) {
    var match = text.match(/\{[\s\S]*\}/);
    return match ? match[0] : null;
}

function getSheetNames() {
    var url = "https://sheets.googleapis.com/v4/spreadsheets/" + SHEET_ID + "?key=" + SHEET_API_KEY;
    try {
        var response = Utils.getWebText(url);
        var jsonStr = extractJSON(response);
        if (!jsonStr) {
            customLog("JSON 데이터를 추출할 수 없습니다.");
            return null;
        }
        var data = JSON.parse(jsonStr);
        return data.sheets.map(function(sheet) {
            return sheet.properties.title;
        });
    } catch (e) {
        customLog("시트 이름 가져오기 에러: " + e.message);
        return null;
    }
}

function getSheetData(sheetName) {
    var url = "https://sheets.googleapis.com/v4/spreadsheets/" + SHEET_ID + "/values/" + 
              encodeURIComponent(sheetName) + "!A1:B1000?key=" + SHEET_API_KEY;
    try {
        var response = Utils.getWebText(url);
        var jsonStr = extractJSON(response);
        if (!jsonStr) {
            customLog("JSON 데이터를 추출할 수 없습니다.");
            return null;
        }
        var data = JSON.parse(jsonStr);
        return data.values;
    } catch (e) {
        customLog(sheetName + " 시트 데이터 가져오기 에러: " + e.message);
        return null;
    }
}

function getAllSheetsData() {
    var sheetNames = getSheetNames();
    if (!sheetNames) {
        return null;
    }
    
    var allData = [];
    sheetNames.forEach(function(sheetName) {
        var sheetData = getSheetData(sheetName);
        if (sheetData) {
            allData = allData.concat(sheetData);
        }
    });
    
    return allData;
}

// Markdown 중 ## 와 ** 제거
function removeSpecificMarkdown(text) {
    return text
        .replace(/^##\s+/gm, '')  // 줄 시작 부분의 ## 제거
        .replace(/\*\*(.*?)\*\*/g, '$1')  // ** 제거
        .replace(/^\s*\*\s/gm, '• ');  // 줄 시작 부분의 * 을 • 로 변경
}

function callGeminiAPI(prompt) {
    var url = GEMINI_ENDPOINT + GEMINI_API_KEY;
    var headers = {
        'Content-Type': 'application/json'
    };
    var data = {
        "contents": [{
            "parts":[{
                "text": prompt
            }]
        }]
    };
    
    try {
        var response = org.jsoup.Jsoup.connect(url)
            .ignoreContentType(true)
            .headers(headers)
            .requestBody(JSON.stringify(data))
            .post();
        
        var jsonResponse = JSON.parse(response.body().text());
        var rawText = jsonResponse.candidates[0].content.parts[0].text;
        
        // 링크 활성화 로직 추가
        rawText = rawText.replace(/(https?:\/\/\S+)/g, '<$1>');
        // ##와 ** 제거
        return removeSpecificMarkdown(rawText);  
    } catch (e) {
        customLog("Gemini API 호출 에러: " + e.message);
        return null;
    }
}

function response(room, msg, sender, isGroupChat, replier, imageDB, packageName) {
    var triggers = ["CS봇", "cs봇", "!질문", "!"];
    var isQuestion = triggers.some(function(trigger) {
        return msg.toLowerCase().startsWith(trigger.toLowerCase());
    });

    if (isQuestion) {
        var question = msg.replace(/^(CS봇\s*|cs봇\s*|!질문\s*|!\s*)/i, "").trim();
        
        if (question === "") {
            replier.reply("질문을 입력해주세요. 예: CS봇 전자책 또는 !질문 강의 다시보기");
            return;
        }
        
        replier.reply("CS봇이 질문을 분석 중입니다. 잠시만 기다려주세요...");
        
        var allSheetsData = getAllSheetsData();
        if (!allSheetsData) {
            replier.reply("죄송합니다. 데이터를 가져오는 데 문제가 발생했습니다.");
            return;
        }
        
        var context = "다음은 FAQ 데이터입니다:\n\n";
        allSheetsData.forEach(function(row) {
            if (row[0] && row[1]) {
                context += "Q: " + row[0] + "\nA: " + row[1] + "\n\n";
            }
        });
        
        var prompt = context + "\n사용자 질문: " + question + 
                     "\n\n위의 FAQ 데이터를 참고하여 사용자의 질문에 친절하고 상세하게 답변해주세요. " +
                     "FAQ에 관련 내용이 없다면 '죄송합니다. 현재 DB에 관련 내용이 없습니다. 더 자세한 정보를 원하시면 스탭에게 문의해 주세요.'라고 답변해주세요. " +
                     "답변은 한국어로 작성해주세요.";
        
        var answer = callGeminiAPI(prompt);
        if (answer) {
            replier.reply("[CS봇의 답변]\n\n" + removeSpecificMarkdown(answer));
        } else {
            replier.reply("죄송합니다. 답변을 생성하는 데 문제가 발생했습니다.");
        }
    }
}