# Load config
. ${HOME}/ayo.conf

model=$coder
modelfile=$coder_modelfile

host=$ollama_remote_host

# Initialize variables
typora_flag=false
other_args=""

# Process arguments to handle -t flag and collect other args
while [[ $# -gt 0 ]]; do
  case $1 in
    -t)
      typora_flag=true
      shift
      ;;
    *)
      other_args="$other_args $1"
      shift
      ;;
  esac
done

# Set other_args to the first argument if it exists, otherwise empty string
if [[ -n "$other_args" ]]; then
  # Remove leading space
  other_args="${other_args# }"

  IFS=' ' read -ra args_array <<< "$other_args"
  if [[ ${#args_array[@]} -gt 1 ]]; then
    # Remove first element and rejoin remaining elements
    other_args="${args_array[*]:1}"
  else
    # If there's only one argument, set other_args to empty string
    other_args=""
  fi
fi

if ! [ "$other_args" = "" ]; then
  if [ "$other_args" = "sleep" ]; then
    OLLAMA_HOST=$host ollama stop $model
  elif [ "$other_args" = "init" ]; then
    OLLAMA_HOST=$host ollama create "$model" -f "$modelfile"
  else
    # If -t flag is set, use typora to display output
    if [ "$typora_flag" = true ]; then
      tempfile="$(mktemp)"
      OLLAMA_HOST=$host ollama run $model "$other_args" --hidethinking > $tempfile
      typora $tempfile
    else
      # If no -t flag, just run the command normally
      OLLAMA_HOST=$host ollama run $model "$other_args" --hidethinking
    fi
  fi
else
  OLLAMA_HOST=$host ollama run $model --hidethinking
fi