updated the script as per suggestion

Author: Priyanka-Microsoft

scripts/validate_model_deployment_quota.ps1

@@ -19,6 +19,18 @@ if ($MissingParams.Count -gt 0) {
     exit 1
 }
 
+# Check Azure login
+try {
+    $accountCheck = az account show --output none 2>$null
+    if ($LASTEXITCODE -ne 0) {
+        Write-Error "❌ ERROR: You are not logged in to Azure CLI. Please run 'az login' to continue."
+        exit 1
+    }
+} catch {
+    Write-Error "❌ ERROR: Failed to verify Azure login. Please run 'az login' to continue."
+    exit 1
+}
+
 # Load model deployments from parameter file
 $JsonContent = Get-Content -Path "./infra/main.parameters.json" -Raw | ConvertFrom-Json
 $aiModelDeployments = $JsonContent.parameters.$ModelsParameter.value

scripts/validate_model_deployment_quota.sh

@@ -40,6 +40,13 @@ if [[ ${#MISSING_PARAMS[@]} -ne 0 ]]; then
   exit 1
 fi
 
+# Check if user is logged in to Azure
+if ! az account show > /dev/null 2>&1; then
+  echo "❌ ERROR: You are not logged in to Azure CLI."
+  echo "👉 Please run 'az login' to continue."
+  exit 1
+fi
+
 # Load model definitions
 aiModelDeployments=$(jq -c ".parameters.$MODELS_PARAMETER.value[]" ./infra/main.parameters.json 2>/dev/null)
 if [[ $? -ne 0 || -z "$aiModelDeployments" ]]; then

scripts/validate_model_quota.ps1

@@ -111,7 +111,13 @@ if ($PrimaryResult) {
             Write-Host "🚨 Your application may not work as expected due to limited quota."
             Write-Host "`nℹ️  Recommended regions: $($RecommendedRegions -join ', ')"
             Write-Host "👉 It's advisable to deploy in one of these regions for optimal app performance."
-            $choice = Read-Host "❓ Do you want to choose a recommended region instead? (y/n)"
+            do {
+                $choice = Read-Host "❓ Do you want to choose a recommended region instead? (y/n)"
+                if ($choice -notmatch "^[YyNn]$") {
+                    Write-Host "❌ Invalid input. Please enter 'y' or 'n'."
+                }
+            } while ($choice -notmatch "^[YyNn]$")
+
             if ($choice -match "^[Yy]$") {
                 Show-Table
                 break
@@ -186,7 +192,13 @@ while ($true) {
         Write-Host "`n⚠️  You have entered a capacity of $ManualCapacity, which is less than the recommended minimum (200)."
         Write-Host "🚨 This may cause performance issues or unexpected behavior."
         Write-Host "ℹ️  Recommended regions (≥ $RECOMMENDED_TOKENS tokens available): $($RecommendedRegions -join ', ')"
-        $proceed = Read-Host "❓ Proceed anyway? (y/n)"
+        do {
+            $proceed = Read-Host "❓ Proceed anyway? (y/n)"
+            if ($proceed -notmatch "^[YyNn]$") {
+                Write-Host "❌ Invalid input. Please enter 'y' or 'n'."
+            }
+        } while ($proceed -notmatch "^[YyNn]$")
+
         if ($proceed -notmatch "^[Yy]$") {
             continue
         }

scripts/validate_model_quota.sh

@@ -5,6 +5,7 @@ MODEL=""
 DEPLOYMENT_TYPE="Standard"
 CAPACITY=0
 RECOMMENDED_TOKENS=200
+TABLE_SHOWN=false
 
 ALL_REGIONS=('australiaeast' 'eastus' 'eastus2' 'francecentral' 'japaneast' 'norwayeast' 'southindia' 'swedencentral' 'uksouth' 'westus' 'westus3')
 
@@ -44,6 +45,9 @@ check_quota() {
   output=$(az cognitiveservices usage list --location "$region" --query "[?name.value=='$MODEL_TYPE']" --output json 2>/dev/null)
 
   if [[ -z "$output" || "$output" == "[]" ]]; then
+    if [[ "$region" == "$LOCATION" ]]; then
+      echo "⚠️ Could not retrieve the quota info for the region: $LOCATION"
+    fi
     return 2
   fi
 
@@ -107,6 +111,10 @@ ask_for_location() {
     echo -e "ℹ️  Recommended regions (≥ $RECOMMENDED_TOKENS tokens available): $recommended_list"
     echo -n "❓ Proceed anyway? (y/n): "
     read -r proceed_anyway < /dev/tty
+    while [[ ! "$proceed_anyway" =~ ^[YyNn]$ ]]; do
+      echo "❌ Invalid input. Please enter 'y' or 'n': "
+      read -r proceed_anyway < /dev/tty
+    done
     if [[ ! "$proceed_anyway" =~ ^[Yy]$ ]]; then
       ask_for_location
       return
@@ -138,7 +146,10 @@ check_fallback_regions() {
     fi
   done
 
-  show_table
+  if [[ "$TABLE_SHOWN" == false ]]; then
+    show_table
+    TABLE_SHOWN=true
+  fi
 
   if [[ ${#FALLBACK_RESULTS[@]} -gt 0 ]]; then
     echo -e "\n➡️  Found fallback regions with sufficient quota."
@@ -189,6 +200,10 @@ manual_region_input() {
     echo -e "ℹ️  Recommended regions (≥ $RECOMMENDED_TOKENS tokens available): ${RECOMMENDED_REGIONS[*]}"
     echo -n "❓ Proceed anyway? (y/n): "
     read -r proceed_anyway < /dev/tty
+    while [[ ! "$proceed_anyway" =~ ^[YyNn]$ ]]; do
+      echo "❌ Invalid input. Please enter 'y' or 'n': "
+      read -r proceed_anyway < /dev/tty
+    done
     if [[ ! "$proceed_anyway" =~ ^[Yy]$ ]]; then
       manual_region_input
       return