inhancing the UI and validation in enums.py file

Author: shreefAhmedM

sprint5-prep-exercises/enums.py

@@ -33,23 +33,28 @@ class Laptop:
 ]
 
 
-# Get the person's name
 name = input("What is your name? ")
 
-
-# Get and convert age
 try:
     age = int(input("What is your age? "))
 except ValueError:
     print("Age must be a number.", file=sys.stderr)
     sys.exit(1)
 
 
-# Get and convert operating system
+print("Available operating systems:")
+for os in OperatingSystem:
+    print(os.value)
+
 try:
-    os_input = input("What is your preferred operating system? ")
-    operating_system = OperatingSystem(os_input)
-except ValueError:
+    os_input = input("What is your preferred operating system? ").strip().lower()
+
+    operating_system = next(
+        os for os in OperatingSystem
+        if os.value.lower() == os_input
+    )
+
+except StopIteration:
     print("Invalid operating system.", file=sys.stderr)
     sys.exit(1)
 
@@ -61,21 +66,18 @@ class Laptop:
 )
 
 
-# Count laptops with the person's preferred OS
 number_available = sum(
     laptop.operating_system == person.preferred_operating_system
     for laptop in laptops
 )
 
-
 print(
     f"We have {number_available} laptop(s) with "
     f"{person.preferred_operating_system.value}."
 )
 
 
-# Find the operating system with the most laptops
-counts = {}
+counts: dict[OperatingSystem, int] = {}
 
 for os in OperatingSystem:
     counts[os] = sum(
@@ -86,7 +88,6 @@ class Laptop:
 
 most_available_os = max(counts, key=counts.get)
 
-
 if most_available_os != person.preferred_operating_system:
     if counts[most_available_os] > number_available:
         print(