app.py

2 months ago · d2345731b4
parent fc476c74c5
commit d2345731b4
1 changed files with 3 additions and 2 deletions
--- a/app.py
+++ b/app.py
@ -384,7 +384,7 @@ def main():
    print("4. Generate the website")
    print("5. Delete a project")
    print("6. Exit")
-    choice = input("Enter your choice (1, 2, 3, 4, or 5): ")
+    choice = input("Enter your choice (1, 2, 3, 4, 5 or 6): ")
    
    if choice == '1':
        generate_template()
@ -399,7 +399,8 @@ def main():
    elif choice == '6':
        exit()
    else:
-        print("Invalid choice. Please enter 1, 2, 3, or 4.")
+        print("Invalid choice. Please enter 1, 2, 3, 4, 5 or 6.")
+        main()

 if __name__ == "__main__":
    main()