紧急求助高人

我们学的是微机原理，8086，题目如下：

从键盘输入一个正整数，计算其平方根并输出。
要求：提示输入一整数；
键盘输入，Enter 键结束输入，并换行显示结果

有下面一个程序，谁能帮忙修改一下，让它具有输入以及显示的功能，达到题目的要求！

data segment
x dw ? ; 待求平方根的数
root dw ? ; 结果保存在这里
data ends
code segment
assume cs:code, ds:data
start:
  mov ax, data
  mov ds, ax
  mov ax, x ; ax被减数
  mov bx, 1 ; 从1开始减
  mov cx, -1 ; cx 计数
next:
  sub ax, bx ; 依次减去1,3,5,....
  jc ok ; 不够减就说明找到了根，值就是循环的次数
  add bx, 2 ; 更新bx
  loop next
ok:
  not cx ; 取反求根
  mov root, cx ; 保存
  mov ax, 4c00h
  int 21h
code ends
end start

望各位不吝赐教，万分感谢

重新编一个也行，不要太专业，稍微要能看懂的

这是一段含有接受键盘输入和显示输出的例子
Assembly code

;
;This Program Compiled Sucess by Masm 6.15
;
assume    cs:code,ds:data
data    segment
buf    db    5,0,5 dup (0)；定义输入缓冲区，接受4位数字
data    ends
code    segment
start:
    mov ax,data
    mov ds,ax
    call input           ；input是输入子程序，结果在AX里
    push ax              ；保存AX
    mov ah,2             ；换行
    mov dl,13
    int 21h
    mov dl,10
    int 21h
    pop dx               ；恢复AX弹至DX，
    call show            ；show是显示十进制子程序，将DX的值以十进制显示输出

    mov ah,4ch
    int 21h
;
;result in AX
input    proc
    mov dx,offset buf
    mov ah,10
    int 21h
    
    xor cx,cx
    mov cl,buf+1
    mov bx,offset buf+2
    mov ax,0
inp:
    mov dx,10
    mul dx
    and byte ptr [bx],0fh
    add al,[bx]
    adc ah,0
    inc bx
    loop inp
    
    ret
input    endp
;
;input:DX
show    proc uses ax bx cx dx
    mov ax,dx
    xor dx,dx
    mov bx,10
    mov cx,0
a:    cmp ax,10
    jb ok
    div bx
    add dl,30h
    push dx
    xor dx,dx
    inc cx
    jmp a
ok:    add al,30h
    push ax
    inc cx
b:    pop dx
    mov ah,2
    int 21h
    loop b
    ret
show    endp
;
code    ends
end    start

有错误啊！