using a new coordinate system with levels, rings and sectors

app.py

@@ -4,7 +4,8 @@ import os
 import numpy
 
 from sentence_transformers import SentenceTransformer
-sentence_model = SentenceTransformer("all-mpnet-base-v2") 
+sentence_model = SentenceTransformer("paraphrase-multilingual-mpnet-base-v2")
+#sentence_model = SentenceTransformer("all-mpnet-base-v2") 
 
 def getText( file ):
     return open( file ).read()
@@ -12,15 +13,15 @@ def getText( file ):
 def axis():
     global v, a, s
     doc = getText( './axis/v' )
-    embedding = sentence_model.encode( doc , show_progress_bar=False)
+    embedding = sentence_model.encode( doc , show_progress_bar=False, normalize_embeddings=True )
     torch.save( embedding , './embeds/v' )
     v = embedding
     doc = getText( './axis/a' )
-    embedding = sentence_model.encode( doc , show_progress_bar=False)
+    embedding = sentence_model.encode( doc , show_progress_bar=False, normalize_embeddings=True )
     torch.save( embedding , './embeds/a' )
     a = embedding
     doc = getText( './axis/s' )
-    embedding = sentence_model.encode( doc , show_progress_bar=False)
+    embedding = sentence_model.encode( doc , show_progress_bar=False, normalize_embeddings=True )
     torch.save( embedding , './embeds/s' )
     s = embedding
 
@@ -34,25 +35,23 @@ def zxy( doc ):
             v = torch.load( './embeds/v' )
             a = torch.load( './embeds/a' )
             s = torch.load( './embeds/s' )
-
-    embedding = sentence_model.encode( doc , show_progress_bar=False)
+    embedding = sentence_model.encode( doc , show_progress_bar=False , normalize_embeddings=True )
     # torch.save( embedding , './embeds/new' )
-
-    z = embedding @ s
-    x = embedding @ a
-    y = embedding @ v
-
-    if z < 0 : z = 0
-    if x < 0 : x = 0
-    if y < 0 : y = 0
-
+    z = ( 1.0 - embedding @ s ) / 2.0 # level 0 on top
+    x = ( 1.0 - embedding @ a ) / 2.0 # ring 0 in the middle
+    y = ( embedding @ v + 1.0 ) / 2.0 # sector 0 at twelve oclock
+    # add one to make positive / divide by two to make range 0-1
     scale = 1
-    axisrange = 44
-
-    zxy = [ numpy.rint( scale * axisrange * z ) , numpy.rint( scale * axisrange * x ) , numpy.rint( scale * axisrange * y ) ]
-
-    return "L" + str( int( zxy[ 0 ] ) ) + "/" + str( int( zxy[ 1 ] ) ) + "/" + str( int( zxy[ 2 ] ) )
-
-iface = gr.Interface(fn=zxy, inputs="text", outputs="text")
+    levels = 44
+    rings = 22
+    sectors = 44 * 4 
+    zxy = [ numpy.rint( scale * levels * z ) , numpy.rint( scale * rings * x ) , numpy.rint( scale * sectors * y ) ]
+    return f"""L{ str( int( zxy[ 0 ] ) ) }/{ str( int( zxy[ 1 ] ) ) }/{ str( int( zxy[ 2 ] ) ) }
+        es={ embedding @ s } ex={ embedding @ a } ey={ embedding @ v }
+        z={ z } x={ x } y={ y }
+        {scale} {levels} {rings} {sectors}
+        """
+
+#iface = gr.Interface(fn=zxy, inputs="text", outputs="text")
 #iface.launch()
-iface.launch(share=True)
+#iface.launch(share=True)  # make public