from math import *

x = int(input("Введите х: "))

if x >= 0:

   y = sqrt(x)

else:

   y = x**2

print(y)

Блок схема на приложенном фото.