#Struktura kodu
# 1. importy
# 2. vlastní funkce/definice
# 3. hlavní spustitelná část kódu

import time

# TODO: Základní matematické funkce
# TODO: základní geometrické výpočty

#ZÁKLADNÍ ARITMETICKÉ FUNKCE
def secti(a,b):
    c = a+b
    return c

def odecti(a,b):
    return a-b

def vynasob(a,b):
    return a*b

def vydel(a,b):
    if b==0:
        print("Chyba:nelze dělit nulou")
        return 0
    else:
        return a/b
    
# ZÁKLADNÍ GEOMETRICKÉ FUNKCE

def obsah_ctverce(strana):
    return strana*strana

def obvod_ctverce(strana):
    return 4*strana

def obsah_obdelnika(a,b):
    return a*b

def obvod_obdelnika(a,b):
    return (a+b)*2

## POMOCNÉ FUNKCE

def nacti_cislo(vyzva):
    while True:
        try:
            vstupni_text = input(vyzva)
            cislo = float(vstupni_text)
            return cislo
        except ValueError:
            print("Chyba! Musíš zadat platné číslo. Zkus to znova")

## HLAVNÍ ČÁST KÓDU
print("Vítej v malíčkatý kalkulačce")

while True:
    print("\n---Hlavní menu---")
    print("1: Základní aritmetika (2 číla)")
    print("2:geometrické výpočty")
    print("3:konec programu")
    hlavni_volba= input("vyber si možnost (1-3):")

    if hlavni_volba == "1":
        print("\n---Základní aritmetika---")
        cislo1 = nacti_cislo("Zadej prvni cislo:")
        cislo2 = nacti_cislo("Zadej druhe cislo:")

        print(f"{cislo1} + {cislo2} = {secti(cislo1,cislo2)}")
        print(f"{cislo1} - {cislo2} = {odecti(cislo1,cislo2)}")
        print(f"{cislo1} * {cislo2} = {vynasob(cislo1,cislo2)}")
        print(f"{cislo1} / {cislo2} = {vydel(cislo1,cislo2)}")
        time.sleep(2)

    #Výběr geometrického výpočtu
    elif hlavni_volba == "2":
        print("\n---Gemotetrické výpočty---")
        print("1: čtverec")
        print("2: obdelník")
        geo_volba= input("vyber si možnost 1/2:")

        #ČTVEREC
        if geo_volba == "1":
            a = nacti_cislo("Zadej stranu čtverce:")
            print(f"Obvod čtverce: {obvod_ctverce(a)}")
            print(f"Obsah čtverce: {obsah_ctverce(a)}")
            time.sleep(2)

        #OBDELNÍK
        elif geo_volba == "2":
            a = nacti_cislo("Zadej stranu obdelníku a:")
            b = nacti_cislo("Zadej stranu obdelníku b:")
            print(f"Obvod obdelníku: {obvod_obdelnika(a,b)}")
            print(f"Obsah obdelníku: {obsah_obdelnika(a,b)}")
            time.sleep(2)

        #chyba zadání    
        else:
            print("Neplatná volba tvaru.")
    elif hlavni_volba == "3":
        print("Děkujeme využití naši aplikace")
        break
    else:
        print("Neplatná volba. Zadej 1, 2 nebo 3.")