def co2_steuer():
    print("Wählen Sie die Antriebsart:")
    print("1 = E-Auto")
    print("2 = Hybrid")
    print("3 = Benziner")
    print("4 = Diesel")

    antriebsart = input("Bitte Antriebsart wählen (1-4): ")
    ps = int(input("Bitte PS-Zahl eingeben: "))

    betrag_monat = 0
    stufe = ""

    if antriebsart == "1":
        stufe = "E-Auto"
        betrag_monat = 0
    elif antriebsart == "2":
        stufe = "Hybrid"
        betrag_monat = 35 if ps <= 150 else 50
    elif antriebsart == "3":
        stufe = "Benziner"
        betrag_monat = 55 if ps <= 150 else 70
    elif antriebsart == "4":
        stufe = "Diesel"
        betrag_monat = 75 if ps <= 150 else 90
    else:
        print("Ungültige Eingabe!")
        return

    betrag_jahr = betrag_monat * 12

    print(f"\nFahrzeugtyp: {stufe}")
    print(f"PS-Zahl: {ps}")
    print(f"Monatliche CO2-Abgabe: {betrag_monat} €")
    print(f"Jährliche CO2-Abgabe: {betrag_jahr} €")

if __name__ == "__main__":
    co2_steuer()
