Update app.py

app.py

@@ -95,40 +95,40 @@ if name in user_data:
             save_user_data(user_data)
             display_message('🎉 PIN created successfully! 🎉', "🎉")
 
-        # comparing pin
-        count = 0
-        with st.form(key='pin_entry_form'):
-            pin = st.text_input('Please enter your PIN:', type='password', key="pin_input_existing")
-            st.form_submit_button('Submit PIN')
-
-        while count < 3:
-            if pin.isdigit() and pin == user_data[name]['pin']:
-                break
-            else:
-                count += 1
-                display_message('❌ Invalid PIN. Please try again. ❌', "❌")
+    # comparing pin
+    count = 0
+    with st.form(key='pin_entry_form'):
+        pin = st.text_input('Please enter your PIN:', type='password', key="pin_input_existing")
+        st.form_submit_button('Submit PIN')
 
-        # in case of a valid pin - continuing, or exiting
-        if count == 3:
-            display_message('❌ 3 UNSUCCESSFUL PIN ATTEMPTS, EXITING. YOUR CARD HAS BEEN LOCKED ❌', "❌")
+    while count < 3:
+        if pin.isdigit() and pin == user_data[name]['pin']:
+            break
         else:
-            display_message(f'🎉 LOGIN SUCCESSFUL, CONTINUE {name.capitalize()}! 🎉', "🎉")
-
-            # Main menu
-            options = ['View Balance', 'Withdraw Cash', 'Deposit Cash', 'Change PIN', 'Quit']
-            response = st.selectbox('Select an option:', options, key="menu_selectbox_existing")
-
-            if response == 'View Balance':
-                view_balance(user_data)
-            elif response == 'Withdraw Cash':
-                withdraw_cash(user_data)
-            elif response == 'Deposit Cash':
-                deposit_cash(user_data)
-            elif response == 'Change PIN':
-                change_pin(user_data)
-            elif response == 'Quit':
-                save_user_data(user_data)
-                display_message('🌟 Thank you for using our ATM. Have a great day! 🌟', "🌟")
+            count += 1
+            display_message('❌ Invalid PIN. Please try again. ❌', "❌")
+
+    # in case of a valid pin - continuing, or exiting
+    if count == 3:
+        display_message('❌ 3 UNSUCCESSFUL PIN ATTEMPTS, EXITING. YOUR CARD HAS BEEN LOCKED ❌', "❌")
+    else:
+        display_message(f'🎉 LOGIN SUCCESSFUL, CONTINUE {name.capitalize()}! 🎉', "🎉")
+
+        # Main menu
+        options = ['View Balance', 'Withdraw Cash', 'Deposit Cash', 'Change PIN', 'Quit']
+        response = st.selectbox('Select an option:', options, key="menu_selectbox_existing")
+
+        if response == 'View Balance':
+            view_balance(user_data)
+        elif response == 'Withdraw Cash':
+            withdraw_cash(user_data)
+        elif response == 'Deposit Cash':
+            deposit_cash(user_data)
+        elif response == 'Change PIN':
+            change_pin(user_data)
+        elif response == 'Quit':
+            save_user_data(user_data)
+            display_message('🌟 Thank you for using our ATM. Have a great day! 🌟', "🌟")
 else:
     # New user; create an entry with default values
     user_data[name] = {'pin': '', 'amount': 0}
@@ -144,3 +144,11 @@ else:
         user_data[name]['pin'] = new_pin
         save_user_data(user_data)
         display_message('🎉 PIN created successfully! 🎉', "🎉")
+
+    # Main menu (for new user)
+    options_new_user = ['Quit']
+    response_new_user = st.selectbox('Select an option:', options_new_user, key="menu_selectbox_new_user")
+
+    if response_new_user == 'Quit':
+        save_user_data(user_data)
+        display_message('🌟 Thank you for using our ATM. Have a great day! 🌟', "🌟")