Don't render user message as if they are template strings

open-cogsci · Jan 20, 2024 · 3970d5a · 3970d5a
1 parent 7b15bdf
commit 3970d5a
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 4 deletions.
diff --git a/heymans/__init__.py b/heymans/__init__.py
@@ -1,3 +1,3 @@
 """AI-based chatbot that provides sensible answers based on documentation"""
 
-__version__ = '0.10.4'
+__version__ = '0.10.5'
diff --git a/heymans/routes/app.py b/heymans/routes/app.py
@@ -42,7 +42,8 @@ def chat_page():
             html_class = 'message-ai'
         else:
             html_body = '<p>' + utils.clean(f'You: {message}', 
-                                            escape_html=True) + '</p>'
+                                            escape_html=True,
+                                            render=False) + '</p>'
             html_class = 'message-user'
         if 'sources' in metadata:
             sources_div = '<div class="message-sources">'

diff --git a/heymans/utils.py b/heymans/utils.py
@@ -21,8 +21,9 @@ def md(text):
                                          TocExtension()])
 
 
-def clean(text, escape_html=True):
-    text = render_template_string(text)
+def clean(text, escape_html=True, render=True):
+    if render:
+        text = render_template_string(text)
     if escape_html:
         text = html.escape(text).replace('\n', '<br>\n')
     return text