各位大佬，怎样编写python程序解答下列题目

import math

def fun(x):
    if x>=0:
        y=math.sqrt(x)+4
    else:
        y=x**2+6
    return y

print('请输入需要进行计算的数字：')
a=input()
print('x取%s时函数的值为：%d'%(a,fun(int(a))))

import math

def fun(x):
    y = 0
    if x>=0:
        y = math.sqrt(x)+4
    else:
        y = x*x+6
    return y

if __name__ == '__main__':
    res = fun(4)
    print(res)