rescribe: Ensure temporary tesseract dir is removed in gui mode too

1 files changed, 5 insertions, 0 deletions

diff --git a/cmd/rescribe/main.go b/cmd/rescribe/main.go
index 8224830..229bf49 100644
--- a/cmd/rescribe/main.go
+++ b/cmd/rescribe/main.go
@@ -238,6 +238,11 @@ These training files are included in rescribe, and are always available:
 
 	if flag.NArg() < 1 || *usegui {
 		err := startGui(*verboselog, tessCommand, trainingName, *systess, tessdir)
+		err = os.RemoveAll(tessdir)
+		if err != nil {
+			log.Printf("Error removing tesseract directory %s: %v", tessdir, err)
+		}
+
 		if err != nil {
 			log.Fatalln("Error in gui:", err)
 		}