Adding better console handling

src/display/log_visualizer.py

@@ -1,3 +1,4 @@
+from io import StringIO
 from pathlib import Path
 
 from bs4 import BeautifulSoup
@@ -12,7 +13,6 @@ proj_dir = Path(__name__).parent
 
 
 def log_file_to_html_string():
-    console = Console(record=True, width=150, style="#272822")
     with open(log_file, "rt") as f:
         # Seek to the end of the file minus 300 lines
         # Read the last 300 lines of the file
@@ -23,7 +23,8 @@ def log_file_to_html_string():
         output = "".join(reversed(lines))
         syntax = Syntax(output, "python", theme="monokai", word_wrap=True)
 
-    console.print(syntax);
+    console = Console(record=True, width=150, style="#272822", file=StringIO())
+    console.print(syntax)
     html_content = console.export_html(inline_styles=True)
 
     # Parse the HTML content using BeautifulSoup