fix handling of empty input #185

ku-nlp · Jul 28, 2023 · 3d33347 · 3d33347
1 parent 2bc13d8
commit 3d33347
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 7 deletions.
diff --git a/src/kwja/cli/cli.py b/src/kwja/cli/cli.py
@@ -372,7 +372,9 @@ def main(
     if input_text is not None:
         if input_text.strip() != "":
             print(processor.run(_split_into_documents(input_text)), end="")
-        processor.refresh()
+            processor.refresh()
+        else:
+            print("EOD")
         raise typer.Exit()
 
     # Interactive mode
@@ -385,11 +387,14 @@ def main(
         except EOFError:
             break
         if input_ == "EOD":
-            processor.refresh()
-            print(processor.run([input_text], interactive=True), end="")
-            if specified_tasks != ["typo"]:
-                print("EOD")  # To indicate the end of the output.
-            input_text = ""
+            if input_text.strip() != "":
+                processor.refresh()
+                print(processor.run([input_text], interactive=True), end="")
+                if specified_tasks != ["typo"]:
+                    print("EOD")  # To indicate the end of the output.
+                input_text = ""
+            else:
+                print("EOD")
         else:
             input_text += input_ + "\n"
 

diff --git a/tests/cli/test_cli.py b/tests/cli/test_cli.py
@@ -30,7 +30,7 @@ def test_text_input():
 @pytest.mark.parametrize(
     "text, output",
     [
-        ("", ""),
+        ("", "EOD\n"),
         # ("EOD", "EOD\nEOD\n"),  # TODO
         ("おはよう", "おはよう\nEOD\n"),
         ("おはよう．", "おはよう.\nEOD\n"),