add colors

app.py

@@ -58,10 +58,18 @@ def sanitize_dimention(dim):
     return out
 
 
-def create_row(dim):
+def create_row(dim,is_dim=None,checks=None):
     out = "| "
-    for i in dim:
-        out = out + str(i) + " | "
+    n_dim = len(dim)
+    for i in range(n_dim):
+        if (is_dim ==1 and i != n_dim-1) or (is_dim ==2 and i ==n_dim-1):
+            color = "green" if checks[i] == "V" else "red"
+            out += f"<strong style='color: {color}'> {dim[i]} </strong>| "
+        elif (is_dim ==1 and i == n_dim-1) or (is_dim ==2 and i ==n_dim-2):
+            color = "blue" if checks[i] == "V" else "red"
+            out += f"<strong style='color: {color}'> {dim[i]} </strong>| "
+        else : 
+            out+= f"{dim[i]} | "
     return out + "\n"
 
 
@@ -78,9 +86,16 @@ def generate_table(dim1, dim2, checks=None):
     n_dim = len(dim1)
     table = create_header(n_dim, checks)
     # tensor 1
-    table += create_row(dim1)
+    if not checks : 
+        table += create_row(dim1)
+    else : 
+        table += create_row(dim1,1,checks)
+
     # tensor 2
-    table += create_row(dim2)
+    if not checks : 
+        table += create_row(dim2)
+    else : 
+        table += create_row(dim2,2,checks)
     return table