Improve demo.py, add demo configuration

LyubomirT · Nov 26, 2023 · c751687 · c751687
1 parent 933f116
commit c751687
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 2 deletions.
diff --git a/ac/__init__.py b/ac/__init__.py
diff --git a/autocorrect.py → ac/autocorrect.py b/autocorrect.py → ac/autocorrect.py
diff --git a/demo.py b/demo.py
@@ -1,4 +1,13 @@
-from main import is_correct, get_similar
+from ac.autocorrect import is_correct, get_similar
+
+def load_config():
+    with open("demo_config", "r") as f:
+        config = f.read()
+    return config
+
+config = load_config()
+# showallsimilarities="True"
+showallsimilarities = config.split("showallsimilarities=\"")[1].split("\"")[0]
 
 def demo():
     while True:
@@ -7,9 +16,13 @@ def demo():
             print("Correct!")
         else:
             print("Incorrect!")
-            similar = get_similar(word, 0.7, chunks=20, upto=5)
+            similar = get_similar(word, 0.5, chunks=20, upto=5)
             if similar == None:
                 print("No similar words found.")
+            elif showallsimilarities == "True":
+                print("Did you mean any of the following?")
+                for w in similar:
+                    print("\"" + w + "\"")
             else:
                 similar = similar[0]
                 print("Did you mean \"" + similar + "\"?")

diff --git a/demo_config b/demo_config
@@ -0,0 +1 @@
+showallsimilarities="True"