Adding the option to start again

Author: Eladkrauz

PasswordGenerator

@@ -95,6 +95,18 @@ def generatePassword(length, complexity):
         segmentsCount[pickSegment] -= 1
     newPassword = ''.join(newPassword)
     return newPassword
+
+def startAgain():
+#a function for letting the user to start again the program.
+    choice = input("\nStart again? (y/n): ")
+    choice = choice.lower()
+    if choice == 'y':
+        main()
+    elif choice == 'n':
+        return
+    else:
+        print("Type y or n only please.")
+        startAgain()
 
 def main():
     print("Hello!")
@@ -106,4 +118,6 @@ def main():
         complexity = getComplexity()
     password = generatePassword(length, complexity)
     print(".\n.\n.\n.\n.\nYour password is: ", password)
+    startAgain()
+    
 main()