From 14193592bddb0896caeccbe9e669894e85ecabc8 Mon Sep 17 00:00:00 2001
From: e-gun <egun@antisigma.classics.utoronto.ca>
Date: Wed, 28 Feb 2024 16:48:27 -0500
Subject: [PATCH] this one can cancel `ldaplot()`...

---
 go.mod            |  2 +-
 vectorquerylda.go | 16 +++++++---------
 2 files changed, 8 insertions(+), 10 deletions(-)

diff --git a/go.mod b/go.mod
index e4d5947..d49719d 100644
--- a/go.mod
+++ b/go.mod
@@ -4,7 +4,7 @@ go 1.21
 
 require (
 	github.com/e-gun/nlp v0.0.2
-	github.com/e-gun/tsnemp v0.0.0
+	github.com/e-gun/tsnemp v0.1.7
 	github.com/e-gun/wego v0.1.0
 	github.com/go-echarts/go-echarts/v2 v2.3.3
 	github.com/google/uuid v1.6.0
diff --git a/vectorquerylda.go b/vectorquerylda.go
index fb93315..dbbfeb4 100644
--- a/vectorquerylda.go
+++ b/vectorquerylda.go
@@ -6,6 +6,7 @@
 package main
 
 import (
+	"context"
 	"fmt"
 	"github.com/e-gun/nlp"
 	"github.com/labstack/echo/v4"
@@ -77,9 +78,9 @@ func (b *BagWithLocus) GetWL() {
 func LDASearch(c echo.Context, srch SearchStruct) error {
 	const (
 		LDAMSG = `Building LDA model for the current selections`
-		ESM1   = "preparing the text for modeling"
+		ESM1   = "Preparing the text for modeling"
 		ESM2   = "Building topic models"
-		ESM3   = "Using t-Distributed Stochastic Neighbor Embedding to build graph (please be patient...)"
+		ESM3   = "Building the graph (please be patient this can be very slow...)"
 	)
 	c.Response().After(func() { messenger.LogPaths("LDASearch()") })
 
@@ -137,7 +138,7 @@ func LDASearch(c echo.Context, srch SearchStruct) error {
 	var img string
 	if se.LDAgraph || srch.ID == "ldamodelbot()" {
 		WSInfo.UpdateSummMsg <- WSSIKVs{vs.WSID, fmt.Sprintf(ESM3)}
-		img = ldaplot(se.LDA2D, ntopics, incl, se.VecTextPrep, dot, bags)
+		img = ldaplot(vs.Context, se.LDA2D, ntopics, incl, se.VecTextPrep, dot, bags)
 	}
 
 	soj := SearchOutputJSON{
@@ -629,10 +630,7 @@ func ldadocbyweight(ntopics int, docsOverTopics mat.Matrix) []float64 {
 // see https://pkg.go.dev/gonum.org/v1/gonum/mat@v0.12.0#pkg-index
 
 // ldaplot - plot the lda results
-func ldaplot(graph2d bool, ntopics int, incl string, bagger string, docsOverTopics mat.Matrix, bags []BagWithLocus) string {
-	// m := mat.NewDense()
-	// func NewDense(r int, c int, data []float64) *Dense
-
+func ldaplot(ctx context.Context, graph2d bool, ntopics int, incl string, bagger string, docsOverTopics mat.Matrix, bags []BagWithLocus) string {
 	const (
 		PERPLEX = 150 // default 300
 		LEARNRT = 100 // default 100
@@ -693,12 +691,12 @@ func ldaplot(graph2d bool, ntopics int, incl string, bagger string, docsOverTopi
 	if graph2d && graph {
 		// t := NewTSNE(2, PERPLEX, LEARNRT, MAXITER, VERBOSE)
 		t := tsnemp.NewMPTSNE(Config.WorkerCount, 2, PERPLEX, LEARNRT, MAXITER, VERBOSE)
-		t.EmbedData(wv, nil)
+		t.EmbedDataWithCtx(ctx, wv, nil)
 		htmlandjs = lda2dscatter(ntopics, incl, bagger, t.Y, Y, bags)
 	} else if graph {
 		// 3d
 		nd := tsnemp.NewMPTSNE(Config.WorkerCount, 3, PERPLEX, LEARNRT, MAXITER, VERBOSE)
-		nd.EmbedData(wv, nil)
+		nd.EmbedDataWithCtx(ctx, wv, nil)
 		htmlandjs = lda3dscatter(ntopics, incl, bagger, nd.Y, Y, bags)
 	} else {
 		p := message.NewPrinter(language.English)