Update app_enhanced.py

app_enhanced.py

@@ -2,35 +2,30 @@ import spaces  # <--- CRITICAL: MUST BE FIRST
 import os
 import time
 import threading
-import uuid
-import shutil
 import json
 import traceback
 import logging
 import string
 import random
+import shutil
 import cv2
+import math
 import numpy as np
 import srt
-from concurrent.futures import ThreadPoolExecutor
-from flask import Flask, render_template, request, jsonify, send_from_directory, send_file
+from flask import Flask, jsonify, request, send_from_directory, send_file
 
 # ======================================================
-# 🚀 ZEROGPU WARMUP (Required for HF Detection)
+# 🚀 ZEROGPU CONFIGURATION
 # ======================================================
 @spaces.GPU
 def gpu_warmup():
     import torch
-    print(f"✅ ZeroGPU Active. CUDA: {torch.cuda.is_available()}")
+    print(f"✅ ZeroGPU Warmup: CUDA Available: {torch.cuda.is_available()}")
     return True
 
 # ======================================================
-# 🔧 CONFIG & LOGGING
+# 🧱 DATA CLASSES (Local Definition)
 # ======================================================
-logging.basicConfig(level=logging.INFO)
-logger = logging.getLogger(__name__)
-
-# --- 2. ROBUST DATA CLASSES (Defined locally to avoid Import Errors) ---
 def bubble(dialog="", bubble_offset_x=50, bubble_offset_y=20, lip_x=-1, lip_y=-1, emotion='normal'):
     return {
         'dialog': dialog, 
@@ -49,7 +44,12 @@ class Page:
         self.panels = panels
         self.bubbles = bubbles
 
-# --- 3. FLASK SETUP ---
+# ======================================================
+# 🔧 APP SETUP
+# ======================================================
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
 app = Flask(__name__)
 BASE_USER_DIR = "userdata" 
 SAVED_COMICS_DIR = "saved_comics"
@@ -65,16 +65,16 @@ def generate_save_code(length=8):
             return code
 
 # ======================================================
-# 🧠 GLOBAL GPU FUNCTIONS (The Engine)
+# 🧠 GLOBAL GPU FUNCTIONS
 # ======================================================
 
 @spaces.GPU(duration=240)
-def generate_comic_gpu(video_path, user_dir, frames_dir, metadata_path):
-    print(f"🚀 GPU Task Started: {video_path}")
+def generate_comic_gpu(video_path, user_dir, frames_dir, metadata_path, target_pages):
+    print(f"🚀 GPU Task Started: {video_path} | Target Pages: {target_pages}")
     
-    # --- Local Imports ensure they run in GPU Context ---
     import cv2
     import srt
+    import math
     import numpy as np
     from backend.keyframes.keyframes import black_bar_crop
     from backend.simple_color_enhancer import SimpleColorEnhancer
@@ -98,7 +98,7 @@ def generate_comic_gpu(video_path, user_dir, frames_dir, metadata_path):
         if os.path.exists('test1.srt'):
             shutil.move('test1.srt', user_srt)
     except Exception as e:
-        print(f"⚠️ Subtitle error: {e}. Using fallback.")
+        print(f"⚠️ Subtitle error: {e}")
         with open(user_srt, 'w') as f: f.write("1\n00:00:01,000 --> 00:00:04,000\nHello\n")
             
     with open(user_srt, 'r', encoding='utf-8') as f:
@@ -107,12 +107,13 @@ def generate_comic_gpu(video_path, user_dir, frames_dir, metadata_path):
     key_moments = [{'index': s.index, 'text': s.content, 'start': s.start.total_seconds(), 'end': s.end.total_seconds()} for s in all_subs]
     key_moments.sort(key=lambda x: x['start'])
     
-    # 3. Extract Keyframes
+    # 3. Extract Frames (Limit max frames to keep it snappy)
     frame_metadata = {}
     cap = cv2.VideoCapture(video_path)
     count = 0
-    # Limit frames for GPU stability
-    for i, moment in enumerate(key_moments[:32]):
+    max_frames_to_extract = 40 
+    
+    for i, moment in enumerate(key_moments[:max_frames_to_extract]):
         mid = (moment['start'] + moment['end']) / 2
         if mid > duration: continue
         cap.set(cv2.CAP_PROP_POS_FRAMES, int(mid * fps))
@@ -138,9 +139,9 @@ def generate_comic_gpu(video_path, user_dir, frames_dir, metadata_path):
     print(f"🎨 Enhancing {len(frame_files)} frames...")
     for f in frame_files:
         p = os.path.join(frames_dir, f)
-        try: se.enhance_single(p, p) # Pass output path!
+        try: se.enhance_single(p, p)
         except: pass
-        try: qe.enhance_single(p, p) # Pass output path!
+        try: qe.enhance_single(p, p)
         except: pass
         
     # 5. Bubbles
@@ -153,23 +154,35 @@ def generate_comic_gpu(video_path, user_dir, frames_dir, metadata_path):
             faces = face_detector.detect_faces(p)
             lip = face_detector.get_lip_position(p, faces[0]) if faces else (-1, -1)
             bx, by = ai_bubble_placer.place_bubble_ai(p, lip)
-            # Use local safe function
             bubbles_list.append(bubble(dialog=dialogue, bubble_offset_x=bx, bubble_offset_y=by, lip_x=lip[0], lip_y=lip[1]))
         except:
             bubbles_list.append(bubble(dialog=dialogue))
             
-    # 6. Layout
+    # 6. Dynamic Layout Calculation
     print("📄 Generating layout...")
-    try:
-        from backend.fixed_12_pages_800x1080 import generate_12_pages_800x1080
-        pages = generate_12_pages_800x1080(frame_files, bubbles_list)
-    except:
-        pages = []
-        for i in range((len(frame_files)+3)//4):
-            start = i*4
-            pg_panels = [panel(image=f) for f in frame_files[start:start+4]]
-            pg_bubbles = bubbles_list[start:start+4]
-            if pg_panels: pages.append(Page(panels=pg_panels, bubbles=pg_bubbles))
+    total_panels = len(frame_files)
+    
+    # Calculate panels per page
+    if target_pages <= 0: target_pages = 1
+    
+    # If we have 32 panels and user wants 4 pages -> 8 panels/page
+    # If we have 32 panels and user wants 10 pages -> ~3 panels/page
+    panels_per_page = math.ceil(total_panels / target_pages)
+    
+    print(f"📊 Layout: {total_panels} panels across {target_pages} pages (~{panels_per_page}/page)")
+    
+    pages = []
+    for i in range(target_pages):
+        start_idx = i * panels_per_page
+        end_idx = start_idx + panels_per_page
+        
+        # Slicing safely
+        page_frames = frame_files[start_idx:end_idx]
+        page_bubbles = bubbles_list[start_idx:end_idx]
+        
+        if page_frames:
+            pg_panels = [panel(image=f) for f in page_frames]
+            pages.append(Page(panels=pg_panels, bubbles=page_bubbles))
 
     # Serialize Results
     result = []
@@ -185,7 +198,6 @@ def regen_frame_gpu(video_path, frames_dir, metadata_path, fname, direction):
     import cv2
     import json
     from backend.simple_color_enhancer import SimpleColorEnhancer
-    from backend.quality_color_enhancer import QualityColorEnhancer
     
     if not os.path.exists(metadata_path): return {"success": False, "message": "No metadata"}
     with open(metadata_path, 'r') as f: meta = json.load(f)
@@ -203,11 +215,12 @@ def regen_frame_gpu(video_path, frames_dir, metadata_path, fname, direction):
     
     if ret:
         p = os.path.join(frames_dir, fname)
+        # Write then Flush to prevent libpng errors
         cv2.imwrite(p, frame)
+        os.sync() 
+        
         try: SimpleColorEnhancer().enhance_single(p, p)
         except: pass
-        try: QualityColorEnhancer().enhance_single(p, p)
-        except: pass
         
         if isinstance(meta[fname], dict): meta[fname]['time'] = new_t
         else: meta[fname] = new_t
@@ -220,7 +233,6 @@ def get_frame_at_ts_gpu(video_path, frames_dir, metadata_path, fname, ts):
     import cv2
     import json
     from backend.simple_color_enhancer import SimpleColorEnhancer
-    from backend.quality_color_enhancer import QualityColorEnhancer
     
     cap = cv2.VideoCapture(video_path)
     cap.set(cv2.CAP_PROP_POS_MSEC, float(ts) * 1000)
@@ -230,10 +242,10 @@ def get_frame_at_ts_gpu(video_path, frames_dir, metadata_path, fname, ts):
     if ret:
         p = os.path.join(frames_dir, fname)
         cv2.imwrite(p, frame)
+        os.sync()
+        
         try: SimpleColorEnhancer().enhance_single(p, p)
         except: pass
-        try: QualityColorEnhancer().enhance_single(p, p)
-        except: pass
         
         if os.path.exists(metadata_path):
             with open(metadata_path, 'r') as f: meta = json.load(f)
@@ -245,7 +257,7 @@ def get_frame_at_ts_gpu(video_path, frames_dir, metadata_path, fname, ts):
     return {"success": False, "message": "Invalid timestamp"}
 
 # ======================================================
-# 💻 BACKEND HELPER CLASS
+# 💻 BACKEND CLASS
 # ======================================================
 class EnhancedComicGenerator:
     def __init__(self, sid):
@@ -264,11 +276,11 @@ class EnhancedComicGenerator:
         os.makedirs(self.frames_dir, exist_ok=True)
         os.makedirs(self.output_dir, exist_ok=True)
 
-    def run(self):
+    def run(self, target_pages):
         try:
             self.write_status("Waiting for GPU...", 5)
-            # Call Global GPU Function
-            data = generate_comic_gpu(self.video_path, self.user_dir, self.frames_dir, self.metadata_path)
+            # Pass target_pages to GPU function
+            data = generate_comic_gpu(self.video_path, self.user_dir, self.frames_dir, self.metadata_path, int(target_pages))
             with open(os.path.join(self.output_dir, 'pages.json'), 'w') as f:
                 json.dump(data, f, indent=2)
             self.write_status("Complete!", 100)
@@ -281,10 +293,9 @@ class EnhancedComicGenerator:
             json.dump({'message': msg, 'progress': prog}, f)
 
 # ======================================================
-# 🌐 ROUTES & HTML
+# 🌐 ROUTES
 # ======================================================
 
-# PASTE YOUR FULL INTERACTIVE HTML HERE
 INDEX_HTML = '''
 <!DOCTYPE html>
 <html lang="en">
@@ -297,22 +308,37 @@ INDEX_HTML = '''
     <style>
         * { box-sizing: border-box; }
         body { background-color: #fdf6e3; font-family: 'Lato', sans-serif; color: #3d3d3d; margin: 0; min-height: 100vh; }
+        
+        /* UPLOAD SCREEN */
         #upload-container { display: flex; flex-direction:column; justify-content: center; align-items: center; min-height: 100vh; width: 100%; padding: 20px; }
         .upload-box { max-width: 500px; width: 100%; padding: 40px; background: white; border-radius: 12px; box-shadow: 0 8px 30px rgba(0,0,0,0.12); text-align: center; }
+        
+        /* EDITOR SCREEN */
         #editor-container { display: none; padding: 20px; width: 100%; box-sizing: border-box; padding-bottom: 100px; }
+        
         h1 { color: #2c3e50; margin-bottom: 20px; font-weight: 600; }
         .file-input { display: none; }
         .file-label { display: block; padding: 15px; background: #2c3e50; color: white; border-radius: 8px; cursor: pointer; font-weight: bold; margin-bottom: 10px; transition:0.2s; }
         .file-label:hover { background: #34495e; }
+        
+        /* PAGE INPUT STYLES */
+        .page-input-group { margin: 15px 0; text-align: left; }
+        .page-input-group label { font-weight: bold; font-size: 14px; display: block; margin-bottom: 5px; color: #333; }
+        .page-input-group input { width: 100%; padding: 10px; border: 2px solid #ddd; border-radius: 8px; font-size: 16px; }
+        
         .submit-btn { width: 100%; padding: 15px; background: #e67e22; color: white; border: none; border-radius: 8px; font-size: 18px; font-weight: bold; cursor: pointer; transition: 0.2s; }
         .submit-btn:hover { background: #d35400; }
         .restore-btn { margin-top: 10px; background: #27ae60; color: white; padding: 12px; width: 100%; border: none; border-radius: 8px; cursor: pointer; font-weight: bold; }
+        
         .load-section { margin-top: 30px; padding-top: 20px; border-top: 2px solid #eee; }
         .load-input-group { display: flex; gap: 10px; margin-top: 10px; }
         .load-input-group input { flex: 1; padding: 12px; border: 2px solid #ddd; border-radius: 8px; font-size: 16px; text-transform: uppercase; letter-spacing: 2px; text-align: center; }
         .load-input-group button { padding: 12px 20px; background: #3498db; color: white; border: none; border-radius: 8px; cursor: pointer; font-weight: bold; }
+        
         .loader { width: 120px; height: 20px; background: radial-gradient(circle 10px, #e67e22 100%, transparent 0); background-size: 20px 20px; animation: ball 1s infinite linear; margin: 20px auto; }
         @keyframes ball { 0%{background-position:0 50%} 100%{background-position:100px 50%} }
+        
+        /* COMIC ELEMENTS */
         .comic-wrapper { max-width: 1000px; margin: 0 auto; }
         .page-wrapper { margin: 30px auto; display: flex; flex-direction: column; align-items: center; }
         .page-title { text-align: center; color: #333; margin-bottom: 10px; font-size: 18px; font-weight: bold; }
@@ -323,6 +349,8 @@ INDEX_HTML = '''
         .panel img { width: 100%; height: 100%; object-fit: cover; transition: transform 0.1s ease-out; transform-origin: center center; }
         .panel img.pannable { cursor: grab; }
         .panel img.panning { cursor: grabbing; }
+        
+        /* BUBBLES */
         .speech-bubble { position: absolute; display: flex; justify-content: center; align-items: center; width: 150px; height: 80px; min-width: 50px; min-height: 30px; box-sizing: border-box; z-index: 10; cursor: move; font-family: 'Comic Neue', cursive; font-weight: bold; font-size: 13px; text-align: center; overflow: visible; }
         .bubble-text { padding: 0.8em; word-wrap: break-word; position: relative; z-index: 5; }
         .speech-bubble.selected { outline: 2px dashed #4CAF50; z-index: 100; }
@@ -332,57 +360,24 @@ INDEX_HTML = '''
         .speech-bubble.speech.tail-bottom:before { top: 100%; left: clamp(0%, calc(var(--p) - (1 - var(--t)) * var(--b) / 2), calc(100% - (1 - var(--t)) * var(--b))); }
         .speech-bubble.speech.tail-top { border-radius: min(var(--r), calc(var(--p) - (1 - var(--t)) * var(--b) / 2)) min(var(--r), calc(100% - var(--p) - (1 - var(--t)) * var(--b) / 2)) var(--r) var(--r) / var(--r); }
         .speech-bubble.speech.tail-top:before { bottom: 100%; left: clamp(0%, calc(var(--p) - (1 - var(--t)) * var(--b) / 2), calc(100% - (1 - var(--t)) * var(--b))); transform: scaleY(-1); }
-        .speech-bubble.speech.tail-left { border-radius: var(--r); }
-        .speech-bubble.speech.tail-left:before { right: 100%; top: clamp(0%, calc(var(--p) - (1 - var(--t)) * var(--b) / 2), calc(100% - (1 - var(--t)) * var(--b))); transform: rotate(90deg); transform-origin: top right; }
-        .speech-bubble.speech.tail-right { border-radius: var(--r); }
-        .speech-bubble.speech.tail-right:before { left: 100%; top: clamp(0%, calc(var(--p) - (1 - var(--t)) * var(--b) / 2), calc(100% - (1 - var(--t)) * var(--b))); transform: rotate(-90deg); transform-origin: top left; }
-        .speech-bubble.thought { background: white; border: 2px dashed #555; color: #333; border-radius: 50%; }
-        .speech-bubble.thought::after { display:none; } 
-        .thought-dot { position: absolute; background-color: white; border: 2px solid #555; border-radius: 50%; z-index: -1; }
-        .thought-dot-1 { width: 20px; height: 20px; bottom: -20px; left: 15px; }
-        .thought-dot-2 { width: 12px; height: 12px; bottom: -32px; left: 5px; }
-        .speech-bubble.flipped .thought-dot-1 { left: auto; right: 15px; }
-        .speech-bubble.flipped .thought-dot-2 { left: auto; right: 5px; }
-        .speech-bubble.flipped-vertical .thought-dot-1 { bottom: auto; top: -20px; }
-        .speech-bubble.flipped-vertical .thought-dot-2 { bottom: auto; top: -32px; }
-        .speech-bubble.reaction { background: #FFD700; border: 3px solid #E53935; color: #D32F2F; font-weight: 900; text-transform: uppercase; width: 180px; clip-path: polygon(0% 25%, 17% 21%, 17% 0%, 31% 16%, 50% 4%, 69% 16%, 83% 0%, 83% 21%, 100% 25%, 85% 45%, 95% 62%, 82% 79%, 100% 97%, 79% 89%, 60% 98%, 46% 82%, 27% 95%, 15% 78%, 5% 62%, 15% 45%); }
-        .speech-bubble.narration { background: #FAFAFA; border: 2px solid #BDBDBD; color: #424242; border-radius: 3px; }
-        .speech-bubble.idea { background: linear-gradient(180deg,#FFFDD0 0%, #FFF8B5 100%); border: 2px solid #FFA500; color: #6a4b00; border-radius: 40% 60% 40% 60% / 60% 40% 60% 40%; }
+        
         .resize-handle { position: absolute; width: 10px; height: 10px; background: #2196F3; border: 1px solid white; border-radius: 50%; display: none; z-index: 11; }
         .speech-bubble.selected .resize-handle { display: block; }
         .resize-handle.se { bottom: -5px; right: -5px; cursor: se-resize; }
-        .resize-handle.sw { bottom: -5px; left: -5px; cursor: sw-resize; }
-        .resize-handle.ne { top: -5px; right: -5px; cursor: ne-resize; }
-        .resize-handle.nw { top: -5px; left: -5px; cursor: nw-resize; }
+        
+        /* CONTROLS */
         .edit-controls { position: fixed; bottom: 20px; right: 20px; width: 260px; background: rgba(44, 62, 80, 0.95); color: white; padding: 15px; border-radius: 8px; box-shadow: 0 5px 15px rgba(0,0,0,0.3); z-index: 900; font-size: 13px; max-height: 90vh; overflow-y: auto; }
         .edit-controls h4 { margin: 0 0 10px 0; color: #4ECDC4; text-align: center; }
         .control-group { margin-top: 10px; border-top: 1px solid #555; padding-top: 10px; }
-        .control-group label { font-size: 11px; font-weight: bold; display: block; margin-bottom: 3px; }
         button, input, select { width: 100%; margin-top: 5px; padding: 6px; border-radius: 4px; border: 1px solid #ddd; cursor: pointer; font-weight: bold; font-size: 12px; }
-        button:hover { background: #f5f5f5; }
         .button-grid { display: grid; grid-template-columns: 1fr 1fr; gap: 5px; }
-        .color-grid { display: grid; grid-template-columns: 1fr 1fr; gap: 10px; }
-        .color-grid div { text-align: center; }
-        .color-grid input[type="color"] { height: 30px; padding: 2px; }
-        .slider-container { display: flex; align-items: center; gap: 5px; margin-top: 5px; }
-        .slider-container label { min-width: 40px; font-size: 11px; }
-        .timestamp-controls { display: grid; grid-template-columns: 1fr auto; gap: 5px; }
-        .timestamp-controls input { color: #333; font-weight: normal; }
         .action-btn { background: #4CAF50; color: white; }
-        .reset-btn { background: #e74c3c; color: white; }
-        .secondary-btn { background: #f39c12; color: white; }
-        .export-btn { background: #2196F3; color: white; }
-        .save-btn { background: #9b59b6; color: white; }
-        .save-code-display { background: #2ecc71; color: white; padding: 15px; border-radius: 8px; text-align: center; margin-top: 10px; display: none; }
-        .save-code-display .code { font-size: 24px; font-weight: bold; letter-spacing: 3px; background: white; color: #2ecc71; padding: 10px 20px; border-radius: 4px; display: inline-block; margin: 10px 0; font-family: monospace; }
-        .save-code-display button { background: white; color: #2ecc71; border: none; padding: 8px 15px; border-radius: 4px; cursor: pointer; font-weight: bold; margin-top: 5px; }
+        .action-btn:disabled { background: #555; cursor: not-allowed; opacity: 0.7; }
+        
+        /* MODAL */
         .modal-overlay { position: fixed; top: 0; left: 0; width: 100%; height: 100%; background: rgba(0,0,0,0.7); display: none; justify-content: center; align-items: center; z-index: 9999; }
         .modal-content { background: white; padding: 30px; border-radius: 12px; max-width: 400px; width: 90%; text-align: center; }
-        .modal-content h2 { color: #2ecc71; margin-bottom: 20px; }
         .modal-content .code { font-size: 32px; font-weight: bold; letter-spacing: 4px; background: #f0f0f0; padding: 15px 25px; border-radius: 8px; display: inline-block; margin: 15px 0; font-family: monospace; user-select: all; }
-        .modal-content p { color: #666; margin: 10px 0; }
-        .modal-content button { background: #3498db; color: white; border: none; padding: 12px 30px; border-radius: 8px; cursor: pointer; font-weight: bold; font-size: 14px; margin: 5px; }
-        .modal-content button.close-btn { background: #95a5a6; }
     </style>
 </head>
 <body>
@@ -392,11 +387,17 @@ INDEX_HTML = '''
             <input type="file" id="file-upload" class="file-input" onchange="document.getElementById('fn').innerText=this.files[0].name">
             <label for="file-upload" class="file-label">📁 Choose Video File</label>
             <span id="fn" style="margin-bottom:10px; display:block; color:#666;">No file selected</span>
+            
+            <div class="page-input-group">
+                <label>📚 Number of Pages:</label>
+                <input type="number" id="page-count" value="4" min="1" max="10" placeholder="e.g. 4">
+            </div>
+            
             <button class="submit-btn" onclick="upload()">🚀 Generate Comic</button>
             <button id="restore-draft-btn" class="restore-btn" style="display:none; margin-top:15px;" onclick="restoreDraft()">📂 Restore Unsaved Draft</button>
+            
             <div class="load-section">
                 <h3>📥 Load Saved Comic</h3>
-                <p style="font-size:12px; color:#888; margin-bottom:10px;">Enter your save code to continue editing</p>
                 <div class="load-input-group">
                     <input type="text" id="load-code-input" placeholder="SAVE CODE" maxlength="8" style="text-transform:uppercase;">
                     <button onclick="loadSavedComic()">Load</button>
@@ -418,58 +419,25 @@ INDEX_HTML = '''
                 <label>💾 Save & Load:</label>
                 <button onclick="saveComic()" class="save-btn">💾 Save Comic (Get Code)</button>
                 <div id="current-save-code" style="display:none; margin-top:8px; padding:8px; background:#2ecc71; border-radius:4px; text-align:center;">
-                    <span style="font-size:11px;">Current Save Code:</span><br>
                     <span id="display-save-code" style="font-size:18px; font-weight:bold; letter-spacing:2px;"></span>
                     <button onclick="copyCode()" style="padding:4px 8px; margin-left:5px; font-size:10px;">📋 Copy</button>
                 </div>
             </div>
             <div class="control-group">
                 <label>💬 Bubble Tools:</label>
-                <select id="bubble-type-select" onchange="changeBubbleType(this.value)" disabled>
-                    <option value="speech">Speech</option>
-                    <option value="thought">Thought</option>
-                    <option value="reaction">Reaction</option>
-                    <option value="narration">Narration</option>
-                    <option value="idea">Idea</option>
-                </select>
-                <select id="font-select" onchange="changeFont(this.value)" disabled>
-                    <option value="'Comic Neue', cursive">Comic Neue</option>
-                    <option value="'Bangers', cursive">Bangers</option>
-                    <option value="'Gloria Hallelujah', cursive">Gloria</option>
-                    <option value="'Lato', sans-serif">Lato</option>
-                </select>
-                <div class="color-grid">
-                    <div><label>Text</label><input type="color" id="bubble-text-color" value="#ffffff" disabled></div>
-                    <div><label>Fill</label><input type="color" id="bubble-fill-color" value="#4ECDC4" disabled></div>
-                </div>
                 <button onclick="addBubble()" class="action-btn">💬 Add Bubble</button>
                 <button onclick="deleteBubble()" class="reset-btn">🗑️ Delete Bubble</button>
             </div>
-            <div class="control-group" id="tail-controls" style="display:none;">
-                <label>📐 Tail Adjustment:</label>
-                <button onclick="rotateTail()" class="secondary-btn">🔄 Rotate Side</button>
-                <div class="slider-container">
-                    <label>Pos:</label>
-                    <input type="range" id="tail-slider" min="10" max="90" value="50" oninput="slideTail(this.value)">
-                </div>
-            </div>
             <div class="control-group">
                 <label>🖼️ Panel Tools:</label>
                 <button onclick="replacePanelImage()" class="action-btn">🖼️ Replace Image</button>
                 <div class="button-grid">
-                    <button onclick="adjustFrame('backward')" class="secondary-btn">⬅️ Prev</button>
-                    <button onclick="adjustFrame('forward')" class="action-btn">Next ➡️</button>
+                    <button onclick="adjustFrame('backward')" class="secondary-btn" id="prev-btn">⬅️ Prev</button>
+                    <button onclick="adjustFrame('forward')" class="action-btn" id="next-btn">Next ➡️</button>
                 </div>
                 <div class="timestamp-controls">
                     <input type="text" id="timestamp-input" placeholder="mm:ss or secs">
-                    <button onclick="gotoTimestamp()" class="action-btn">Go</button>
-                </div>
-            </div>
-            <div class="control-group">
-                <label>🔍 Zoom & Pan:</label>
-                <div class="button-grid">
-                    <button onclick="resetPanelTransform()" class="secondary-btn">Reset</button>
-                    <input type="range" id="zoom-slider" min="100" max="300" value="100" step="5" disabled oninput="handleZoom(this)">
+                    <button onclick="gotoTimestamp()" class="action-btn" id="go-btn">Go</button>
                 </div>
             </div>
             <div class="control-group">
@@ -479,12 +447,11 @@ INDEX_HTML = '''
         </div>
     </div>
     
+    <!-- MODAL -->
     <div class="modal-overlay" id="save-modal">
         <div class="modal-content">
             <h2>✅ Comic Saved!</h2>
-            <p>Your unique save code is:</p>
             <div class="code" id="modal-save-code">XXXXXXXX</div>
-            <p style="font-size:12px;">Write this code down or copy it.<br>Anyone can load this comic using this code.</p>
             <button onclick="copyModalCode()">📋 Copy Code</button>
             <button class="close-btn" onclick="closeModal()">Close</button>
         </div>
@@ -494,53 +461,62 @@ INDEX_HTML = '''
         function genUUID(){ return 'xxxxxxxx-xxxx-4xxx-yxxx-xxxxxxxxxxxx'.replace(/[xy]/g,c=>{var r=Math.random()*16|0,v=c=='x'?r:(r&0x3|0x8);return v.toString(16);}); }
         let sid = localStorage.getItem('comic_sid') || genUUID();
         localStorage.setItem('comic_sid', sid);
+        
         let currentSaveCode = null;
+        let isProcessing = false; // Lock to prevent 429 errors
         let interval, selectedBubble = null, selectedPanel = null;
-        let isDragging = false, isResizing = false, isPanning = false;
-        let startX, startY, initX, initY, initW, initH;
-        let panStartX, panStartY, panStartTx, panStartTy;
+        let isDragging = false, isResizing = false;
+        let startX, startY, initX, initY;
         let resizeHandle = '', originalWidth, originalHeight, originalX, originalY, originalMouseX, originalMouseY;
         let currentlyEditing = null;
         
         if(localStorage.getItem('comic_draft_'+sid)) document.getElementById('restore-draft-btn').style.display = 'block';
         
+        // --- HELPERS ---
         function showSaveModal(code) { document.getElementById('modal-save-code').textContent = code; document.getElementById('save-modal').style.display = 'flex'; }
         function closeModal() { document.getElementById('save-modal').style.display = 'none'; }
-        function copyModalCode() { navigator.clipboard.writeText(document.getElementById('modal-save-code').textContent).then(() => alert('Code copied to clipboard!')); }
+        function copyModalCode() { navigator.clipboard.writeText(document.getElementById('modal-save-code').textContent).then(() => alert('Code copied!')); }
         function copyCode() { if(currentSaveCode) navigator.clipboard.writeText(currentSaveCode).then(() => alert('Code copied!')); }
-        
+        function setProcessing(busy) {
+            isProcessing = busy;
+            const btns = ['prev-btn', 'next-btn', 'go-btn'];
+            btns.forEach(id => {
+                const el = document.getElementById(id);
+                if(el) { el.disabled = busy; el.style.opacity = busy ? '0.5' : '1'; }
+            });
+        }
+
+        // --- CORE ACTIONS ---
         async function saveComic() {
             const state = getCurrentState();
             if(!state || state.length === 0) { alert('No comic to save!'); return; }
             try {
-                const response = await fetch(`/save_comic?sid=${sid}`, { method: 'POST', headers: {'Content-Type': 'application/json'}, body: JSON.stringify({ pages: state, savedAt: new Date().toISOString() }) });
-                const data = await response.json();
-                if(data.success) { currentSaveCode = data.code; document.getElementById('display-save-code').textContent = data.code; document.getElementById('current-save-code').style.display = 'block'; showSaveModal(data.code); saveDraft(); } 
-                else { alert('Failed to save: ' + data.message); }
+                const r = await fetch(`/save_comic?sid=${sid}`, { method: 'POST', headers: {'Content-Type': 'application/json'}, body: JSON.stringify({ pages: state, savedAt: new Date().toISOString() }) });
+                const d = await r.json();
+                if(d.success) { currentSaveCode = d.code; document.getElementById('display-save-code').textContent = d.code; document.getElementById('current-save-code').style.display = 'block'; showSaveModal(d.code); saveDraft(); } 
+                else { alert('Failed to save: ' + d.message); }
             } catch(e) { console.error(e); alert('Error saving comic'); }
         }
         
         async function loadSavedComic() {
             const code = document.getElementById('load-code-input').value.trim().toUpperCase();
-            if(!code || code.length < 4) { alert('Please enter a valid save code'); return; }
+            if(!code || code.length < 4) { alert('Invalid code'); return; }
             try {
-                const response = await fetch(`/load_comic/${code}`);
-                const data = await response.json();
-                if(data.success) { currentSaveCode = code; sid = data.originalSid || sid; localStorage.setItem('comic_sid', sid); renderFromState(data.pages); document.getElementById('upload-container').style.display = 'none'; document.getElementById('editor-container').style.display = 'block'; document.getElementById('display-save-code').textContent = code; document.getElementById('current-save-code').style.display = 'block'; saveDraft(); } 
-                else { alert('Could not load comic: ' + data.message); }
-            } catch(e) { console.error(e); alert('Error loading comic. Check the code and try again.'); }
+                const r = await fetch(`/load_comic/${code}`);
+                const d = await r.json();
+                if(d.success) { currentSaveCode = code; sid = d.originalSid || sid; localStorage.setItem('comic_sid', sid); renderFromState(d.pages); document.getElementById('upload-container').style.display = 'none'; document.getElementById('editor-container').style.display = 'block'; document.getElementById('display-save-code').textContent = code; document.getElementById('current-save-code').style.display = 'block'; saveDraft(); } 
+                else { alert('Load failed: ' + d.message); }
+            } catch(e) { console.error(e); alert('Error loading comic.'); }
         }
         
         function restoreDraft() {
-            const savedData = localStorage.getItem('comic_draft_'+sid);
-            if(!savedData) { alert("No draft found."); return; }
             try {
-                const state = JSON.parse(savedData);
+                const state = JSON.parse(localStorage.getItem('comic_draft_'+sid));
                 if(state.saveCode) { currentSaveCode = state.saveCode; document.getElementById('display-save-code').textContent = state.saveCode; document.getElementById('current-save-code').style.display = 'block'; }
                 renderFromState(state.pages || state);
                 document.getElementById('upload-container').style.display = 'none';
                 document.getElementById('editor-container').style.display = 'block';
-            } catch(e) { console.error(e); alert("Failed to restore draft."); }
+            } catch(e) { console.error(e); alert("Failed to restore."); }
         }
         
         function getCurrentState() {
@@ -552,9 +528,9 @@ INDEX_HTML = '''
                     const bubbles = [];
                     pan.querySelectorAll('.speech-bubble').forEach(b => {
                         const textEl = b.querySelector('.bubble-text');
-                        bubbles.push({ text: textEl ? textEl.textContent : '', left: b.style.left, top: b.style.top, width: b.style.width, height: b.style.height, classes: b.className, type: b.dataset.type, font: b.style.fontFamily, tailPos: b.style.getPropertyValue('--tail-pos'), colors: { fill: b.style.getPropertyValue('--bubble-fill-color'), text: b.style.getPropertyValue('--bubble-text-color') } });
+                        bubbles.push({ text: textEl ? textEl.textContent : '', left: b.style.left, top: b.style.top, width: b.style.width, height: b.style.height, classes: b.className });
                     });
-                    panels.push({ src: img.src, zoom: img.dataset.zoom, tx: img.dataset.translateX, ty: img.dataset.translateY, bubbles: bubbles });
+                    panels.push({ src: img.src, bubbles: bubbles });
                 });
                 pages.push({ panels: panels });
             });
@@ -575,8 +551,8 @@ INDEX_HTML = '''
                     const pDiv = document.createElement('div'); pDiv.className = 'panel';
                     pDiv.onclick = (e) => { e.stopPropagation(); selectPanel(pDiv); };
                     const img = document.createElement('img');
-                    img.src = pan.src; img.dataset.zoom = pan.zoom || 100; img.dataset.translateX = pan.tx || 0; img.dataset.translateY = pan.ty || 0;
-                    updateImageTransform(img); img.onmousedown = (e) => startPan(e, img);
+                    // Add timestamp to src to force refresh
+                    img.src = pan.src.includes('?') ? pan.src : pan.src + `?sid=${sid}`; 
                     pDiv.appendChild(img);
                     (pan.bubbles || []).forEach(bData => { pDiv.appendChild(createBubbleHTML(bData)); });
                     grid.appendChild(pDiv);
@@ -587,10 +563,17 @@ INDEX_HTML = '''
         
         async function upload() {
             const f = document.getElementById('file-upload').files[0];
-            if(!f) return alert("Select a video file first");
+            const pCount = document.getElementById('page-count').value;
+            if(!f) return alert("Select a video");
             sid = genUUID(); localStorage.setItem('comic_sid', sid); currentSaveCode = null;
-            document.querySelector('.upload-box').style.display='none'; document.getElementById('loading-view').style.display='flex';
-            const fd = new FormData(); fd.append('file', f);
+            
+            document.querySelector('.upload-box').style.display='none'; 
+            document.getElementById('loading-view').style.display='flex';
+            
+            const fd = new FormData(); 
+            fd.append('file', f);
+            fd.append('target_pages', pCount); // Send page count to backend
+            
             const r = await fetch(`/uploader?sid=${sid}`, {method:'POST', body:fd});
             if(r.ok) interval = setInterval(checkStatus, 2000);
             else { alert("Upload failed"); location.reload(); }
@@ -610,36 +593,26 @@ INDEX_HTML = '''
                 const cleanData = data.map((p, pi) => ({
                     panels: p.panels.map((pan, j) => ({
                         src: `/frames/${pan.image}?sid=${sid}`,
-                        bubbles: (p.bubbles && p.bubbles[j] && p.bubbles[j].dialog) ? [{ text: p.bubbles[j].dialog, left: (p.bubbles[j].bubble_offset_x || 50) + 'px', top: (p.bubbles[j].bubble_offset_y || 20) + 'px', type: 'speech', tailPos: '50%', colors: { text: '#ffffff', fill: '#4ECDC4' } }] : []
+                        bubbles: (p.bubbles && p.bubbles[j] && p.bubbles[j].dialog) ? [{ text: p.bubbles[j].dialog, left: (p.bubbles[j].bubble_offset_x || 50) + 'px', top: (p.bubbles[j].bubble_offset_y || 20) + 'px', classes: 'speech-bubble speech tail-bottom' }] : []
                     }))
                 }));
                 renderFromState(cleanData); saveDraft();
             });
         }
         
+        // --- INTERACTIVE ELEMENTS ---
         function createBubbleHTML(data) {
-            const b = document.createElement('div'); b.dataset.type = data.type || 'speech';
-            applyBubbleType(b, data.type || 'speech', data.classes);
+            const b = document.createElement('div');
+            b.className = data.classes || 'speech-bubble speech tail-bottom';
             b.style.left = data.left; b.style.top = data.top;
-            if(data.width) b.style.width = data.width; if(data.height) b.style.height = data.height; if(data.font) b.style.fontFamily = data.font;
-            if(data.colors) { b.style.setProperty('--bubble-fill-color', data.colors.fill || '#4ECDC4'); b.style.setProperty('--bubble-text-color', data.colors.text || '#ffffff'); }
-            if(data.tailPos) b.style.setProperty('--tail-pos', data.tailPos);
+            if(data.width) b.style.width = data.width; if(data.height) b.style.height = data.height;
             const textSpan = document.createElement('span'); textSpan.className = 'bubble-text'; textSpan.textContent = data.text || ''; b.appendChild(textSpan);
             ['nw', 'ne', 'sw', 'se'].forEach(dir => { const handle = document.createElement('div'); handle.className = `resize-handle ${dir}`; handle.onmousedown = (e) => startResize(e, dir); b.appendChild(handle); });
             b.onmousedown = (e) => { if(e.target.classList.contains('resize-handle')) return; e.stopPropagation(); selectBubble(b); isDragging = true; startX = e.clientX; startY = e.clientY; initX = b.offsetLeft; initY = b.offsetTop; };
-            b.ondblclick = (e) => { e.stopPropagation(); editBubbleText(b); }; b.onclick = (e) => { e.stopPropagation(); selectBubble(b); };
+            b.ondblclick = (e) => { e.stopPropagation(); editBubbleText(b); };
             return b;
         }
         
-        function applyBubbleType(bubble, type, existingClasses) {
-            bubble.querySelectorAll('.thought-dot').forEach(el => el.remove());
-            let baseClasses = 'speech-bubble ' + type;
-            if (type === 'speech') baseClasses += ' tail-bottom';
-            if (existingClasses && existingClasses.includes('selected')) baseClasses += ' selected';
-            bubble.className = baseClasses; bubble.dataset.type = type;
-            if (type === 'thought') { for (let i = 1; i <= 2; i++) { const dot = document.createElement('div'); dot.className = `thought-dot thought-dot-${i}`; bubble.appendChild(dot); } }
-        }
-        
         function editBubbleText(bubble) {
             if (currentlyEditing) return; currentlyEditing = bubble;
             const textSpan = bubble.querySelector('.bubble-text'); const textarea = document.createElement('textarea');
@@ -649,30 +622,94 @@ INDEX_HTML = '''
             textarea.addEventListener('keydown', e => { if (e.key === 'Enter' && !e.shiftKey) { e.preventDefault(); textarea.blur(); } });
         }
         
-        document.addEventListener('mousemove', (e) => { if(isDragging && selectedBubble) { selectedBubble.style.left = (initX + e.clientX - startX) + 'px'; selectedBubble.style.top = (initY + e.clientY - startY) + 'px'; } if(isResizing && selectedBubble) { resizeBubble(e); } if(isPanning && selectedPanel) { panImage(e); } });
-        document.addEventListener('mouseup', () => { if(isDragging || isResizing || isPanning) saveDraft(); isDragging = false; isResizing = false; isPanning = false; });
-        function startResize(e, dir) { e.preventDefault(); e.stopPropagation(); isResizing = true; resizeHandle = dir; const rect = selectedBubble.getBoundingClientRect(); originalWidth = rect.width; originalHeight = rect.height; originalX = selectedBubble.offsetLeft; originalY = selectedBubble.offsetTop; originalMouseX = e.clientX; originalMouseY = e.clientY; }
-        function resizeBubble(e) { if (!isResizing || !selectedBubble) return; const dx = e.clientX - originalMouseX; const dy = e.clientY - originalMouseY; if (resizeHandle.includes('e')) selectedBubble.style.width = `${originalWidth + dx}px`; if (resizeHandle.includes('w')) { selectedBubble.style.width = `${originalWidth - dx}px`; selectedBubble.style.left = `${originalX + dx}px`; } if (resizeHandle.includes('s')) selectedBubble.style.height = `${originalHeight + dy}px`; if (resizeHandle.includes('n')) { selectedBubble.style.height = `${originalHeight - dy}px`; selectedBubble.style.top = `${originalY + dy}px`; } }
-        function selectBubble(el) { if(selectedBubble) selectedBubble.classList.remove('selected'); if(selectedPanel) { selectedPanel.classList.remove('selected'); selectedPanel = null; } selectedBubble = el; el.classList.add('selected'); const bubbleType = el.dataset.type || 'speech'; document.getElementById('tail-controls').style.display = (bubbleType === 'speech' || bubbleType === 'thought') ? 'block' : 'none'; ['bubble-text-color','bubble-fill-color','bubble-type-select','font-select'].forEach(i => document.getElementById(i).disabled = false); document.getElementById('zoom-slider').disabled = true; document.getElementById('bubble-type-select').value = bubbleType; const styles = window.getComputedStyle(el); document.getElementById('bubble-text-color').value = rgbToHex(styles.getPropertyValue('--bubble-text-color').trim() || styles.color); document.getElementById('bubble-fill-color').value = rgbToHex(styles.getPropertyValue('--bubble-fill-color').trim() || styles.backgroundColor); document.getElementById('tail-slider').value = parseInt(styles.getPropertyValue('--tail-pos').trim() || 50); }
-        function selectPanel(el) { if(selectedPanel) selectedPanel.classList.remove('selected'); if(selectedBubble) { selectedBubble.classList.remove('selected'); selectedBubble = null; } selectedPanel = el; el.classList.add('selected'); const img = el.querySelector('img'); document.getElementById('zoom-slider').disabled = false; document.getElementById('zoom-slider').value = img.dataset.zoom || 100; ['bubble-text-color','bubble-fill-color','bubble-type-select','font-select'].forEach(i => document.getElementById(i).disabled = true); document.getElementById('tail-controls').style.display = 'none'; }
-        function addBubble() { if(!selectedPanel) return alert("Select a panel first"); const b = createBubbleHTML({ text: "New Text", left: "50px", top: "30px", type: 'speech', colors: { text: '#ffffff', fill: '#4ECDC4' } }); selectedPanel.appendChild(b); selectBubble(b); saveDraft(); }
-        function deleteBubble() { if(!selectedBubble) return alert("Select a bubble first"); if(confirm("Delete this bubble?")) { selectedBubble.remove(); selectedBubble = null; saveDraft(); } }
-        function changeBubbleType(type) { if(!selectedBubble) return; applyBubbleType(selectedBubble, type); selectedBubble.classList.add('selected'); document.getElementById('tail-controls').style.display = (type === 'speech' || type === 'thought') ? 'block' : 'none'; saveDraft(); }
-        function changeFont(font) { if(!selectedBubble) return; selectedBubble.style.fontFamily = font; saveDraft(); }
-        function rotateTail() { if(!selectedBubble) return; const type = selectedBubble.dataset.type; if(type === 'speech') { const positions = ['tail-bottom', 'tail-right', 'tail-top', 'tail-left']; let current = 0; positions.forEach((pos, i) => { if(selectedBubble.classList.contains(pos)) current = i; }); selectedBubble.classList.remove(positions[current]); selectedBubble.classList.add(positions[(current + 1) % 4]); } else if(type === 'thought') { const isFlippedH = selectedBubble.classList.contains('flipped'); const isFlippedV = selectedBubble.classList.contains('flipped-vertical'); if (!isFlippedH && !isFlippedV) selectedBubble.classList.add('flipped'); else if (isFlippedH && !isFlippedV) selectedBubble.classList.add('flipped-vertical'); else if (isFlippedH && isFlippedV) selectedBubble.classList.remove('flipped'); else selectedBubble.classList.remove('flipped-vertical'); } saveDraft(); }
-        function slideTail(v) { if(selectedBubble) { selectedBubble.style.setProperty('--tail-pos', v+'%'); saveDraft(); } }
-        document.getElementById('bubble-text-color').addEventListener('input', (e) => { if(selectedBubble) { selectedBubble.style.setProperty('--bubble-text-color', e.target.value); saveDraft(); } });
-        document.getElementById('bubble-fill-color').addEventListener('input', (e) => { if(selectedBubble) { selectedBubble.style.setProperty('--bubble-fill-color', e.target.value); saveDraft(); } });
-        function handleZoom(el) { if(!selectedPanel) return; const img = selectedPanel.querySelector('img'); img.dataset.zoom = el.value; updateImageTransform(img); saveDraft(); }
-        function startPan(e, img) { if(parseFloat(img.dataset.zoom || 100) <= 100) return; e.preventDefault(); isPanning = true; selectedPanel = img.closest('.panel'); panStartX = e.clientX; panStartY = e.clientY; panStartTx = parseFloat(img.dataset.translateX || 0); panStartTy = parseFloat(img.dataset.translateY || 0); img.classList.add('panning'); }
-        function panImage(e) { if(!isPanning || !selectedPanel) return; const img = selectedPanel.querySelector('img'); img.dataset.translateX = panStartTx + (e.clientX - panStartX); img.dataset.translateY = panStartTy + (e.clientY - panStartY); updateImageTransform(img); }
-        function updateImageTransform(img) { const z = (img.dataset.zoom || 100) / 100; const x = img.dataset.translateX || 0; const y = img.dataset.translateY || 0; img.style.transform = `translateX(${x}px) translateY(${y}px) scale(${z})`; img.classList.toggle('pannable', z > 1); }
-        function resetPanelTransform() { if(!selectedPanel) return alert("Select a panel first"); const img = selectedPanel.querySelector('img'); img.dataset.zoom = 100; img.dataset.translateX = 0; img.dataset.translateY = 0; document.getElementById('zoom-slider').value = 100; updateImageTransform(img); saveDraft(); }
-        function replacePanelImage() { if(!selectedPanel) return alert("Select a panel first"); const inp = document.getElementById('image-uploader'); inp.onchange = async (e) => { const fd = new FormData(); fd.append('image', e.target.files[0]); const img = selectedPanel.querySelector('img'); img.style.opacity = '0.5'; const r = await fetch(`/replace_panel?sid=${sid}`, {method:'POST', body:fd}); const d = await r.json(); if(d.success) { img.src = `/frames/${d.new_filename}?sid=${sid}`; resetPanelTransform(); saveDraft(); } else { alert('Error: ' + d.error); } img.style.opacity = '1'; inp.value = ''; }; inp.click(); }
-        async function adjustFrame(dir) { if(!selectedPanel) return alert("Select a panel first"); const img = selectedPanel.querySelector('img'); let fname = img.src.split('/').pop().split('?')[0]; img.style.opacity = '0.5'; const r = await fetch(`/regenerate_frame?sid=${sid}`, { method:'POST', headers:{'Content-Type':'application/json'}, body:JSON.stringify({filename:fname, direction:dir}) }); const d = await r.json(); if(d.success) { img.src = `/frames/${fname}?sid=${sid}&t=${Date.now()}`; } else { alert('Error: ' + d.message); } img.style.opacity = '1'; saveDraft(); }
-        async function gotoTimestamp() { if(!selectedPanel) return alert("Select a panel first"); let v = document.getElementById('timestamp-input').value.trim(); if(!v) return; if(v.includes(':')) { let p = v.split(':'); v = parseInt(p[0]) * 60 + parseFloat(p[1]); } else { v = parseFloat(v); } if(isNaN(v)) return alert("Invalid time format"); const img = selectedPanel.querySelector('img'); let fname = img.src.split('/').pop().split('?')[0]; img.style.opacity = '0.5'; const r = await fetch(`/goto_timestamp?sid=${sid}`, { method:'POST', headers:{'Content-Type':'application/json'}, body:JSON.stringify({filename:fname, timestamp:v}) }); const d = await r.json(); if(d.success) { img.src = `/frames/${fname}?sid=${sid}&t=${Date.now()}`; document.getElementById('timestamp-input').value = ''; resetPanelTransform(); } else { alert('Error: ' + d.message); } img.style.opacity = '1'; saveDraft(); }
-        async function exportComic() { const pgs = document.querySelectorAll('.comic-page'); if(pgs.length === 0) return alert("No pages found"); const bubbles = document.querySelectorAll('.speech-bubble'); bubbles.forEach(b => { const rect = b.getBoundingClientRect(); b.style.width = rect.width + 'px'; b.style.height = rect.height + 'px'; }); alert(`Exporting ${pgs.length} page(s)...`); for(let i = 0; i < pgs.length; i++) { try { const u = await htmlToImage.toPng(pgs[i], {pixelRatio: 3}); const a = document.createElement('a'); a.href = u; a.download = `Comic-Page-${i+1}.png`; a.click(); } catch(err) { console.error(err); alert(`Failed to export page ${i+1}`); } } bubbles.forEach(b => { b.style.width = ''; b.style.height = ''; }); }
-        function rgbToHex(rgb) { if (!rgb || !rgb.startsWith('rgb')) return '#ffffff'; let sep = rgb.indexOf(",") > -1 ? "," : " "; rgb = rgb.substr(4).split(")")[0].split(sep); let r = (+rgb[0]).toString(16), g = (+rgb[1]).toString(16), b = (+rgb[2]).toString(16); if (r.length == 1) r = "0" + r; if (g.length == 1) g = "0" + g; if (b.length == 1) b = "0" + b; return "#" + r + g + b; }
+        document.addEventListener('mousemove', (e) => { if(isDragging && selectedBubble) { selectedBubble.style.left = (initX + e.clientX - startX) + 'px'; selectedBubble.style.top = (initY + e.clientY - startY) + 'px'; } if(isResizing && selectedBubble) { resizeBubble(e); } });
+        document.addEventListener('mouseup', () => { if(isDragging || isResizing) saveDraft(); isDragging = false; isResizing = false; });
+        function startResize(e, dir) { e.preventDefault(); e.stopPropagation(); isResizing = true; resizeHandle = dir; const rect = selectedBubble.getBoundingClientRect(); originalWidth = rect.width; originalHeight = rect.height; originalMouseX = e.clientX; originalMouseY = e.clientY; }
+        function resizeBubble(e) { if (!isResizing || !selectedBubble) return; const dx = e.clientX - originalMouseX; const dy = e.clientY - originalMouseY; if(resizeHandle.includes('e')) selectedBubble.style.width = (originalWidth + dx)+'px'; if(resizeHandle.includes('s')) selectedBubble.style.height = (originalHeight + dy)+'px'; }
+        
+        function selectBubble(el) { if(selectedBubble) selectedBubble.classList.remove('selected'); if(selectedPanel) { selectedPanel.classList.remove('selected'); selectedPanel = null; } selectedBubble = el; el.classList.add('selected'); }
+        function selectPanel(el) { if(selectedPanel) selectedPanel.classList.remove('selected'); if(selectedBubble) { selectedBubble.classList.remove('selected'); selectedBubble = null; } selectedPanel = el; el.classList.add('selected'); }
+        function addBubble() { if(!selectedPanel) return alert("Select a panel first"); const b = createBubbleHTML({ text: "Text", left: "50px", top: "30px", classes: "speech-bubble speech tail-bottom" }); selectedPanel.appendChild(b); selectBubble(b); saveDraft(); }
+        function deleteBubble() { if(!selectedBubble) return alert("Select a bubble"); selectedBubble.remove(); selectedBubble=null; saveDraft(); }
+        
+        function replacePanelImage() { 
+            if(!selectedPanel) return alert("Select a panel"); 
+            const inp = document.getElementById('image-uploader'); 
+            inp.onchange = async (e) => { 
+                const fd = new FormData(); fd.append('image', e.target.files[0]); 
+                const img = selectedPanel.querySelector('img'); 
+                const r = await fetch(`/replace_panel?sid=${sid}`, {method:'POST', body:fd}); 
+                const d = await r.json(); 
+                if(d.success) { img.src = `/frames/${d.new_filename}?sid=${sid}`; saveDraft(); } 
+                inp.value = ''; 
+            }; 
+            inp.click(); 
+        }
+        
+        async function adjustFrame(dir) { 
+            if(isProcessing) return; // LOCK
+            if(!selectedPanel) return alert("Select a panel"); 
+            const img = selectedPanel.querySelector('img'); 
+            let fname = img.src.split('/').pop().split('?')[0]; 
+            
+            setProcessing(true); // ENABLE LOCK
+            img.style.opacity = '0.5';
+            
+            try {
+                const r = await fetch(`/regenerate_frame?sid=${sid}`, { method:'POST', headers:{'Content-Type':'application/json'}, body:JSON.stringify({filename:fname, direction:dir}) }); 
+                const d = await r.json(); 
+                if(d.success) { img.src = `/frames/${fname}?sid=${sid}&t=${Date.now()}`; } 
+                else { alert('Error: ' + d.message); }
+            } catch(e) { console.error(e); }
+            
+            img.style.opacity = '1';
+            setProcessing(false); // RELEASE LOCK
+            saveDraft(); 
+        }
+        
+        async function gotoTimestamp() { 
+            if(isProcessing) return; // LOCK
+            if(!selectedPanel) return alert("Select a panel"); 
+            let v = document.getElementById('timestamp-input').value.trim(); 
+            if(!v) return;
+            if(v.includes(':')) { let p = v.split(':'); v = parseInt(p[0]) * 60 + parseFloat(p[1]); } else { v = parseFloat(v); }
+            if(isNaN(v)) return alert("Invalid time"); 
+            
+            const img = selectedPanel.querySelector('img'); 
+            let fname = img.src.split('/').pop().split('?')[0]; 
+            
+            setProcessing(true); // ENABLE LOCK
+            img.style.opacity = '0.5';
+            
+            try {
+                const r = await fetch(`/goto_timestamp?sid=${sid}`, { method:'POST', headers:{'Content-Type':'application/json'}, body:JSON.stringify({filename:fname, timestamp:v}) }); 
+                const d = await r.json(); 
+                if(d.success) { img.src = `/frames/${fname}?sid=${sid}&t=${Date.now()}`; document.getElementById('timestamp-input').value = ''; } 
+                else { alert('Error: ' + d.message); }
+            } catch(e) { console.error(e); }
+            
+            img.style.opacity = '1';
+            setProcessing(false); // RELEASE LOCK
+            saveDraft(); 
+        }
+        
+        async function exportComic() { 
+            const pgs = document.querySelectorAll('.comic-page'); 
+            if(pgs.length === 0) return alert("No pages found"); 
+            alert(`Exporting ${pgs.length} page(s)...`); 
+            for(let i = 0; i < pgs.length; i++) { 
+                try { const u = await htmlToImage.toPng(pgs[i], {pixelRatio: 2}); const a = document.createElement('a'); a.href = u; a.download = `Comic-Page-${i+1}.png`; a.click(); } 
+                catch(err) { console.error(err); alert(`Failed to export page ${i+1}`); } 
+            } 
+        }
+        
+        function goBackToUpload() {
+            if(confirm('Go home? Unsaved changes will be lost.')) {
+                document.getElementById('editor-container').style.display = 'none';
+                document.getElementById('upload-container').style.display = 'flex';
+                document.getElementById('loading-view').style.display = 'none';
+            }
+        }
     </script>
 </body>
 </html>
@@ -689,13 +726,16 @@ def upload():
     if 'file' not in request.files or not request.files['file'].filename:
         return jsonify({'success': False, 'message': 'No file selected'}), 400
     
+    # GET PAGE COUNT FROM FORM
+    target_pages = request.form.get('target_pages', 4)
+    
     f = request.files['file']
     gen = EnhancedComicGenerator(sid)
     gen.cleanup()
     f.save(gen.video_path)
     gen.write_status("Starting...", 5)
     
-    threading.Thread(target=gen.run).start()
+    threading.Thread(target=gen.run, args=(target_pages,)).start()
     return jsonify({'success': True, 'message': 'Generation started.'})
 
 @app.route('/status')
@@ -720,7 +760,6 @@ def regen():
     sid = request.args.get('sid')
     d = request.get_json()
     gen = EnhancedComicGenerator(sid)
-    # CALL GLOBAL GPU FUNCTION
     return jsonify(regen_frame_gpu(gen.video_path, gen.frames_dir, gen.metadata_path, d['filename'], d['direction']))
 
 @app.route('/goto_timestamp', methods=['POST'])
@@ -728,7 +767,6 @@ def go_time():
     sid = request.args.get('sid')
     d = request.get_json()
     gen = EnhancedComicGenerator(sid)
-    # CALL GLOBAL GPU FUNCTION
     return jsonify(get_frame_at_ts_gpu(gen.video_path, gen.frames_dir, gen.metadata_path, d['filename'], float(d['timestamp'])))
 
 @app.route('/replace_panel', methods=['POST'])