add examples

app.py

@@ -23,6 +23,7 @@ for i, j in zip(t1, t2):
 ```
 """
 
+
 def generate_example(dim1: list, dim2: list):
     n1 = 1
     n2 = 1
@@ -64,35 +65,35 @@ def sanitize_dimention(dim):
     return out
 
 
-def create_row(dim,is_dim=None,checks=None,version=1):
+def create_row(dim, is_dim=None, checks=None, version=1):
     out = "| "
     n_dim = len(dim)
     for i in range(n_dim):
-        if version == 1 :
+        if version == 1:
             # infered last dims
-            if (is_dim ==1 and i == n_dim-2) or (is_dim ==2 and i ==n_dim-1):
+            if (is_dim == 1 and i == n_dim - 2) or (is_dim == 2 and i == n_dim - 1):
                 color = "green"
                 out += f"<strong style='color: {color}'> {dim[i]} </strong>| "
             # check every normal dimension
-            elif (is_dim ==1 and i != n_dim-1) or (is_dim ==2 and i ==n_dim-1):
+            elif (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>| "
             # checks last 2 dims
-            elif (is_dim ==1 and i == n_dim-1) or (is_dim ==2 and i ==n_dim-2):
+            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 "yellow"
                 out += f"<strong style='color: {color}'> {dim[i]} </strong>| "
             # when using this function without checks
-            else : 
-                out+= f"{dim[i]} | "
-        if version == 2 : 
-            if (is_dim == 1 and i != n_dim-1) : 
+            else:
+                out += f"{dim[i]} | "
+        if version == 2:
+            if is_dim == 1 and i != n_dim - 1:
                 out += f"<strong style='color: green'> {dim[i]} </strong>| "
-            elif i == n_dim-1 :
+            elif i == n_dim - 1:
                 color = "blue" if checks[i] == "V" else "yellow"
                 out += f"<strong style='color: {color}'> {dim[i]} </strong>| "
-            else : 
+            else:
                 out += f"{dim[i]} | "
-            
+
     return out + "\n"
 
 
@@ -105,20 +106,20 @@ def create_header(n_dim, checks=None):
     return out
 
 
-def generate_table(dim1, dim2, checks=None,version=1):
+def generate_table(dim1, dim2, checks=None, version=1):
     n_dim = len(dim1)
     table = create_header(n_dim, checks)
     # tensor 1
-    if not checks : 
+    if not checks:
         table += create_row(dim1)
-    else : 
-        table += create_row(dim1,1,checks,version)
+    else:
+        table += create_row(dim1, 1, checks, version)
 
     # tensor 2
-    if not checks : 
+    if not checks:
         table += create_row(dim2)
-    else : 
-        table += create_row(dim2,2,checks,version)
+    else:
+        table += create_row(dim2, 2, checks, version)
     return table
 
 
@@ -137,63 +138,71 @@ def alignment_and_fill_with_ones(dim1, dim2):
         dim2 = placeholder
     return dim1, dim2
 
-def check_validity(dim1,dim2):
+
+def check_validity(dim1, dim2):
     out = []
-    for i in range(len(dim1)-2):
+    for i in range(len(dim1) - 2):
         if dim1[i] == dim2[i]:
             out.append("V")
-        else : 
+        else:
             out.append("X")
     # final dims
     if dim1[-1] == dim2[-2]:
-        out.extend(["V","V"])
-    else : 
-        out.extend(["X","X"])
+        out.extend(["V", "V"])
+    else:
+        out.extend(["X", "X"])
     return out
 
 
-def substitute_ones_with_concat(dim1,dim2,version = 1):
-    n = len(dim1)-2 if version ==1 else len(dim1)-1 
+def substitute_ones_with_concat(dim1, dim2, version=1):
+    n = len(dim1) - 2 if version == 1 else len(dim1) - 1
     for i in range(n):
         dim1[i] = dim2[i] if dim1[i] == 1 else dim1[i]
         dim2[i] = dim1[i] if dim2[i] == 1 else dim2[i]
     return dim1, dim2
 
+
 def predict(dim1, dim2):
     dim1 = sanitize_dimention(dim1)
     dim2 = sanitize_dimention(dim2)
-    n1 , n2 = len(dim1) , len(dim2)
+    n1, n2 = len(dim1), len(dim2)
     dim1, dim2, out = generate_example(dim1, dim2)
-    # TODO 
-    if n1 >1 and n2 > 1 : 
-        # Table 1 
+    # TODO
+    if n1 > 1 and n2 > 1:
+        # Table 1
         dim1, dim2 = alignment_and_fill_with_ones(dim1, dim2)
         table1 = generate_table(dim1, dim2)
-        # Table 2 
-        dim1, dim2 = substitute_ones_with_concat(dim1,dim2)
+        # Table 2
+        dim1, dim2 = substitute_ones_with_concat(dim1, dim2)
         table2 = generate_table(dim1, dim2)
-        # Table 3 
-        checks = check_validity(dim1,dim2)
-        table3 = generate_table(dim1,dim2,checks)
+        # Table 3
+        checks = check_validity(dim1, dim2)
+        table3 = generate_table(dim1, dim2, checks)
 
         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 += (
+            "\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 (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 : 
+        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 += "\n# Final dimension\n"
-            out+="as highlighted in <strong style='color:green'> green </strong> \n\n"
-            out+= f"`output.shape = {dim1}`"
+            out += "as highlighted in <strong style='color:green'> green </strong> \n\n"
+            out += f"`output.shape = {dim1}`"
     # case single dims
-    elif n1 == 1 and n2  == 1 :
+    elif n1 == 1 and n2 == 1:
         out += "# Single Dimensional Cases\n"
         out += "When both matricies have only single dims they should both have the same number of values in the first dimension\n"
         out += "meaning that `t1.shape == t2.shape`\n"
         out += "the output is a single value, think : \n"
         out += matrix_loop
-    else : 
+    else:
         out += "# One of the dimensions is a single dimension\n"
         out += "In this case we need to assert that the last dimension of `t1` "
         out += "is equal to the last dimension of `t2`\n"
@@ -204,20 +213,18 @@ def predict(dim1, dim2):
         out += table
         out += "\n# Step2 (susbtitute columns that have 1 with concat)\n"
         out += ""
-        dim1, dim2 = substitute_ones_with_concat(dim1,dim2,2)
-        checks = ["V"] * (len(dim1)-1)
-        if dim1[-1] == dim2[-1] :
+        dim1, dim2 = substitute_ones_with_concat(dim1, dim2, 2)
+        checks = ["V"] * (len(dim1) - 1)
+        if dim1[-1] == dim2[-1]:
             checks.append("V")
-        else : 
+        else:
             checks.append("X")
-        table = generate_table(dim1, dim2,checks,2)
-        out+= table
-        if "X" not in checks : 
+        table = generate_table(dim1, dim2, checks, 2)
+        out += table
+        if "X" not in checks:
             out += "\n#Final dimension"
             out += "The final dimension is everything colored in <strong style='color:green'> green </strong> \n"
             out += f"\nfinal dimension = `{dim1[:-1]}` "
-        
-        
 
     return out
 
@@ -226,7 +233,14 @@ demo = gr.Interface(
     predict,
     inputs=["text", "text"],
     outputs=["markdown"],
-    examples=[["9,2,1,3,3", "5,3,7"], ["7,4,2,3", "5,2,7"]],
+    examples=[
+        ["9,2,1,3,3", "5,3,7"],
+        ["7,4,2,3", "5,2,7"],
+        ["4,5,6,7", "7"],
+        ["7,5,3", "4"],
+        ["5", "5"],
+        ["8", "2"],
+    ],
 )
 
 demo.launch(debug=True)