Spinner in zsh script

Makefile

@@ -13,7 +13,7 @@ pull:
 test:
 	curl $(LOCALAI)/v1/chat/completions -H "Content-Type: application/json" -d '{"model": "gpt4all-j", "messages": [{"role": "user", "content": "How are you?"}],"temperature": 0.1 }'
 
-SILENT:
+.SILENT:
 get_models:
 	curl $(LOCALAI)/models/apply -H "Content-Type: application/json" -d '{"url": "github:go-skynet/model-gallery/gpt4all-j.yaml","name": "gpt4all-j"}'
 	rm -rf models/*.tmpl

zsh-local-ai/create_completion.py

@@ -2,7 +2,6 @@
 
 import subprocess
 import sys
-import time
 
 
 ai_url = 'http://localhost:8085/v1'
@@ -39,18 +38,8 @@
     shell=True,
 )
 
-spinner = ['-', '/', '|', '\\']
-i = 0
-while process.poll() is None:
-    sys.stdout.write('\r' + spinner[i % len(spinner)])
-    i += 1
-    time.sleep(0.1)
-
-
 stdout, stderr = process.communicate()
 
-sys.stdout.write('\r')
-
 # print the output
 if process.returncode == 0:
     completed_command = eval(stdout)['choices'][0]['message']['content']

zsh-local-ai/zsh-local-ai.plugin.zsh

@@ -2,12 +2,26 @@
 
 create_completion() {
     text=${BUFFER}
-    completion=$(echo -n "$text" | $ZSH_CUSTOM/plugins/zsh-local-ai/create_completion.py $CURSOR)
     text_before_cursor=${text:0:$CURSOR}
     text_after_cursor=${text:$CURSOR}
+    spinner="/-\|"
+    i=0
+    set +m
+    echo -ne "|"
+    while [ true ]
+    do
+        i=$(( (i+1) %4 ))
+        echo -ne "\b${spinner:$i:1}"
+        sleep .1
+    done &
+    spinner_pid=$!
+    completion=$(echo -n "$text" | $ZSH_CUSTOM/plugins/zsh-local-ai/create_completion.py $CURSOR)
+    kill $spinner_pid
+    wait $spinner_pid 2>/dev/null
+    echo -ne "\b"
     BUFFER="${text_before_cursor}${completion}${text_after_cursor}"
     prefix_and_completion="${text_before_cursor}${completion}"
     CURSOR=${#prefix_and_completion}
 }
 
-zle -N create_completion
+zle -N create_completion