Merge pull request #40 from 0xacx/add-generate-command-option

WIP: First draft of adding command generation and execution and command ex…

Author: 0xacx

chatgpt.sh

@@ -5,6 +5,8 @@ CHAT_INIT_PROMPT="You are ChatGPT, a Large Language Model trained by OpenAI. You
 
 SYSTEM_PROMPT="You are ChatGPT, a large language model trained by OpenAI. Answer as concisely as possible. Current date: $(date +%d/%m/%Y). Knowledge cutoff: 9/1/2021."
 
+COMMAND_GENERATION_PROMPT="Return a one-line bash command with the functionality I will describe. Return ONLY the command ready to run in the terminal. The command should do the following:"
+
 CHATGPT_CYAN_LABEL="\n\033[36mchatgpt \033[0m"
 
 # error handling function
@@ -116,7 +118,7 @@ build_user_chat_message() {
 		chat_message="$chat_message, {\"role\": \"user\", \"content\": \"$escaped_prompt\"}"
 	fi
 
-	request_prompt=$chat_message
+	request_prompt="$chat_message"
 }
 
 # adds the assistant response to the message in (chatml) format
@@ -222,7 +224,7 @@ if [ -n "$prompt" ]; then
 	pipe_mode_prompt=${prompt}
 # if input file_descriptor is a terminal, run on chat mode
 elif [ -t 0 ]; then
-	echo -e "Welcome to chatgpt. You can quit with '\033[36mexit\033[0m'."
+	echo -e "Welcome to chatgpt. You can quit with '\033[36mexit\033[0m' or '\033[36mq\033[0m'."
 # prompt from pipe or redirected stdin, run on pipe mode
 else
 	pipe_mode_prompt+=$(cat -)
@@ -240,7 +242,7 @@ while $running; do
 		CHATGPT_CYAN_LABEL=""
 	fi
 
-	if [ "$prompt" == "exit" ]; then
+	if [ "$prompt" == "exit" ] || [ "$prompt" == "q" ]; then
 		running=false
 	elif [[ "$prompt" =~ ^image: ]]; then
 		request_to_image "$prompt"
@@ -275,6 +277,41 @@ while $running; do
 		handle_error "$models_response"
 		model_data=$(echo $models_response | jq -r -C '.data[] | select(.id=="'"${prompt#*model:}"'")')
 		echo -e "${CHATGPT_CYAN_LABEL}Complete details for model: ${prompt#*model:}\n ${model_data}"
+	elif [[ "$prompt" =~ ^command: ]]; then
+		# escape quotation marks
+		escaped_prompt=$(echo "$prompt" | sed 's/"/\\"/g')
+		# escape new lines
+		if [[ "$prompt" =~ ^command: ]]; then
+			escaped_prompt=${prompt#command:}
+			request_prompt=$COMMAND_GENERATION_PROMPT${escaped_prompt//$'\n'/' '}
+		fi
+		build_user_chat_message "$chat_message" "$request_prompt"
+		request_to_chat "$request_prompt"
+		handle_error "$response"
+		response_data=$(echo $response | jq -r '.choices[].message.content')
+
+		if [[ "$prompt" =~ ^command: ]]; then
+			echo -e "${CHATGPT_CYAN_LABEL} ${response_data}\n"
+			dangerous_commands=("rm" ">" "mv" "mkfs" ":(){:|:&};" "dd" "chmod" "wget" "curl")
+
+			for dangerous_command in "${dangerous_commands[@]}"; do
+				if [[ "$response_data" == *"$dangerous_command"* ]]; then
+					echo "Warning! This command can change your file system or download external scripts & data. Please do not execute code that you don't understand completely."
+				fi
+			done
+			echo "Would you like to execute it? (Yes/No)"
+			read run_answer
+			if [ "$run_answer" == "Yes" ] || [ "$run_answer" == "yes" ] || [ "$run_answer" == "y" ] || [ "$run_answer" == "Y" ]; then
+				echo -e "\nExecuting command: $response_data\n"
+				eval $response_data
+			fi
+		fi
+		response_data=$(echo "$response_data" | sed 's/"/\\"/g')
+		add_assistant_response_to_chat_message "$chat_message" "$response_data"
+
+		timestamp=$(date +"%d/%m/%Y %H:%M")
+		echo -e "$timestamp $prompt \n$response_data \n" >>~/.chatgpt_history
+
 	elif [[ "$MODEL" =~ ^gpt- ]]; then
 		# escape quotation marks
 		escaped_prompt=$(echo "$prompt" | sed 's/"/\\"/g')