From 538998b43b87678cb6b7ba9c401404401b05629f Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Sat, 7 Jan 2023 01:20:10 -0300
Subject: [PATCH] Fix a bug with the greentexts

---
 html_generator.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/html_generator.py b/html_generator.py
index 665f2e7..e6c9e85 100644
--- a/html_generator.py
+++ b/html_generator.py
@@ -148,7 +148,8 @@ def generate_html(f):
     output += '</div>'
     output = output.split('\n')
     for i in range(len(output)):
-        output[i] = re.sub('^(&gt;[^\n]*(<br>|</div>))', '<span class="greentext">\\1</span>\n', output[i])
+        output[i] = re.sub(r'^(&gt;(.*?)(<br>|</div>))', r'<span class="greentext">\1</span>', output[i])
+        output[i] = re.sub(r'^<blockquote class="message">(&gt;(.*?)(<br>|</div>))', r'<blockquote class="message"><span class="greentext">\1</span>', output[i])
     output = '\n'.join(output)
 
     return output