Update app.py

app.py

@@ -108,13 +108,16 @@ class Measurer:
                 map_dist = self.map_dist_func[l2][l1]
         else:
             map_dist = self.map_dist_func[l2][l1]
-        if l1 in self.asp_dist_func:
-            if l2 in self.asp_dist_func[l1]:
-                asp_dist = self.asp_dist_func[l1][l2]
+        try:
+            if l1 in self.asp_dist_func:
+                if l2 in self.asp_dist_func[l1]:
+                    asp_dist = self.asp_dist_func[l1][l2]
+                else:
+                    asp_dist = self.asp_dist_func[l2][l1]
             else:
                 asp_dist = self.asp_dist_func[l2][l1]
-        else:
-            asp_dist = self.asp_dist_func[l2][l1]
+        except KeyError:
+            asp_dist = tree_dist  # dirty hack, but like 4 codes are not part of phonepiece
         if l1 in self.learned_dist_func:
             if l2 in self.learned_dist_func[l1]:
                 learned_dist = self.learned_dist_func[l1][l2]