update text color

app_text.py

@@ -1,7 +1,7 @@
 # %%
+import copy
 import os
 USE_HUGGINGFACE_ZEROGPU = os.getenv("USE_HUGGINGFACE_ZEROGPU", "False").lower() in ["true", "1", "yes"]
-#%%
 if USE_HUGGINGFACE_ZEROGPU:  # huggingface ZeroGPU, dynamic GPU allocation 
     try:
         import spaces
@@ -98,14 +98,18 @@ def make_plot(token_texts, rgb, num_lines=50, title=""):
     x_pos = 0.0
     max_word_length = max(len(word) for word in words)
     count = 0
+    prev_word = ""
     for word, color in zip(words, colors):
+        _word = copy.deepcopy(word)
         if '\n' in word:
             word = word.replace('\n', '')
+        if '\n' in prev_word:
             y_pos -= 0.025
             x_pos = 0.0
             count += 1
             if count >= num_lines:
                 break
+        prev_word = _word
         
         text_color = 'black' if sum(mcolors.hex2color(color)) > 1.3 else 'white'  # Choose text color based on background color
         # text_color = 'black'