function prompt::awsenv
    if test -n "$env"
        if [ $env = "prod" ]
            printf '%s ' \U2601
            echo -n -s (yellow)"["(red)(string upper $env)(yellow)"]"(off)
        else
            printf '%s ' \U2601
            echo -n -s (yellow)"["(off)(string upper $env)(yellow)"]"(off)
        end
    end
end

function prompt::pyenv
    if test -n "$PYENV_VERSION"
        echo -n -s (yellow)(printf '%s' \U1F40D)"["(cyan)"$PYENV_VERSION"(off)"@"(python -c "import sys;t='{v[0]}.{v[1]}.{v[2]}'.format(v=list(sys.version_info[:3]));sys.stdout.write(t)")(yellow)"]"(off)
    # else if test (command -v pyenv)
    else if test -n "$VIRTUAL_ENV"
        echo -n -s (yellow)(printf '%s' \U1F40D)"["(cyan)(pyenv version-name | sed 's/:.*$//' )(off)"@"(python -c "import sys;t='{v[0]}.{v[1]}.{v[2]}'.format(v=list(sys.version_info[:3]));sys.stdout.write(t)")(yellow)"]"(off)
    else if test (command -v python)
       echo -n -s (yellow)(printf '%s' \U1F40D)"["(cyan)(python -c "import sys;t='{v[0]}.{v[1]}.{v[2]}'.format(v=list(sys.version_info[:3]));sys.stdout.write(t)")(yellow)"]"(off) 
    end
end

function t31m_prompt

    set awsenv (prompt::awsenv)
    set pyenv (prompt::pyenv) 
    echo -n -s "$pyenv"
    echo -n -s "$awsenv"

	if test $CMD_DURATION
        # Show duration of the last command in seconds
        set duration (echo "$CMD_DURATION 1000" | awk '{printf "%.2fs", $1 / $2}')
        printf '%s ' \U23F1
        echo -n -s (yellow)"["(off)$duration(yellow)"]"(off)
    end
end