diff --git a/src/gui_2.py b/src/gui_2.py
index 0fac41a..68da9cd 100644
--- a/src/gui_2.py
+++ b/src/gui_2.py
@@ -110,6 +110,11 @@ class App:
   self.show_preset_manager_window = False
   self.show_tool_preset_manager_window = False
   self.show_persona_editor_window = False
+  self.show_text_viewer = False
+  self.text_viewer_title = ''
+  self.text_viewer_content = ''
+  self.text_viewer_type = 'text'
+  self.text_viewer_wrap = True
   self.ui_active_tool_preset = ""
   self.ui_active_bias_profile = ""
   self.ui_active_persona = ""
@@ -281,7 +286,8 @@ class App:
    f.write(data)
    # ---------------------------------------------------------------- helpers
 
- def _render_text_viewer(self, label: str, content: str) -> None:
+ def _render_text_viewer(self, label: str, content: str, text_type: str = 'text') -> None:
+  self.text_viewer_type = text_type
   if imgui.button("[+]##" + str(id(content))):
    self.show_text_viewer = True
    self.text_viewer_title = label