Python - Ayuda con código de juego

import random as r

a = r.randint(1,1001)

print(a)

b = int(input('Juguemos a adivinar el número \n\

adivina el número del 1 al 1000 que estoy pensando:\n'))

while a != b:

  if a < b:

    print("Demasiado alto. Inténtelo de nuevo.")

    b = int(input("Ingresa otro número: \n"))

  if a > b:

    print("Demasiado bajo. Inténtelo de nuevo.")

    b = int(input("Ingresa otro número: \n"))

if a == b:

    print("Felicidades. Ha adivinado el número.")

¿Quieres volver a jugar?

import random as r

jugar="s"

while jugar=="s":

    a = r.randint(1,1001)

    print(a)

    b = int(input('Juguemos a adivinar el número \n\

    adivina el número del 1 al 1000 que estoy pensando:\n'))

    while a != b:

        if a < b:

            print("Demasiado alto. Inténtelo de nuevo.")

            b = int(input("Ingresa otro número: \n"))

        if a > b:

           print("Demasiado bajo. Inténtelo de nuevo.")

           b = int(input("Ingresa otro número: \n"))

    if a == b:

         print("Felicidades. Ha adivinado el número.\n\n\n")

         jugar=input("¿Quieres volver a jugar? ")