diff --git a/internal/telegram/generate.go b/internal/telegram/generate.go index 42688f2..2b20553 100644 --- a/internal/telegram/generate.go +++ b/internal/telegram/generate.go @@ -22,6 +22,7 @@ import ( // generateImages is called when the user enters /generate or /generate // asks the user for a prompt if not given func (bot *TipBot) generateImages(ctx intercept.Context) (intercept.Context, error) { + bot.anyTextHandler(ctx) user := LoadUser(ctx) if user.Wallet == nil { return ctx, fmt.Errorf("user has no wallet") @@ -141,7 +142,7 @@ func (bot *TipBot) generateDalleImages(event Event) { log.Errorf("rejected: %s", t.ID) } - fmt.Println("task still pending") + log.Debugln("task still pending") } // download the first generated image diff --git a/internal/telegram/handler.go b/internal/telegram/handler.go index 6c0b017..40faa00 100644 --- a/internal/telegram/handler.go +++ b/internal/telegram/handler.go @@ -99,20 +99,21 @@ func (bot TipBot) getHandler() []InterceptionWrapper { }, }, }, - // { - // Endpoints: []interface{}{"/generate"}, - // Handler: bot.generateImages, - // Interceptor: &Interceptor{ - // Before: []intercept.Func{ - // bot.localizerInterceptor, - // bot.loadUserInterceptor, - // bot.lockInterceptor, - // }, - // OnDefer: []intercept.Func{ - // bot.unlockInterceptor, - // }, - // }, - // }, + { + Endpoints: []interface{}{"/generate"}, + Handler: bot.generateImages, + Interceptor: &Interceptor{ + Before: []intercept.Func{ + bot.localizerInterceptor, + bot.logMessageInterceptor, + bot.loadUserInterceptor, + bot.lockInterceptor, + }, + OnDefer: []intercept.Func{ + bot.unlockInterceptor, + }, + }, + }, { Endpoints: []interface{}{"/tip", "/t", "/honk"}, Handler: bot.tipHandler,