Update app.py

app.py

@@ -101,13 +101,14 @@ def editTwoLetters(word, allowSwitches=True):
     return editTwoSet
 
 
-def getCorrections(word, probablity, vocabulary):
-    suggestion = []
-    n_best = []
+def getCorrections(word, probablity, vocabulary, max_suggestions=6):
+    suggestions = []
     suggestions = list((word in vocabulary and word) or editOneLetter(word).intersection(vocabulary) or editTwoLetters(
         word).intersection(vocabulary))
-    n_best = [[s, probablity[s]] for s in list(reversed(suggestions))]
-    return n_best
+    suggestions_with_prob = [[s, probablity[s]] for s in list(reversed(suggestions))]
+    suggestions_with_prob.sort(key=lambda x: x[1], reverse=True)
+    top_suggestions = suggestions_with_prob[:max_suggestions]
+    return top_suggestions
 
 
 @app.route('/')