Green-Sky/solanaceae_sdbot-webui
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

make more stuff configurable

Browse Source
This commit is contained in:
Green Sky 2024-04-07 12:24:29 +02:00
parent 3d9e963ca4
commit 5e96dfd82e
No known key found for this signature in database
2 changed files with 51 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
README.md Normal file
View File

@ -0,0 +1,29 @@
## Solanaceae extention and plugin to serve StableDiffusion
!! currently only works with automatic1111's api !!
### example config
```json
{
"PluginManager": {
"autoload": {
"entries": {
"./build/bin/libplugin_sdbot-webui.so": true
}
}
},
"SDBot": {
"server_host": "127.0.0.1",
"server_port": 8080,
"url_txt2img": "/sdapi/v1/txt2img",
"width": 512,
"height": 512,
"prompt_prefix": "<lora:lcm-lora-sdv1-5:1>",
"steps": 8,
"cfg_scale": 1.0,
"sampler_index": "LCM Test"
}
}
```

26
src/sd_bot.cpp
View File

@ -22,6 +22,15 @@ SDBot::SDBot(
_rng.seed(std::random_device{}());
_rng.discard(3137);
if (!_conf.has_string("SDBot", "server_host")) {
_conf.set("SDBot", "server_host", std::string_view{"127.0.0.1"});
}
if (!_conf.has_int("SDBot", "server_port")) {
_conf.set("SDBot", "server_port", int64_t(7860)); // automatic11 default
}
if (!_conf.has_string("SDBot", "url_txt2img")) {
_conf.set("SDBot", "url_txt2img", std::string_view{"/sdapi/v1/txt2img"}); // automatic11 default
}
if (!_conf.has_int("SDBot", "width")) {
_conf.set("SDBot", "width", int64_t(512));
}
@ -50,8 +59,11 @@ float SDBot::iterate(void) {
_current_task = task_id;
// TODO: reuse connection?
// TODO: read from config
_con = std::make_unique<happyhttp::Connection>("127.0.0.1", 7860);
const std::string server_host {_conf.get_string("SDBot", "server_host").value()};
_con = std::make_unique<happyhttp::Connection>(
server_host.c_str(),
_conf.get_int("SDBot", "server_port").value()
);
_con->setcallbacks(
+[](const happyhttp::Response* r, void* ud) { static_cast<SDBot*>(ud)->onHttpBegin(r); },
+[](const happyhttp::Response* r, void* ud, const uint8_t* data, int n) { static_cast<SDBot*>(ud)->onHttpData(r, data, n); },
@ -98,7 +110,8 @@ float SDBot::iterate(void) {
std::string body = j_body.dump();
try {
_con->request("POST", "/sdapi/v1/txt2img", headers, reinterpret_cast<const uint8_t*>(body.data()), body.size());
const std::string url {_conf.get_string("SDBot", "url_txt2img").value()};
_con->request("POST", url.c_str(), headers, reinterpret_cast<const uint8_t*>(body.data()), body.size());
} catch (const happyhttp::Wobbly& e) {
std::cerr << "SDB http request error: " << e.what() << "\n";
// cleanup
@ -209,7 +222,12 @@ void SDBot::onHttpComplete(const happyhttp::Response* r) {
//std::cout << std::string_view{reinterpret_cast<const char*>(_con_data.data()), _con_data.size()} << "\n";
// extract json result
const auto j = nlohmann::json::parse(std::string_view{reinterpret_cast<const char*>(_con_data.data()), _con_data.size()});
const auto j = nlohmann::json::parse(
std::string_view{reinterpret_cast<const char*>(_con_data.data()), _con_data.size()},
nullptr,
false
);
//std::cout << "json dump: " << j.dump() << "\n";
if (j.count("images") && !j.at("images").empty() && j.at("images").is_array()) {
for (const auto& i_j : j.at("images").items()) {