Update app.py

app.py

@@ -65,7 +65,7 @@ function_choices = ['Broadening Discussion',
 default_choices = function_choices
 
 fields: List[Field] = [
-    Field(name="patient", type="input_col", title="**Patient:** "),
+    Field(name="patient", type="input_col", title=" "),
     Field(type="expander", title="**Session Transcription:** *(expand)*", children=[
         Field(name="dialogue_name", type="input_col", title=""),
     ]),
@@ -167,7 +167,7 @@ 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.container(f"{title} (line numbers)", expanded=True):
+    with st.expander(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>",
@@ -175,26 +175,29 @@ def display_dialogue_with_numbers(text: str, title="Dialogue"):
         )
         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)
-    escaped = py_html.escape(text)
-    body = "".join(f"<span>{ln or '&nbsp;'}</span>" for ln in escaped.splitlines())
-
-    st_html(f"""
-<div class="ln"><pre>{body}</pre></div>
-<style>
-  .ln {{ counter-reset: line; font-family: ui-monospace, SFMono-Regular, Menlo, monospace;
-        font-size: 0.95rem; border: 1px solid #ddd; border-radius: .5rem;
-        padding: .75rem; background: #fff; max-height: 100%; overflow: auto; }}
-  .ln pre {{ white-space: pre-wrap; margin: 0; }}
-  .ln pre span {{ display: block; }}
-  .ln pre span::before {{
-    counter-increment: line;
-    content: counter(line);
-    display: inline-block; width: 4ch; margin-right: 1ch; text-align: right; opacity: .6;
-  }}
-</style>
-""", height=height, scrolling=True)
+def _enable_code_wrap_once():
+    if not st.session_state.get("_code_wrap_css_added"):
+        st.markdown(
+            "<style>div[data-testid='stCode'] pre {white-space: pre-wrap !important; word-break: break-word !important;}</style>",
+            unsafe_allow_html=True,
+        )
+        st.session_state["_code_wrap_css_added"] = True
+
+def display_dialogue_container(hf_path: str, title: str = "Dialogue", wrap: bool = True):
+    # read file from HF or local (uses your existing flags/objects)
+    if filesystem == 'hf':
+        with hf_fs.open(hf_path, "rb") as f:
+            txt = f.read().decode("utf-8")
+    else:
+        with open(hf_path, "r", encoding="utf-8") as f:
+            txt = f.read()
+
+    if wrap:
+        _enable_code_wrap_once()
+        
+    with st.container(border=True):
+        st.markdown(f"**{title}**")
+        st.code(txt, language="markdown", line_numbers=True)
 
 def display_image(image_path):
     with hf_fs.open(image_path) as f:
@@ -250,8 +253,10 @@ def show_field(f: Field, index: int, data_collected):
                     if f.name == 'image_name':
                         display_image(os.path.join(input_repo_path, 'images', value))
                     elif f.name == 'dialogue_name':
-                        display_dialogue(os.path.join(input_repo_path, 'dialogues', value))
-                        #display_dialogue(os.path.join(input_repo_path, value))
+                        #display_dialogue(os.path.join(input_repo_path, 'dialogues', value))
+                        display_dialogue_container(os.path.join(input_repo_path, 'dialogues', value), title="Dialogue Transcription"))
+                    elif f.name == 'patient':
+                        st.markdown(f"**Patient:**&nbsp;&nbsp;{value}")
                     else:
                         st.write(value)
             case 'markdown':