Update app.py

app.py

@@ -167,13 +167,13 @@ import html as py_html
 from streamlit.components.v1 import html as st_html
 
 def display_dialogue_with_numbers(text: str, title="Dialogue"):
-    with st.expander(f"{title} (line numbers)", expanded=True):
+    with st.container(f"{title} (line numbers)", expanded=True):
         # optional: wrap long lines instead of horizontal scrolling
         st.markdown(
             "<style>.stCode pre {white-space: pre-wrap !important; word-break: break-word !important;}</style>",
             unsafe_allow_html=True,
         )
-        st.code(text, language="text", line_numbers=True)
+        st.code(text, language="markdown", line_numbers=True)
 
 def markdown_with_line_numbers(text: str, height: int = 500):
     # Number *source lines* (not wrapped visual lines)