reintegration complete

modules/Archimedes-Modules/Ollama/Ollama.cpp

@@ -0,0 +1,126 @@
+#include "Ollama.h"
+#include "modules/ImguiModule/ImguiModule.h"
+
+#include <sstream>
+
+Ollama::Ollama(Archimedes::App* a, void* h) : Archimedes::Module(a, h) {
+
+    name = "Ollama";
+
+    ImguiModule* im = new ImguiModule(a, h);
+    deps[*im] = im;
+}
+
+Ollama::~Ollama() {
+    if(app) {
+        if(curl) {
+            curl_easy_cleanup(curl);
+            curl = nullptr;
+        }
+        curl_global_cleanup();
+
+        ImguiModule* im; { im = (ImguiModule*) moduleInstances[ImguiModule()]; }
+        im->releaseContext(ImGui::GetCurrentContext());
+    }
+}
+
+static size_t WriteCallback(void* contents, size_t size, size_t nmemb, std::ostream* userp) 
+{
+    userp->write(static_cast<char*>(contents), size * nmemb);
+    return size * nmemb;
+}
+
+void Ollama::onLoad() {
+    
+    ImguiModule* im; { im = (ImguiModule*) moduleInstances[ImguiModule()]; }
+
+    if(!im) {
+        std::cout << "No ImguiModule for Ollama!\n";
+        std::abort();
+    }
+
+    ImGui::SetCurrentContext(im->aquireContext());
+
+    curl_global_init(CURL_GLOBAL_ALL);
+    curl = curl_easy_init();
+
+}
+
+void Ollama::run() {
+
+    ImGuiIO& io = ImGui::GetIO();
+
+    static std::string s, url = "https://ollama.blunkall.us", response = "";
+
+    static nlohmann::json sendObj;
+
+    static std::ostringstream oss;
+
+    static nlohmann::json jsonObj;
+
+    static std::future<CURLcode> result;
+
+    static bool inFlight = false;
+
+    ImGui::Begin("Ollama Module");
+
+    ImGui::InputText("url: ", &url);
+
+    static const std::string replace = "\\n";
+    static const std::string replace_by = "\n";
+    static std::string print = jsonObj["response"].dump(); print = jsonObj["response"].dump();
+    static auto pos = print.find(replace); pos = print.find(replace);
+
+    while (pos != std::string::npos) {
+        // Replace the substring with the specified string
+        print.replace(pos, replace.size(), replace_by);
+
+        // Find the next occurrence of the substring
+        pos = print.find(replace,
+                pos + replace_by.size());
+    }
+    print = "\n\n\n" + print + "\n\n\n";
+    ImGui::TextWrapped("%s", print.c_str());
+
+    ImGui::InputTextMultiline("prompt: ", &s);
+
+    if(ImGui::Button("send")) {
+        sendObj["model"] = "llama3.2";
+        sendObj["stream"] = false;
+        sendObj["prompt"] = s;
+        curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, WriteCallback);
+        curl_easy_setopt(curl, CURLOPT_URL, (url + "/api/generate").c_str());
+        curl_easy_setopt(curl, CURLOPT_WRITEDATA, &oss);
+        //curl_easy_setopt(curl, CURLOPT_SSL_VERIFYPEER, 1L);
+        //curl_easy_setopt(curl, CURLOPT_SSL_VERIFYHOST, 2L);
+        curl_easy_setopt(curl, CURLOPT_POST, 1L);
+        curl_easy_setopt(curl, CURLOPT_COPYPOSTFIELDS, sendObj.dump().c_str());
+        result = std::async(std::launch::async, curl_easy_perform, curl);
+        inFlight = true;
+    }
+
+    ImGui::Text("ms per frame: %f", 1000 / io.Framerate);
+
+    ImGui::End();
+
+    if(curl) {
+    }
+
+    if(inFlight && result.wait_for(std::chrono::seconds(0)) == std::future_status::ready) {
+        CURLcode code = result.get();
+        response = oss.str();
+        oss.str("");
+        
+
+        if(code != CURLE_OK) {
+            std::cerr << "curl_easy_perform() failed!: " << curl_easy_strerror(code) << std::endl;
+            app->emitEvent(new Archimedes::DoUnloadModuleEvent(name));
+            return;
+        } else {
+            jsonObj = nlohmann::json::parse(response);
+
+            std::cout << "Full json object:\n" << jsonObj.dump() << std::endl;
+        }
+        inFlight = false;
+    }
+}