feat: add --interactive flag to img_api_generate.sh

img_api_generate.sh

@@ -40,6 +40,7 @@ DIM='\033[2m'
 
 CHECK_MARK="✓"
 CROSS_MARK="✗"
+QUESTION_MARK="?"
 PALETTE="🎨"
 SPARKLES="✨"
 CAMERA="📸"
@@ -76,6 +77,7 @@ ASYNC_MODE=false
 DRY_RUN=false
 VERBOSE=false
 UPSCALE=false
+INTERACTIVE=false
 
 # Temp files for cleanup
 _TMP_FILES=()
@@ -136,6 +138,22 @@ print_verbose() {
     fi
 }
 
+prompt_confirm() {
+    local message="$1"
+    if [[ "$INTERACTIVE" == false ]]; then
+        return 0
+    fi
+
+    echo -en "${BOLD_YELLOW}${QUESTION_MARK} ${message} [y/N] ${RESET}"
+    read -n 1 -r
+    echo ""
+    if [[ ! $REPLY =~ ^[Yy]$ ]]; then
+        print_info "Operation cancelled by user."
+        return 1
+    fi
+    return 0
+}
+
 # ============================================================================
 # SPINNER
 # ============================================================================
@@ -286,12 +304,15 @@ validate_inputs() {
 # OUTPUT FILENAME GENERATION
 # ============================================================================
 
+_GENERATED_TIMESTAMP=""
+
 generate_output_filename() {
     local suffix="${1:-}"
-    local timestamp
+    if [[ -z "$_GENERATED_TIMESTAMP" ]]; then
-    timestamp=$(date +%Y%m%d_%H%M%S)
+        _GENERATED_TIMESTAMP=$(date +%Y%m%d_%H%M%S)
+    fi
     local seed_part="${SEED:-0}"
-    local base="img_${MODEL}_${timestamp}_${seed_part}"
+    local base="img_${MODEL}_${_GENERATED_TIMESTAMP}_${seed_part}"
 
     if [[ -n "$suffix" ]]; then
         echo "${base}_${suffix}.png"
@@ -548,6 +569,8 @@ generate_image() {
     [[ -n "$INPUT_IMAGE" ]] && print_field " " "Input Image" "$INPUT_IMAGE"
     echo ""
 
+    prompt_confirm "Proceed with image generation?" || return 1
+
     local response
     if [[ "$ASYNC_MODE" == false ]]; then
         # Synchronous mode - wait with spinner
@@ -684,6 +707,8 @@ download_result() {
 
     print_detail "Downloading result..."
 
+    prompt_confirm "Download generated image?" || return 1
+
     api_curl_binary GET "$url" "$_dl_output_path" || return 1
 
     if [[ -f "$_dl_output_path" && -s "$_dl_output_path" ]]; then
@@ -727,6 +752,8 @@ swap_face() {
     print_field "$CAMERA" "Target" "$target_image"
     echo ""
 
+    prompt_confirm "Proceed with face swap?" || return 1
+
     local options='{"processors":["face_swapper","face_enhancer"],"face_swapper":{"model":"hyperswap_1a_256"},"face_enhancer":{"model":"gfpgan_1.4","blend":80}}'
 
     local tmp_resp
@@ -783,6 +810,8 @@ realesrgan_upscale() {
     print_field "$CAMERA" "Input" "$input_image"
     echo ""
 
+    prompt_confirm "Proceed with image upscaling?" || return 1
+
     local tmp_resp
     tmp_resp=$(mktemp)
     _TMP_FILES+=("$tmp_resp")
@@ -848,6 +877,7 @@ OUTPUT:
 MODES:
       --async                 Use async mode with polling instead of sync
       --dry-run               Show curl commands without executing
+  -i, --interactive           Prompt before each API call
       --verbose               Verbose output
   -h, --help                  Show this help message
 
@@ -918,6 +948,10 @@ parse_args() {
                 ASYNC_MODE=true
                 shift
                 ;;
+            -i|--interactive)
+                INTERACTIVE=true
+                shift
+                ;;
             --dry-run)
                 DRY_RUN=true
                 shift
@@ -970,6 +1004,9 @@ main() {
     # Validate inputs
     validate_inputs
 
+    # Initialize timestamp for consistent naming across subshells
+    _GENERATED_TIMESTAMP=$(date +%Y%m%d_%H%M%S)
+
     # Banner
     print_banner "${PALETTE} Pivoine Image Generator ${SPARKLES}"
 
@@ -1044,13 +1081,20 @@ main() {
             local base="${OUTPUT_FILE%.*}"
             upscale_path="${base}_upscaled.${ext}"
         else
-            upscale_path="$(generate_output_filename "upscaled")"
+            # Include faceswap in name if it was performed
+            if [[ -n "${faceswap_path:-}" && -f "${faceswap_path}" ]]; then
+                upscale_path="$(generate_output_filename "faceswap_upscaled")"
+            else
+                upscale_path="$(generate_output_filename "upscaled")"
+            fi
         fi
 
-        realesrgan_upscale "$final_input" upscale_path || exit 1
+        realesrgan_upscale "$final_input" upscale_path || true
 
-        # Use the upscaled image for summary/output
+        # Use the upscaled image for summary/output if it was created
-        output_path="$upscale_path"
+        if [[ -f "${upscale_path:-}" ]]; then
+            output_path="$upscale_path"
+        fi
     fi
 
     # Summary