Prime number searcher very basic and its no optimized

szam = int(input("Adjon meg egy számot: "))
prim =True
#itt lehetne optimalizálni
if (szam==1 or szam == 0):
    print (szam, " nem prímszám!\n")
    prim = False
else: #itt is lehetne még optimalizálni
    for i in range(2,round(szam/2,None),1):
        if (szam%i==0):
            prim = False
            print(szam," nem primszám!")
            break

if (prim == True):
    print (szam, " primszám!")
input("Nyomjon le egy billentyűt")