fix double linebreak issue in jinja rendering #694 (#719)

* fix double linebreak issue in jinja rendering * make style * more simply replacing all `\n`

Author: VictorSanh

promptsource/app.py

@@ -82,12 +82,21 @@ def reset_template_state():
  "<style>" + HtmlFormatter(style="friendly").get_style_defs(".highlight") + "</style>", unsafe_allow_html=True
 )
 
-WIDTH = 80
+WIDTH = 140
 
 
 def show_jinja(t, width=WIDTH):
+ def replace_linebreaks(t):
+ """
+ st.write does not handle double breaklines very well. When it encounters `\n\n`, it exit the curent <div> block.
+ Explicitely replacing all `\n` with their html equivalent to bypass this issue.
+ Also stripping the trailing `\n` first.
+ """
+ return t.strip("\n").replace("\n", "<br/>")
+
  wrap = textwrap.fill(t, width=width, replace_whitespace=False)
  out = highlight(wrap, DjangoLexer(), HtmlFormatter())
+ out = replace_linebreaks(out)
  st.write(out, unsafe_allow_html=True)