#include <iostream>
#include <json/json.h>
#include <curl/curl.h>
#include <vector>
#include <Windows.h>
#include <sstream>
using namespace std;

// 回調(diào)函數(shù)
static size_t WriteCallback(void* contents, size_t size, size_t nmemb, void* userp) {
    ((std::string*)userp)->append((char*)contents, size * nmemb);
    return size * nmemb;
}

void addMessage(std::vector<Json::Value>& messages, const std::string& role, const std::string& content) {
    Json::Value message;
    message["role"] = role;
    message["content"] = content;
    messages.push_back(message);
}

int main() {
    // 設(shè)置控制臺為UTF-8模式
    system("chcp 65001 > nul");
    SetConsoleOutputCP(CP_UTF8);
    SetConsoleCP(CP_UTF8);

    std::string apiKey = "這里放你自己申請的APIkey（注意不要尖括號）";
    std::vector<Json::Value> messages;
    addMessage(messages, "system", "You are a helpful assistant.");

    CURL* curl = curl_easy_init();
    if (curl) {
        std::string readBuffer;
        std::string userInput;

        while (true) {
            std::cout << "User: ";
            std::getline(std::cin, userInput);
            if (userInput == "exit") break;

            addMessage(messages, "user", userInput);

            // 構(gòu)建JSON請求
            Json::Value jsonData;
            jsonData["model"] = "deepseek-chat";
            jsonData["stream"] = false;
            for (const auto& msg : messages) {
                jsonData["messages"].append(msg);
            }

            // 生成JSON字符串
            Json::StreamWriterBuilder writerBuilder;
            writerBuilder.settings_["emitUTF8"] = true;
            std::string jsonString = Json::writeString(writerBuilder, jsonData);

            // 設(shè)置CURL選項(xiàng)
            curl_easy_setopt(curl, CURLOPT_URL, "https://api.deepseek.com/chat/completions");
            curl_easy_setopt(curl, CURLOPT_POSTFIELDS, jsonString.c_str());
            curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, WriteCallback);
            curl_easy_setopt(curl, CURLOPT_WRITEDATA, &readBuffer);

            struct curl_slist* headers = NULL;
            headers = curl_slist_append(headers, "Content-Type: application/json");
            headers = curl_slist_append(headers, ("Authorization: Bearer " + apiKey).c_str());
            curl_easy_setopt(curl, CURLOPT_HTTPHEADER, headers);

            // 執(zhí)行請求
            CURLcode res = curl_easy_perform(curl);
            if (res != CURLE_OK) {
                std::cerr << "Request failed: " << curl_easy_strerror(res) << std::endl;
            }

            // 處理響應(yīng)
            Json::CharReaderBuilder readerBuilder;
            std::unique_ptr<Json::CharReader> reader(readerBuilder.newCharReader());
            Json::Value root;
            std::string errs;

            if (reader->parse(readBuffer.c_str(), readBuffer.c_str() + readBuffer.size(), &root, &errs)) {
                if (root.isMember("choices") && root["choices"].size() > 0) {
                    std::string response = root["choices"][0]["message"]["content"].asString();
                    std::cout << "Assistant: " << response << std::endl;
                    addMessage(messages, "assistant", response);
                }
                else {
                    std::cerr << "API Error: " << root.toStyledString() << std::endl;
                }
            }
            else {
                std::cerr << "JSON Parse Error: " << errs << std::endl;
            }

            readBuffer.clear();
            curl_slist_free_all(headers);
            curl_easy_reset(curl);
        }
        curl_easy_cleanup(curl);
    }
    return 0;
}