Update interactivestory.py

interactivestory.py

@@ -6,18 +6,16 @@ def printSlowly(text):
         time.sleep(0.01)
 
 def text(story1, story2):
-    loop = True
-    while loop:
-        choice = int(input())
-        if choice == 1:
+    choice = 0
+    while choice != "1" and choice != "2":
+        choice = input()
+        if choice == "1":
             printSlowly(story1)
-            loop = False
-        elif choice == 2:
+        elif choice == "2":
             printSlowly(story2)
-            loop = False
         else:
             print("INVALID INPUT, try again")
-    return(choice)
+    return(int(choice))