Mitigate rendering issues when models output a newline

Author: djcopley

src/shelloracle/providers/__init__.py

@@ -11,8 +11,8 @@
     "on interpreting the requirements and translating them into a single, executable Bash command. "
     "Ensure accuracy and relevance to the user's description. The output should be a valid Bash "
     "command that directly aligns with the user's intent, ready for execution in a command-line "
-    "environment. Output nothing except for the command. No code block, no English explanation, "
-    "no start/end tags."
+    "environment. Do not output anything except for the command. No code block, no English explanation, "
+    "no newlines, and no start/end tags."
 )
 
 

src/shelloracle/shelloracle.py

@@ -61,6 +61,8 @@ async def shelloracle() -> None:
     shell_command = ""
     with create_app_session_from_tty(), patch_stdout(raw=True), yaspin() as sp:
         async for token in provider.generate(prompt):
+            # some models may erroneously return a newline, which causes issues with the status spinner
+            token = token.replace("\n", "")
             shell_command += token
             sp.text = shell_command
     sys.stdout.write(shell_command)