Add more fun continuous input for users on example 5b in fw

Author: malcolm-sparkfun

source_only_examples/project5_b.py

@@ -136,7 +136,14 @@ def drive(inputString):
         return
 
     direction = inputList[0].lower()  # Get the first word and convert it to lowercase (so we can accept 'f' or 'F')
-    distance = int(inputList[1])  # Get the second word and convert it to an integer
+    # Code that we put in the "try" block might cause an error, so we can catch that error with "except" and handle it gracefully
+    # For example if the user enters something weird for distance like "ten" instead of "10", the int() function will throw an error
+    # and we can catch that error and print a message instead of the program crashing
+    try:
+        distance = int(inputList[1])  # Get the second word and convert it to an integer
+    except:
+        print("Invalid distance. Please enter a valid number.")
+        return
 
     # check the switch state and only drive if the switch is pressed
     if switchPin.value() == 1:  # If the switch is in OFF position
@@ -171,3 +178,8 @@ def drive(inputString):
 #         print("Executing command:", c)
 #         drive(c)  # Execute each command in the list
 #         sleep(1)  # Add a delay between commands to see what's happening more clearly
+
+# The below code will run forever and allow us to enter further commands interactively!
+while True:
+    c = input("Enter a new command!: ") # Get user input in the c variable in the form of "f 10" or "r 90"
+    drive(c)