Updating for better logging

src/visualize_logs.py

@@ -1,15 +1,17 @@
+from io import StringIO
+from pathlib import Path
+
 from bs4 import BeautifulSoup
 from rich.console import Console
 from rich.syntax import Syntax
-from pathlib import Path
 
 proj_dir = Path(__file__).parents[1]
 
+
 def log_file_to_html_string():
-    log_file = proj_dir/"mylog.log"
+    log_file = proj_dir / "mylog.log"
     num_lines_visualize = 50
 
-    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
@@ -20,7 +22,8 @@ def log_file_to_html_string():
         output = "".join(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