From 7d9c0d0c8a627d6c6c1aeaeef4f5786ab8d4e780 Mon Sep 17 00:00:00 2001
From: callebtc <93376500+callebtc@users.noreply.github.com>
Date: Tue, 30 Aug 2022 18:56:05 +0200
Subject: [PATCH] retry

---
 internal/telegram/generate.go |  3 ++-
 internal/telegram/handler.go  | 29 +++++++++++++++--------------
 2 files changed, 17 insertions(+), 15 deletions(-)

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 <prompt>
 // 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,