From 5796b2c30bc6277800371accfe179c45d30ce8fe Mon Sep 17 00:00:00 2001
From: AnushaTiwari5 <atiwari3@albany.edu>
Date: Mon, 27 Nov 2023 21:41:22 -0500
Subject: [PATCH] changing theme redraws based on mode

---
 src/Themes.ts | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/Themes.ts b/src/Themes.ts
index 425f1268..2c490193 100644
--- a/src/Themes.ts
+++ b/src/Themes.ts
@@ -4,7 +4,7 @@
  * for the application from the CSS themes
  */
 import {treeContext} from "./treeContext";
-import {redrawTree} from "./DrawModes/DrawUtils";
+import {redrawProof, redrawTree} from "./DrawModes/DrawUtils";
 
 const themeSelector: HTMLSelectElement = <HTMLSelectElement>document.getElementById("theme-select");
 
@@ -36,7 +36,7 @@ function setTheme() {
         legalColorStr = cssVar("--good-placement");
         illegalColorStr = cssVar("--bad-placement");
         placedColorStr = cssVar("--canvas-items");
-        redrawTree(treeContext.tree);
+        treeContext.modeState === "Draw" ? redrawTree(treeContext.tree) : redrawProof();
     });
 }