diff options
Diffstat (limited to 'cmd')
| -rw-r--r-- | cmd/xedn/stablediffusion.go | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/cmd/xedn/stablediffusion.go b/cmd/xedn/stablediffusion.go index 6bc18c0..0b8c3cf 100644 --- a/cmd/xedn/stablediffusion.go +++ b/cmd/xedn/stablediffusion.go @@ -3,6 +3,7 @@ package main import ( "bytes" "context" + "expvar" "fmt" "image" "image/jpeg" @@ -20,6 +21,11 @@ import ( "within.website/x/internal/stablediffusion" ) +var ( + stableDiffusionHits = expvar.NewInt("xedn_avatar_hits") + stableDiffusionCreation = expvar.NewInt("xedn_avatar_creation") +) + type StableDiffusion struct { client *stablediffusion.Client db *bbolt.DB @@ -55,6 +61,8 @@ func (sd *StableDiffusion) RenderImage(ctx context.Context, w http.ResponseWrite return nil, err } + stableDiffusionCreation.Add(1) + img, _, err := image.Decode(bytes.NewBuffer(imgs.Images[0])) if err != nil { return nil, err @@ -140,6 +148,8 @@ func (sd *StableDiffusion) ServeHTTP(w http.ResponseWriter, r *http.Request) { w.Write(data) } + stableDiffusionHits.Add(1) + return nil }) |