final comment shape

app.py

@@ -66,7 +66,7 @@ def create_row(dim,is_dim=None,checks=None):
             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"
+            color = "blue" if checks[i] == "V" else "yellow"
             out += f"<strong style='color: {color}'> {dim[i]} </strong>| "
         else : 
             out+= f"{dim[i]} | "
@@ -157,8 +157,11 @@ def predict(dim1, dim2):
     out += "\n# Step1 (alignment and pre_append with ones)\n" + table1
     out += "\n# Step2 (susbtitute columns that have 1 with concat)\nexcept for last 2 dimensions\n" + table2
     out += "\n# Step3 (check if matrix multiplication is valid)\n" 
-    out += "* last dimension of dim1 should equal before last dimension of dim2\n"
-    out += "* all the other dimensions should be equal to one another\n\n" + table3
+    out += "* last dimension of dim1 should equal before last dimension of dim2 (blue or yellow colors)\n"
+    out += "* all the other dimensions should be equal to one another (green or red colors)\n\n" + table3
+    if "X" not in checks : 
+        dim1[-1] = dim2[-1]
+        out += f"\n# Final dimension\n `output.shape = {dim1}`"
     return out