Improve UX

Author: zohrehKazemianpour

exercises-sprint5/enums.py

@@ -91,11 +91,14 @@ def find_possible_laptops(laptops: List[Laptop], person: Person) -> List[Laptop]
         print(f"Error: {age_str} is not a valid age. Please enter a number.")
 
 # Convert OS string to enum with error handling
-try:
-    os = OperatingSystem(os_str) 
-except ValueError:
-    print(f"Error: '{os_str}' is not a valid OS. Valid options are: Ubuntu, macOS, Arch Linux", file=sys.stderr)
-    sys.exit(1)
+# Convert OS string to enum with error handling
+os = None
+while os is None:
+    try:
+        os = OperatingSystem(os_str) 
+    except ValueError:
+        print(f"Error: '{os_str}' is not a valid OS. Valid options are: Ubuntu, macOS, Arch Linux", file=sys.stderr)
+        os_str = input("Enter preferred OS: ")
 
 
 # Create Person object from validated input