can save animated gifs

Author: LostRuins

expose.h

@@ -204,6 +204,7 @@ struct sd_generation_inputs
     const char * sample_method = nullptr;
     const int clip_skip = -1;
     const int vid_req_frames = 1;
+    const int vid_req_avi = 0;
 };
 struct sd_generation_outputs
 {

koboldcpp.py

@@ -317,7 +317,8 @@ class sd_generation_inputs(ctypes.Structure):
                 ("seed", ctypes.c_int),
                 ("sample_method", ctypes.c_char_p),
                 ("clip_skip", ctypes.c_int),
-                ("vid_req_frames", ctypes.c_int)]
+                ("vid_req_frames", ctypes.c_int),
+                ("vid_req_avi", ctypes.c_int)]
 
 class sd_generation_outputs(ctypes.Structure):
     _fields_ = [("status", ctypes.c_int),
@@ -1824,6 +1825,7 @@ def sd_generate(genparams):
     clip_skip = tryparseint(genparams.get("clip_skip", -1),-1)
     vid_req_frames = tryparseint(genparams.get("frames", 1),1)
     vid_req_frames = 1 if (not vid_req_frames or vid_req_frames < 1) else vid_req_frames
+    vid_req_avi = 1 if genparams.get("avi_video", False) else 0
     extra_images_arr = genparams.get("extra_images", [])
     extra_images_arr = ([] if not extra_images_arr else extra_images_arr)
     extra_images_arr = [img for img in extra_images_arr if img not in (None, "")]
@@ -1856,6 +1858,7 @@ def sd_generate(genparams):
     inputs.sample_method = sample_method.lower().encode("UTF-8")
     inputs.clip_skip = clip_skip
     inputs.vid_req_frames = vid_req_frames
+    inputs.vid_req_avi = vid_req_avi
     ret = handle.sd_generate(inputs)
     outstr = ""
     if ret.status==1:

otherarch/sdcpp/avi_writer.h

@@ -7,6 +7,7 @@
 #include <string.h>
 
 #include "stable-diffusion.h"
+#include "./gif.h"   // charlietangora/gif-h
 
 #ifndef INCLUDE_STB_IMAGE_WRITE_H
 #include "stb_image_write.h"
@@ -390,4 +391,107 @@ int create_mjpg_avi_membuf_from_sd_images(sd_image_t* images, int num_images, in
     return 0;
 }
 
+/// kcpp gif writer
+
+// ---------------- Helper: create_gif_buf_from_sd_images ----------------
+// Builds a GIF in memory from an array of sd_image_t. Returns 0 on success, -1 on failure.
+// Caller must free(*out_data) when done.
+int create_gif_buf_from_sd_images(sd_image_t* images, int num_images, int fps, int quality,  uint8_t** out_data, size_t *out_len)
+{
+    if(!images || num_images <= 0 || !out_data || !out_len) return -1;
+
+    // basic parameter heuristics
+    if(fps <= 0) fps = 16;
+    uint32_t delay = (uint32_t)(100 / fps); // hundredths of a second per frame
+
+    // map quality [1..100] to bitDepth and dithering
+    if(quality < 1) quality = 1;
+    if(quality > 100) quality = 100;
+    int bitDepth = 8;
+    bool dither = false;
+    // if(quality >= 80) { bitDepth = 8; dither = false; }
+    // else if(quality >= 50) { bitDepth = 6; dither = false; }
+    // else { bitDepth = 5; dither = true; }
+
+    // assume all images same size; use first
+    uint32_t width = images[0].width;
+    uint32_t height = images[0].height;
+
+    GifWriter gw;
+    memset(&gw, 0, sizeof(gw));
+
+    if(!GifBegin(&gw, width, height, delay, bitDepth, dither))
+    {
+        if(gw.oldImage) GIF_FREE(gw.oldImage);
+
+        fprintf(stderr, "Error: GifBegin failed.\n");
+        return -1;
+    }
+
+    // Feed frames
+    for (int i = 0; i < num_images; i++)
+    {
+        sd_image_t* img = &images[i];
+
+        if (img->width != width || img->height != height) {
+            fprintf(stderr, "Frame %d has mismatched dimensions.\n", i);
+            GifEnd(&gw);
+            memfile_free(&gw.mem);
+            return -1;
+        }
+
+        // gif-h expects 4 channels (RGBA) or 3 channels (RGB). It quantizes internally.
+        // If your images have 3 channels, that’s fine. If 4 channels, it also works.
+        int channels = img->channel;
+        if (channels != 3 && channels != 4) {
+            fprintf(stderr, "Unsupported channel count: %d\n", channels);
+            GifEnd(&gw);
+            memfile_free(&gw.mem);
+            return -1;
+        }
+
+        // gif-h requires 4 channels (RGBA). If you only have RGB, add opaque alpha.
+        uint8_t* frame_rgba = NULL;
+        if (channels == 3) {
+            frame_rgba = (uint8_t*)malloc(width * height * 4);
+            for (int p = 0; p < width * height; p++) {
+                frame_rgba[p*4+0] = img->data[p*3+0];
+                frame_rgba[p*4+1] = img->data[p*3+1];
+                frame_rgba[p*4+2] = img->data[p*3+2];
+                frame_rgba[p*4+3] = 255;
+            }
+        } else {
+            frame_rgba = img->data; // already RGBA
+        }
+
+        if(!GifWriteFrame(&gw, frame_rgba, width, height, delay, bitDepth, dither))
+        {
+            fprintf(stderr, "GIF Write Failed\n");
+            GifEnd(&gw);
+            memfile_free(&gw.mem);
+            return -1;
+        }
+
+        if (channels == 3) {
+            free(frame_rgba);
+        }
+    }
+
+    if(!GifEnd(&gw))
+    {
+        memfile_free(&gw.mem);
+        return -1;
+    }
+
+    uint8_t* buf = memfile_detach(&gw.mem, out_len);
+    if(!buf)
+    {
+        memfile_free(&gw.mem);
+        return -1;
+    }
+
+    *out_data = buf;
+    return 0;
+}
+
 #endif  // __AVI_WRITER_H__