From 800d1b7792118a9c225fc81a2a71305b6f3c8337 Mon Sep 17 00:00:00 2001 From: Harivansh Rathi Date: Sun, 14 Dec 2025 17:26:18 -0500 Subject: [PATCH] fix problem diff --- veetcode/app.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/veetcode/app.py b/veetcode/app.py index 07b5c64..d05fe2f 100644 --- a/veetcode/app.py +++ b/veetcode/app.py @@ -256,9 +256,14 @@ class ProblemListScreen(Screen): return for p in self.problems: - diff_char = {"easy": "Easy", "medium": "Medium", "hard": "Hard"}.get(p.difficulty, "?") - check = "✓ " if p.name in self.solved else " " - ol.add_option(Option(f"{check}[{diff_char}] {p.name}", id=p.name)) + diff_colors = { + "easy": "[green]Easy[/green]", + "medium": "[yellow]Medium[/yellow]", + "hard": "[red]Hard[/red]", + } + diff = diff_colors.get(p.difficulty, "?") + check = "[green]✓[/green] " if p.name in self.solved else " " + ol.add_option(Option(f"{check}{diff} {p.name}", id=p.name)) def action_rescan(self) -> None: self.load_problems()