gui.py

#!/usr/bin/env python3
import os
import json
import tkinter as tk
from tkinter import Listbox, Scrollbar, END, RIGHT, Y, LEFT, BOTH

from tkinterweb import HtmlFrame
from mistletoe import markdown
from mistletoe.contrib.pygments_renderer import PygmentsRenderer

from core import DATA_DIRECTORY


class JsonViewerApp(tk.Tk):
    def __init__(self):
        super().__init__()

        self.title("JSON Viewer")
        self.geometry("800x600")

        self.left_frame = tk.Frame(self)
        self.left_frame.pack(side=LEFT, fill=tk.Y)

        self.right_frame = tk.Frame(self)
        self.right_frame.pack(side=LEFT, fill=BOTH, expand=True)

        self.file_listbox = Listbox(self.left_frame, width=40)
        self.file_listbox.pack(side=LEFT, fill=tk.Y)
        self.scrollbar = Scrollbar(self.left_frame)
        self.scrollbar.pack(side=RIGHT, fill=Y)
        self.file_listbox.config(yscrollcommand=self.scrollbar.set)
        self.scrollbar.config(command=self.file_listbox.yview)

        self.file_content_text = HtmlFrame(self.right_frame, messages_enabled=False)
        self.file_content_text.pack(fill=BOTH, expand=True)

        # Load the list of JSON files
        self.load_json_files()

        # Bind the listbox selection event to a function
        self.file_listbox.bind("<<ListboxSelect>>", self.on_file_select)

        # Killing with CTRL+C in the console
        self.check()

    def load_json_files(self):
        """Load the list of JSON files from the hardcoded folder."""
        if os.path.exists(DATA_DIRECTORY):
            json_files = sorted(
                [f for f in os.listdir(DATA_DIRECTORY) if f.endswith(".json")]
            )
            for file in json_files:
                self.file_listbox.insert(END, file)
        else:
            self.file_listbox.insert(END, "Folder not found")

    def on_file_select(self, event):
        """Display the content of the selected JSON file."""
        # Get the selected file name
        selected_index = self.file_listbox.curselection()
        if selected_index:
            selected_file = self.file_listbox.get(selected_index)

            # Construct the full file path
            file_path = os.path.join(DATA_DIRECTORY, selected_file)

            with open(file_path, "r") as file:
                file_content = json.load(file)
                formatted_content = markdown(
                    format_json(file_content), CustomPygmentsRenderer
                )
                self.file_content_text.load_html(formatted_content)

    # context switch to allow code to check CTRL+C in console
    def check(self):
        self.after(250, self.check)


def format_json(j):
    return "\n\n".join([f"**{m['role'].upper()}:**\n\n{m['content']}" for m in j])


class CustomPygmentsRenderer(PygmentsRenderer):
    def __init__(self):
        super().__init__()
        self.formatter.nobackground = True
        self.formatter.prestyles = "font-family: monospace, monospace; font-weight: 500"


if __name__ == "__main__":
    app = JsonViewerApp()
    app.mainloop()