Merge pull request #101 from 0xacx/multiline-prompt-chat-mode

Multiline prompt chat mode

chatgpt.sh

@@ -12,7 +12,6 @@ CHATGPT_CYAN_LABEL="\033[36mchatgpt \033[0m"
 PROCESSING_LABEL="\n\033[90mProcessing... \033[0m\033[0K\r"
 OVERWRITE_PROCESSING_LINE="             \033[0K\r"
 
-
 if [[ -z "$OPENAI_KEY" ]]; then
 	echo "You need to set your OPENAI_KEY to use this script"
 	echo "You can set it temporarily by running this on your terminal: export OPENAI_KEY=YOUR_KEY_HERE"
@@ -37,15 +36,34 @@ Commands:
   *If a command modifies your file system or dowloads external files the script will show a warning before executing.
 
 Options:
-  -i, --init-prompt - Provide initial chat prompt to use in context
-  --init-prompt-from-file - Provide initial prompt from file
-  -p, --prompt - Provide prompt instead of starting chat
-  --prompt-from-file - Provide prompt from file
-  -t, --temperature - Temperature
-  --max-tokens - Max number of tokens
-  -m, --model - Model
-  -s, --size - Image size. (The sizes that are accepted by the OpenAI API are 256x256, 512x512, 1024x1024)
-  -c, --chat-context - For models that do not support chat context by default (all models except gpt-3.5-turbo and gpt-4), you can enable chat context, for the model to remember your previous questions and its previous answers. It also makes models aware of todays date and what data it was trained on.
+  -i, --init-prompt          Provide initial chat prompt to use in context
+
+  --init-prompt-from-file    Provide initial prompt from file
+
+  -p, --prompt               Provide prompt instead of starting chat
+
+  --prompt-from-file         Provide prompt from file
+
+  -b, --big-prompt           Allow multi-line prompts during chat mode
+
+  -t, --temperature          Temperature
+
+  --max-tokens               Max number of tokens
+
+  -l, --list                 List available openAI models
+
+  -m, --model                Model to use
+
+  -s, --size                 Image size. (The sizes that are accepted by the
+                             OpenAI API are 256x256, 512x512, 1024x1024)
+
+  -c, --chat-context         For models that do not support chat context by
+                             default (all models except gpt-3.5-turbo and
+                             gpt-4), you can enable chat context, for the
+                             model to remember your previous questions and
+                             its previous answers. It also makes models
+                             aware of todays date and what data it was trained
+                             on.
 
 EOF
 }
@@ -60,6 +78,17 @@ handle_error() {
 	fi
 }
 
+# request to openAI API models endpoint. Returns a list of models
+# takes no input parameters
+list_models() {
+	models_response=$(curl https://api.openai.com/v1/models \
+		-sS \
+		-H "Authorization: Bearer $OPENAI_KEY")
+	handle_error "$models_response"
+	models_data=$(echo $models_response | jq -r -C '.data[] | {id, owned_by, created}')
+	echo -e "$OVERWRITE_PROCESSING_LINE"
+	echo -e "${CHATGPT_CYAN_LABEL}This is a list of models currently available at OpenAI API:\n ${models_data}"
+}
 # request to OpenAI API completions endpoint function
 # $1 should be the request prompt
 request_to_completions() {
@@ -124,8 +153,8 @@ build_chat_context() {
 	fi
 }
 
-escape(){
-  echo "$1" | jq -Rrs 'tojson[1:-1]'
+escape() {
+	echo "$1" | jq -Rrs 'tojson[1:-1]'
 }
 
 # maintain chat context function for /completions (all models except
@@ -218,6 +247,10 @@ while [[ "$#" -gt 0 ]]; do
 		shift
 		shift
 		;;
+	-l | --list)
+		list_models
+		exit 0
+		;;
 	-m | --model)
 		MODEL="$2"
 		shift
@@ -228,6 +261,10 @@ while [[ "$#" -gt 0 ]]; do
 		shift
 		shift
 		;;
+	--multi-line-prompt)
+		MULTI_LINE_PROMPT=true
+		shift
+		;;
 	-c | --chat-context)
 		CONTEXT=true
 		shift
@@ -249,6 +286,13 @@ MAX_TOKENS=${MAX_TOKENS:-1024}
 MODEL=${MODEL:-gpt-3.5-turbo}
 SIZE=${SIZE:-512x512}
 CONTEXT=${CONTEXT:-false}
+MULTI_LINE_PROMPT=${MULTI_LINE_PROMPT:-false}
+
+# create our temp file for multi-line input
+if [ $MULTI_LINE_PROMPT = true ]; then
+	USER_INPUT_TEMP_FILE=$(mktemp)
+	trap 'rm -f ${USER_INPUT}' EXIT
+fi
 
 # create history file
 if [ ! -f ~/.chatgpt_history ]; then
@@ -273,9 +317,16 @@ fi
 while $running; do
 
 	if [ -z "$pipe_mode_prompt" ]; then
-		echo -e "\nEnter a prompt:"
-		read -e prompt
-		if [ "$prompt" != "exit" ] && [ "$prompt" != "q" ]; then
+		if [ $MULTI_LINE_PROMPT = true ]; then
+			echo -e "\nEnter a prompt: (Press Enter then Ctrl-D to send)"
+			cat > "${USER_INPUT_TEMP_FILE}"
+			input_from_temp_file=$(cat "${USER_INPUT_TEMP_FILE}")
+			prompt=$(escape "$input_from_temp_file")
+		else
+			echo -e "\nEnter a prompt:"
+			read -e prompt
+		fi
+		if [[ ! $prompt =~ ^(exit|q)$ ]]; then
 			echo -ne $PROCESSING_LABEL
 		fi
 	else
@@ -285,7 +336,7 @@ while $running; do
 		CHATGPT_CYAN_LABEL=""
 	fi
 
-	if [ "$prompt" == "exit" ] || [ "$prompt" == "q" ]; then
+	if [[ $prompt =~ ^(exit|q)$ ]]; then
 		running=false
 	elif [[ "$prompt" =~ ^image: ]]; then
 		request_to_image "$prompt"
@@ -312,13 +363,7 @@ while $running; do
 	elif [[ "$prompt" == "history" ]]; then
 		echo -e "\n$(cat ~/.chatgpt_history)"
 	elif [[ "$prompt" == "models" ]]; then
-		models_response=$(curl https://api.openai.com/v1/models \
-			-sS \
-			-H "Authorization: Bearer $OPENAI_KEY")
-		handle_error "$models_response"
-		models_data=$(echo $models_response | jq -r -C '.data[] | {id, owned_by, created}')
-		echo -e "$OVERWRITE_PROCESSING_LINE"
-		echo -e "${CHATGPT_CYAN_LABEL}This is a list of models currently available at OpenAI API:\n ${models_data}"
+		list_models
 	elif [[ "$prompt" =~ ^model: ]]; then
 		models_response=$(curl https://api.openai.com/v1/models \
 			-sS \
@@ -373,7 +418,6 @@ while $running; do
 		if command -v glow &>/dev/null; then
 			echo -e "${CHATGPT_CYAN_LABEL}"
 			echo "${response_data}" | glow -
-			#echo -e "${formatted_text}" 
 		else
 			echo -e "${CHATGPT_CYAN_LABEL}${response_data}" | fold -s -w "$COLUMNS"
 		fi
@@ -399,7 +443,7 @@ while $running; do
 			echo -e "${CHATGPT_CYAN_LABEL}"
 			echo "${response_data}" | glow -
 		else
-		# else remove empty lines and print
+			# else remove empty lines and print
 			formatted_text=$(echo "${response_data}" | sed '1,2d; s/^A://g')
 			echo -e "${CHATGPT_CYAN_LABEL}${formatted_text}" | fold -s -w $COLUMNS
 		fi