Update app_enhanced.py

app_enhanced.py

@@ -28,22 +28,22 @@ def gpu_warmup():
 # ======================================================
 def bubble(dialog="", bubble_offset_x=50, bubble_offset_y=20, lip_x=-1, lip_y=-1, emotion='normal', type='speech'):
     return {
-        'dialog': dialog, 
-        'bubble_offset_x': int(bubble_offset_x), 
+        'dialog': dialog,
+        'bubble_offset_x': int(bubble_offset_x),
         'bubble_offset_y': int(bubble_offset_y),
-        'lip_x': int(lip_x), 
-        'lip_y': int(lip_y), 
+        'lip_x': int(lip_x),
+        'lip_y': int(lip_y),
         'emotion': emotion,
         'type': type,
         'tail_pos': '50%',
         'classes': f'speech-bubble {type} tail-bottom'
     }
 
-def panel(image=""): 
+def panel(image=""):
     return {'image': image}
 
 class Page:
-    def __init__(self, panels, bubbles): 
+    def __init__(self, panels, bubbles):
         self.panels = panels
         self.bubbles = bubbles
 
@@ -54,7 +54,7 @@ logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
 app = Flask(__name__)
-BASE_USER_DIR = "userdata" 
+BASE_USER_DIR = "userdata"
 SAVED_COMICS_DIR = "saved_comics"
 
 os.makedirs(BASE_USER_DIR, exist_ok=True)
@@ -70,7 +70,6 @@ def generate_save_code(length=8):
 # ======================================================
 # 🧠 GLOBAL GPU FUNCTIONS
 # ======================================================
-
 @spaces.GPU(duration=300)
 def generate_comic_gpu(video_path, user_dir, frames_dir, metadata_path, target_pages):
     print(f"🚀 GPU Task Started: {video_path} | Pages: {target_pages}")
@@ -84,7 +83,7 @@ def generate_comic_gpu(video_path, user_dir, frames_dir, metadata_path, target_p
     from backend.subtitles.subs_real import get_real_subtitles
     from backend.ai_bubble_placement import ai_bubble_placer
     from backend.ai_enhanced_core import face_detector
-    
+
     # 1. Analyze Video
     cap = cv2.VideoCapture(video_path)
     if not cap.isOpened(): raise Exception("Cannot open video")
@@ -92,7 +91,7 @@ def generate_comic_gpu(video_path, user_dir, frames_dir, metadata_path, target_p
     total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
     duration = total_frames / fps
     cap.release()
-    
+
     # 2. Subtitles Generation
     user_srt = os.path.join(user_dir, 'subs.srt')
     try:
@@ -103,16 +102,19 @@ def generate_comic_gpu(video_path, user_dir, frames_dir, metadata_path, target_p
         with open(user_srt, 'w') as f: f.write("1\n00:00:01,000 --> 00:00:04,000\n...\n")
             
     with open(user_srt, 'r', encoding='utf-8') as f:
-        all_subs = list(srt.parse(f.read()))
-    
+        try:
+            all_subs = list(srt.parse(f.read()))
+        except:
+            all_subs = []
+
     # 3. Smart Keyframe Selection
     valid_subs = [s for s in all_subs if s.content.strip()]
     raw_moments = [{'text': s.content, 'start': s.start.total_seconds(), 'end': s.end.total_seconds()} for s in valid_subs]
-    
+
     if target_pages <= 0: target_pages = 1
     panels_per_page = 4
     total_panels_needed = target_pages * panels_per_page
-    
+
     selected_moments = []
     if not raw_moments:
         times = np.linspace(1, duration-1, total_panels_needed)
@@ -128,7 +130,7 @@ def generate_comic_gpu(video_path, user_dir, frames_dir, metadata_path, target_p
     cap = cv2.VideoCapture(video_path)
     count = 0
     frame_files_ordered = []
-    
+
     for i, moment in enumerate(selected_moments):
         mid = (moment['start'] + moment['end']) / 2
         if mid > duration: mid = duration - 1
@@ -143,16 +145,16 @@ def generate_comic_gpu(video_path, user_dir, frames_dir, metadata_path, target_p
             frame_files_ordered.append(fname)
             count += 1
     cap.release()
-    
+
     with open(metadata_path, 'w') as f: json.dump(frame_metadata, f, indent=2)
         
     # 5. Image Enhancement
     try: black_bar_crop() 
     except: pass
-    
+
     se = SimpleColorEnhancer()
     qe = QualityColorEnhancer()
-    
+
     for f in frame_files_ordered:
         p = os.path.join(frames_dir, f)
         try: se.enhance_single(p, p)
@@ -208,17 +210,17 @@ def regen_frame_gpu(video_path, frames_dir, metadata_path, fname, direction):
     if not os.path.exists(metadata_path): return {"success": False, "message": "No metadata"}
     with open(metadata_path, 'r') as f: meta = json.load(f)
     if fname not in meta: return {"success": False, "message": "Frame not found"}
-    
+
     t = meta[fname]['time'] if isinstance(meta[fname], dict) else meta[fname]
     cap = cv2.VideoCapture(video_path)
     fps = cap.get(cv2.CAP_PROP_FPS) or 25
     offset = (1.0/fps) * (1 if direction == 'forward' else -1)
     new_t = max(0, t + offset)
-    
+
     cap.set(cv2.CAP_PROP_POS_MSEC, new_t * 1000)
     ret, frame = cap.read()
     cap.release()
-    
+
     if ret:
         p = os.path.join(frames_dir, fname)
         cv2.imwrite(p, frame)
@@ -242,7 +244,7 @@ def get_frame_at_ts_gpu(video_path, frames_dir, metadata_path, fname, ts):
     cap.set(cv2.CAP_PROP_POS_MSEC, float(ts) * 1000)
     ret, frame = cap.read()
     cap.release()
-    
+
     if ret:
         p = os.path.join(frames_dir, fname)
         cv2.imwrite(p, frame)
@@ -297,559 +299,636 @@ class EnhancedComicGenerator:
 # ======================================================
 # 🌐 ROUTES & FULL UI
 # ======================================================
-
 INDEX_HTML = '''
-<!DOCTYPE html>
-<html lang="en">
-<head>
-    <meta charset="UTF-8">
-    <meta name="viewport" content="width=device-width, initial-scale=1.0">
-    <title>🎬 Enhanced Comic Generator</title>
-    <script src="https://cdnjs.cloudflare.com/ajax/libs/html-to-image/1.11.11/html-to-image.min.js"></script>
-    <link href="https://fonts.googleapis.com/css2?family=Bangers&family=Comic+Neue:wght@700&family=Gloria+Hallelujah&family=Lato&display=swap" rel="stylesheet">
-    <style>
-        * { box-sizing: border-box; }
-        body { background-color: #fdf6e3; font-family: 'Lato', sans-serif; color: #3d3d3d; margin: 0; min-height: 100vh; }
-        
-        #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-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-group { margin: 20px 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: 12px; border: 2px solid #ddd; border-radius: 8px; font-size: 16px; box-sizing: border-box; }
-        
-        .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 LAYOUT */
-        .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; }
-        .comic-page { background: white; width: 600px; height: 400px; box-shadow: 0 4px 10px rgba(0,0,0,0.1); position: relative; overflow: hidden; border: 2px solid #000; padding: 10px; }
-        .comic-grid { display: grid; grid-template-columns: 285px 285px; grid-template-rows: 185px 185px; gap: 10px; width: 100%; height: 100%; }
-        .panel { position: relative; overflow: hidden; border: 2px solid #000; background: #eee; cursor: pointer; }
-        .panel.selected { outline: 3px solid #2196F3; outline-offset: -3px; }
-        .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; }
-        
-        /* SPEECH 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;
-            --tail-pos: 50%;
-        }
-        /* POINTER EVENTS NONE ON TEXT TO FIX SELECTION BUG */
-        .bubble-text { padding: 0.8em; word-wrap: break-word; position: relative; z-index: 5; pointer-events: none; user-select: none; }
-        .speech-bubble.selected { outline: 2px dashed #4CAF50; z-index: 100; }
-        .speech-bubble textarea { position: absolute; top:0; left:0; width:100%; height:100%; box-sizing:border-box; border:1px solid #4CAF50; background:rgba(255,255,255,0.95); text-align:center; padding:8px; z-index:102; resize:none; font: inherit; }
-        
-        /* SPEECH BUBBLE CSS (Tails) */
-        .speech-bubble.speech {
-            --b: 3em; --h: 1.8em; --t: 0.6; --p: var(--tail-pos, 50%); --r: 1.2em;
-            background: var(--bubble-fill-color, #4ECDC4);
-            color: var(--bubble-text-color, #fff);
-            padding: 1em;
-            border-radius: var(--r) var(--r) min(var(--r), calc(100% - var(--p) - (1 - var(--t)) * var(--b) / 2)) min(var(--r), calc(var(--p) - (1 - var(--t)) * var(--b) / 2)) / var(--r);
-        }
-        .speech-bubble.speech:before {
-            content: ""; position: absolute; width: var(--b); height: var(--h);
-            background: inherit; border-bottom-left-radius: 100%; pointer-events: none; z-index: 1;
-            -webkit-mask: radial-gradient(calc(var(--t)*100%) 105% at 100% 0,#0000 99%,#000 101%);
-            mask: radial-gradient(calc(var(--t)*100%) 105% at 100% 0,#0000 99%,#000 101%);
-        }
+<!DOCTYPE html><html lang="en"> <head> <meta charset="UTF-8"> <meta name="viewport" content="width=device-width, initial-scale=1.0"> <title>🎬 Enhanced Comic Generator</title> <script src="https://cdnjs.cloudflare.com/ajax/libs/html-to-image/1.11.11/html-to-image.min.js"></script> <link href="https://fonts.googleapis.com/css2?family=Bangers&family=Comic+Neue:wght@700&family=Gloria+Hallelujah&family=Lato&display=swap" rel="stylesheet"> <style> * { box-sizing: border-box; } body { background-color: #fdf6e3; font-family: 'Lato', sans-serif; color: #3d3d3d; margin: 0; min-height: 100vh; }
+    
+    #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-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-group { margin: 20px 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: 12px; border: 2px solid #ddd; border-radius: 8px; font-size: 16px; box-sizing: border-box; }
+    
+    .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 LAYOUT */
+    .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; }
+    .comic-page { background: white; width: 600px; height: 400px; box-shadow: 0 4px 10px rgba(0,0,0,0.1); position: relative; overflow: hidden; border: 2px solid #000; padding: 10px; }
+    .comic-grid { display: grid; grid-template-columns: 285px 285px; grid-template-rows: 185px 185px; gap: 10px; width: 100%; height: 100%; }
+    .panel { position: relative; overflow: hidden; border: 2px solid #000; background: #eee; cursor: pointer; }
+    .panel.selected { outline: 3px solid #2196F3; outline-offset: -3px; }
+    .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; }
+    
+    /* SPEECH 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;
+        --tail-pos: 50%;
+    }
+    /* FIX FOR MISMATCH: Ensure pre-wrap so spaces and newlines render identically */
+    .bubble-text { padding: 0.8em; word-wrap: break-word; white-space: pre-wrap; position: relative; z-index: 5; pointer-events: none; user-select: none; width: 100%; }
+    
+    .speech-bubble.selected { outline: 2px dashed #4CAF50; z-index: 100; }
+    .speech-bubble textarea { position: absolute; top:0; left:0; width:100%; height:100%; box-sizing:border-box; border:1px solid #4CAF50; background:rgba(255,255,255,0.95); text-align:center; padding:8px; z-index:102; resize:none; font: inherit; white-space: pre-wrap; }
+    
+    /* SPEECH BUBBLE CSS (Tails) */
+    .speech-bubble.speech {
+        --b: 3em; --h: 1.8em; --t: 0.6; --p: var(--tail-pos, 50%); --r: 1.2em;
+        background: var(--bubble-fill-color, #4ECDC4);
+        color: var(--bubble-text-color, #fff);
+        padding: 1em;
+        border-radius: var(--r) var(--r) min(var(--r), calc(100% - var(--p) - (1 - var(--t)) * var(--b) / 2)) min(var(--r), calc(var(--p) - (1 - var(--t)) * var(--b) / 2)) / var(--r);
+    }
+    .speech-bubble.speech:before {
+        content: ""; position: absolute; width: var(--b); height: var(--h);
+        background: inherit; border-bottom-left-radius: 100%; pointer-events: none; z-index: 1;
+        -webkit-mask: radial-gradient(calc(var(--t)*100%) 105% at 100% 0,#0000 99%,#000 101%);
+        mask: radial-gradient(calc(var(--t)*100%) 105% at 100% 0,#0000 99%,#000 101%);
+    }
+    
+    .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; }
+    /* THOUGHT BUBBLE CSS (Fixed Rotation) */
+    .speech-bubble.thought { background: white; border: 2px dashed #555; color: #333; border-radius: 50%; }
+    .speech-bubble.thought::before { 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; }
+    .thought-dot-2 { width: 12px; height: 12px; }
+    
+    /* Thought Tail Positions */
+    .speech-bubble.thought.pos-bl .thought-dot-1 { left: 20px; bottom: -20px; }
+    .speech-bubble.thought.pos-bl .thought-dot-2 { left: 10px; bottom: -32px; }
+    
+    .speech-bubble.thought.pos-br .thought-dot-1 { right: 20px; bottom: -20px; }
+    .speech-bubble.thought.pos-br .thought-dot-2 { right: 10px; bottom: -32px; }
+    
+    .speech-bubble.thought.pos-tr .thought-dot-1 { right: 20px; top: -20px; }
+    .speech-bubble.thought.pos-tr .thought-dot-2 { right: 10px; top: -32px; }
+    
+    .speech-bubble.thought.pos-tl .thought-dot-1 { left: 20px; top: -20px; }
+    .speech-bubble.thought.pos-tl .thought-dot-2 { left: 10px; top: -32px; }
+    
+    .speech-bubble.reaction { background: #FFD700; border: 3px solid #E53935; color: #D32F2F; font-weight: 900; text-transform: uppercase; 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; }
+    
+    .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-grid { display: grid; grid-template-columns: 1fr 1fr; gap: 5px; }
+    .color-grid { display: grid; grid-template-columns: 1fr 1fr; gap: 5px; }
+    .slider-container { display: flex; align-items: center; gap: 5px; margin-top: 5px; }
+    .slider-container label { min-width: 40px; font-size: 11px; }
+    .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; }
+    .undo-btn { background: #7f8c8d; color: white; margin-bottom: 5px; }
+    
+    /* 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 .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 button { background: #3498db; color: white; border: none; padding: 12px 30px; border-radius: 8px; cursor: pointer; font-weight: bold; margin: 5px; }
+</style>
+</head> <body> <div id="upload-container"> <div class="upload-box"> <h1>🎬 Enhanced Comic Generator</h1> <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>📚 Total Comic Pages:</label>
+            <input type="number" id="page-count" value="4" min="1" max="15" placeholder="e.g. 4 (Video will be divided evenly)">
+            <small style="color:#666; font-size:11px; display:block; margin-top:5px;">System calculates ~4 panels per page.</small>
+        </div>
         
-        .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; }
-
-        /* THOUGHT BUBBLE CSS (Fixed Rotation) */
-        .speech-bubble.thought { background: white; border: 2px dashed #555; color: #333; border-radius: 50%; }
-        .speech-bubble.thought::before { 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; }
-        .thought-dot-2 { width: 12px; height: 12px; }
-        
-        /* Thought Tail Positions */
-        .speech-bubble.thought.pos-bl .thought-dot-1 { left: 20px; bottom: -20px; }
-        .speech-bubble.thought.pos-bl .thought-dot-2 { left: 10px; bottom: -32px; }
-        
-        .speech-bubble.thought.pos-br .thought-dot-1 { right: 20px; bottom: -20px; }
-        .speech-bubble.thought.pos-br .thought-dot-2 { right: 10px; bottom: -32px; }
-        
-        .speech-bubble.thought.pos-tr .thought-dot-1 { right: 20px; top: -20px; }
-        .speech-bubble.thought.pos-tr .thought-dot-2 { right: 10px; top: -32px; }
-        
-        .speech-bubble.thought.pos-tl .thought-dot-1 { left: 20px; top: -20px; }
-        .speech-bubble.thought.pos-tl .thought-dot-2 { left: 10px; top: -32px; }
-        
-        .speech-bubble.reaction { background: #FFD700; border: 3px solid #E53935; color: #D32F2F; font-weight: 900; text-transform: uppercase; 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; }
-        
-        .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-grid { display: grid; grid-template-columns: 1fr 1fr; gap: 5px; }
-        .color-grid { display: grid; grid-template-columns: 1fr 1fr; gap: 5px; }
-        .slider-container { display: flex; align-items: center; gap: 5px; margin-top: 5px; }
-        .slider-container label { min-width: 40px; font-size: 11px; }
-        .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; }
-        
-        /* 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 .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 button { background: #3498db; color: white; border: none; padding: 12px 30px; border-radius: 8px; cursor: pointer; font-weight: bold; margin: 5px; }
-    </style>
-</head>
-<body>
-    <div id="upload-container">
-        <div class="upload-box">
-            <h1>🎬 Enhanced Comic Generator</h1>
-            <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>📚 Total Comic Pages:</label>
-                <input type="number" id="page-count" value="4" min="1" max="15" placeholder="e.g. 4 (Video will be divided evenly)">
-                <small style="color:#666; font-size:11px; display:block; margin-top:5px;">System calculates ~4 panels per page.</small>
-            </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>
-                <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>
-                </div>
-            </div>
-            <div class="loading-view" id="loading-view" style="display:none; margin-top:20px;">
-                <div class="loader" style="margin:0 auto;"></div>
-                <p id="status-text" style="margin-top:10px;">Starting...</p>
+        <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>
+            <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>
             </div>
         </div>
+        <div class="loading-view" id="loading-view" style="display:none; margin-top:20px;">
+            <div class="loader" style="margin:0 auto;"></div>
+            <p id="status-text" style="margin-top:10px;">Starting...</p>
+        </div>
     </div>
-    
-    <div id="editor-container">
-        <div class="comic-wrapper" id="comic-container"></div>
-        <input type="file" id="image-uploader" style="display: none;" accept="image/*">
-        <div class="edit-controls">
-            <h4>✏️ Interactive Editor</h4>
-            
-            <div class="control-group">
-                <label>💾 Save & Load:</label>
-                <button onclick="saveComic()" class="save-btn">💾 Save Comic</button>
-                <div id="current-save-code" style="display:none; margin-top:5px; text-align:center;">
-                    <span id="display-save-code" style="font-weight:bold; background:#eee; padding:2px 5px; border-radius:3px;"></span>
-                    <button onclick="copyCode()" style="padding:2px; width:auto; font-size:10px;">Copy</button>
-                </div>
+</div>
+
+<div id="editor-container">
+    <div class="comic-wrapper" id="comic-container"></div>
+    <input type="file" id="image-uploader" style="display: none;" accept="image/*">
+    <div class="edit-controls">
+        <h4>✏️ Interactive Editor</h4>
+        
+        <button onclick="undoLastAction()" class="undo-btn">↩️ Undo</button>
+        
+        <div class="control-group">
+            <label>💾 Save & Load:</label>
+            <button onclick="saveComic()" class="save-btn">💾 Save Comic</button>
+            <div id="current-save-code" style="display:none; margin-top:5px; text-align:center;">
+                <span id="display-save-code" style="font-weight:bold; background:#eee; padding:2px 5px; border-radius:3px;"></span>
+                <button onclick="copyCode()" style="padding:2px; width:auto; font-size:10px;">Copy</button>
             </div>
-            
-            <div class="control-group">
-                <label>💬 Bubble Styling:</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 (Shout)</option>
-                    <option value="narration">Narration (Box)</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>
-                <div class="button-grid">
-                    <button onclick="addBubble()" class="action-btn">Add</button>
-                    <button onclick="deleteBubble()" class="reset-btn">Delete</button>
-                </div>
+        </div>
+        
+        <div class="control-group">
+            <label>💬 Bubble Styling:</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 (Shout)</option>
+                <option value="narration">Narration (Box)</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>
-            
-            <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 class="button-grid">
+                <button onclick="addBubble()" class="action-btn">Add</button>
+                <button onclick="deleteBubble()" class="reset-btn">Delete</button>
             </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" 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" id="go-btn">Go</button>
-                </div>
+        </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 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)">
-                </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" id="prev-btn">⬅️ Prev</button>
+                <button onclick="adjustFrame('forward')" class="action-btn" id="next-btn">Next ➡️</button>
             </div>
-            
-            <div class="control-group">
-                <button onclick="exportComic()" class="export-btn">📥 Export as PNG</button>
-                <button onclick="goBackToUpload()" class="reset-btn" style="margin-top:10px;">🏠 Home</button>
+            <div class="timestamp-controls">
+                <input type="text" id="timestamp-input" placeholder="mm:ss or secs">
+                <button onclick="gotoTimestamp()" class="action-btn" id="go-btn">Go</button>
             </div>
         </div>
-    </div>
-    
-    <div class="modal-overlay" id="save-modal">
-        <div class="modal-content">
-            <h2>✅ Comic Saved!</h2>
-            <div class="code" id="modal-save-code">XXXXXXXX</div>
-            <button onclick="copyModalCode()">📋 Copy Code</button>
-            <button class="close-btn" onclick="closeModal()">Close</button>
+        
+        <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)">
+            </div>
+        </div>
+        
+        <div class="control-group">
+            <button onclick="exportComic()" class="export-btn">📥 Export as PNG</button>
+            <button onclick="goBackToUpload()" class="reset-btn" style="margin-top:10px;">🏠 Home</button>
         </div>
     </div>
+</div>
+
+<div class="modal-overlay" id="save-modal">
+    <div class="modal-content">
+        <h2>✅ Comic Saved!</h2>
+        <div class="code" id="modal-save-code">XXXXXXXX</div>
+        <button onclick="copyModalCode()">📋 Copy Code</button>
+        <button class="close-btn" onclick="closeModal()">Close</button>
+    </div>
+</div>
+
+<script>
+    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);
     
-    <script>
-        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;
-        let interval, selectedBubble = null, selectedPanel = null;
-        let isDragging = false, isResizing = false, isPanning = false;
-        let startX, startY, initX, initY, panStartX, panStartY, panStartTx, panStartTy;
-        let resizeHandle, originalWidth, originalHeight, originalMouseX, originalMouseY;
-        let currentlyEditing = null;
-        
-        if(localStorage.getItem('comic_draft_'+sid)) document.getElementById('restore-draft-btn').style.display = 'block';
-        
-        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!')); }
-        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'; el.innerText = busy ? '⏳' : el.getAttribute('data-txt') || el.innerText; }
-            });
-        }
+    let currentSaveCode = null;
+    let isProcessing = false;
+    let interval, selectedBubble = null, selectedPanel = null;
+    let isDragging = false, isResizing = false, isPanning = false;
+    let startX, startY, initX, initY, panStartX, panStartY, panStartTx, panStartTy;
+    let resizeHandle, originalWidth, originalHeight, originalMouseX, originalMouseY;
+    let currentlyEditing = null;
+    
+    // UNDO SYSTEM
+    let historyStack = [];
+    let historyIndex = -1;
 
-        async function saveComic() {
-            const state = getCurrentState();
-            if(!state || state.length === 0) { alert('No comic to save!'); return; }
-            try {
-                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'); }
+    function addToHistory() {
+        // If we are in the middle of history and add new action, cut off the future
+        if (historyIndex < historyStack.length - 1) {
+            historyStack = historyStack.slice(0, historyIndex + 1);
         }
+        const state = JSON.stringify(getCurrentState());
+        // Avoid duplicate states
+        if (historyStack.length > 0 && historyStack[historyStack.length - 1] === state) return;
         
-        async function loadSavedComic() {
-            const code = document.getElementById('load-code-input').value.trim().toUpperCase();
-            if(!code || code.length < 4) { alert('Invalid code'); return; }
-            try {
-                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.'); }
-        }
+        historyStack.push(state);
+        historyIndex++;
         
-        function restoreDraft() {
-            try {
-                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."); }
+        // Limit stack size
+        if (historyStack.length > 30) {
+            historyStack.shift();
+            historyIndex--;
         }
-        
-        function getCurrentState() {
-            const pages = [];
-            document.querySelectorAll('.comic-page').forEach(p => {
-                const panels = [];
-                p.querySelectorAll('.panel').forEach(pan => {
-                    const img = pan.querySelector('img');
-                    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') }
-                        });
-                    });
-                    panels.push({ 
-                        src: img.src, 
-                        zoom: img.dataset.zoom, tx: img.dataset.translateX, ty: img.dataset.translateY,
-                        bubbles: bubbles 
+    }
+
+    function undoLastAction() {
+        if (historyIndex > 0) {
+            historyIndex--;
+            const previousState = JSON.parse(historyStack[historyIndex]);
+            renderFromState(previousState);
+            saveDraft(false); // Save to local storage without adding to history stack
+        }
+    }
+    
+    if(localStorage.getItem('comic_draft_'+sid)) document.getElementById('restore-draft-btn').style.display = 'block';
+    
+    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!')); }
+    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'; el.innerText = busy ? '⏳' : el.getAttribute('data-txt') || el.innerText; }
+        });
+    }
+    async function saveComic() {
+        const state = getCurrentState();
+        if(!state || state.length === 0) { alert('No comic to save!'); return; }
+        try {
+            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(false); } 
+            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('Invalid code'); return; }
+        try {
+            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(true); } 
+            else { alert('Load failed: ' + d.message); }
+        } catch(e) { console.error(e); alert('Error loading comic.'); }
+    }
+    
+    function restoreDraft() {
+        try {
+            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';
+            addToHistory(); // Initialize history
+        } catch(e) { console.error(e); alert("Failed to restore."); }
+    }
+    
+    function getCurrentState() {
+        const pages = [];
+        document.querySelectorAll('.comic-page').forEach(p => {
+            const panels = [];
+            p.querySelectorAll('.panel').forEach(pan => {
+                const img = pan.querySelector('img');
+                const bubbles = [];
+                pan.querySelectorAll('.speech-bubble').forEach(b => {
+                    const textEl = b.querySelector('.bubble-text');
+                    // Compute styles to fix mismatch
+                    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.replace(' selected', ''), 
+                        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') }
                     });
                 });
-                pages.push({ panels: panels });
-            });
-            return pages;
-        }
-        
-        function saveDraft() { localStorage.setItem('comic_draft_'+sid, JSON.stringify({ pages: getCurrentState(), saveCode: currentSaveCode, savedAt: new Date().toISOString() })); }
-        
-        function renderFromState(pagesData) {
-            const con = document.getElementById('comic-container'); con.innerHTML = '';
-            pagesData.forEach((page, pageIdx) => {
-                const pageWrapper = document.createElement('div'); pageWrapper.className = 'page-wrapper';
-                const pageTitle = document.createElement('h2'); pageTitle.className = 'page-title'; pageTitle.textContent = `Page ${pageIdx + 1}`;
-                pageWrapper.appendChild(pageTitle);
-                const div = document.createElement('div'); div.className = 'comic-page';
-                const grid = document.createElement('div'); grid.className = 'comic-grid';
-                page.panels.forEach((pan) => {
-                    const pDiv = document.createElement('div'); pDiv.className = 'panel';
-                    pDiv.onclick = (e) => { e.stopPropagation(); selectPanel(pDiv); };
-                    const img = document.createElement('img');
-                    img.src = pan.src.includes('?') ? pan.src : pan.src + `?sid=${sid}`; 
-                    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);
-                    pDiv.appendChild(img);
-                    (pan.bubbles || []).forEach(bData => { pDiv.appendChild(createBubbleHTML(bData)); });
-                    grid.appendChild(pDiv);
+                panels.push({ 
+                    src: img.src, 
+                    zoom: img.dataset.zoom, tx: img.dataset.translateX, ty: img.dataset.translateY,
+                    bubbles: bubbles 
                 });
-                div.appendChild(grid); pageWrapper.appendChild(div); con.appendChild(pageWrapper);
             });
-        }
-        
-        async function upload() {
-            const f = document.getElementById('file-upload').files[0];
-            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); fd.append('target_pages', pCount); 
-            const r = await fetch(`/uploader?sid=${sid}`, {method:'POST', body:fd});
-            if(r.ok) interval = setInterval(checkStatus, 2000);
-            else { alert("Upload failed"); location.reload(); }
-        }
-        
-        async function checkStatus() {
-            try {
-                const r = await fetch(`/status?sid=${sid}`); const d = await r.json();
-                document.getElementById('status-text').innerText = d.message;
-                if(d.progress >= 100) { clearInterval(interval); document.getElementById('upload-container').style.display='none'; document.getElementById('editor-container').style.display='block'; loadNewComic(); }
-                else if (d.progress < 0) { clearInterval(interval); document.getElementById('status-text').textContent = "Error: " + d.message; document.querySelector('.loader').style.display = 'none'; }
-            } catch(e) {}
-        }
-        
-        function loadNewComic() {
-            fetch(`/output/pages.json?sid=${sid}`).then(r=>r.json()).then(data => {
-                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: (p.bubbles[j].type || 'speech'),
-                            classes: `speech-bubble ${p.bubbles[j].type || 'speech'} tail-bottom`
-                        }] : []
-                    }))
-                }));
-                renderFromState(cleanData); saveDraft();
+            pages.push({ panels: panels });
+        });
+        return pages;
+    }
+    
+    function saveDraft(recordHistory = true) { 
+        if(recordHistory) addToHistory();
+        localStorage.setItem('comic_draft_'+sid, JSON.stringify({ pages: getCurrentState(), saveCode: currentSaveCode, savedAt: new Date().toISOString() })); 
+    }
+    
+    function renderFromState(pagesData) {
+        const con = document.getElementById('comic-container'); con.innerHTML = '';
+        pagesData.forEach((page, pageIdx) => {
+            const pageWrapper = document.createElement('div'); pageWrapper.className = 'page-wrapper';
+            const pageTitle = document.createElement('h2'); pageTitle.className = 'page-title'; pageTitle.textContent = `Page ${pageIdx + 1}`;
+            pageWrapper.appendChild(pageTitle);
+            const div = document.createElement('div'); div.className = 'comic-page';
+            const grid = document.createElement('div'); grid.className = 'comic-grid';
+            page.panels.forEach((pan) => {
+                const pDiv = document.createElement('div'); pDiv.className = 'panel';
+                pDiv.onclick = (e) => { e.stopPropagation(); selectPanel(pDiv); };
+                const img = document.createElement('img');
+                img.src = pan.src.includes('?') ? pan.src : pan.src + `?sid=${sid}`; 
+                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);
+                pDiv.appendChild(img);
+                (pan.bubbles || []).forEach(bData => { pDiv.appendChild(createBubbleHTML(bData)); });
+                grid.appendChild(pDiv);
             });
-        }
-        
-        function createBubbleHTML(data) {
-            const b = document.createElement('div');
-            const type = data.type || 'speech';
-            b.className = data.classes || `speech-bubble ${type} tail-bottom`;
-            if (type === 'thought' && !b.className.includes('pos-')) b.className += ' pos-bl'; // Default position for thought
-            
-            b.dataset.type = type;
-            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);
-            
-            const textSpan = document.createElement('span'); textSpan.className = 'bubble-text'; textSpan.textContent = data.text || ''; b.appendChild(textSpan);
-            
-            if(type === 'thought') { for(let i=1; i<=2; i++){ const d = document.createElement('div'); d.className = `thought-dot thought-dot-${i}`; b.appendChild(d); } }
-            
-            ['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.onclick = (e) => { e.stopPropagation(); };
-            b.ondblclick = (e) => { e.stopPropagation(); editBubbleText(b); };
-            return b;
-        }
+            div.appendChild(grid); pageWrapper.appendChild(div); con.appendChild(pageWrapper);
+        });
+        // Clear selections
+        selectedBubble = null;
+        selectedPanel = null;
+        document.getElementById('bubble-type-select').disabled = true;
+        document.getElementById('font-select').disabled = true;
+    }
+    
+    async function upload() {
+        const f = document.getElementById('file-upload').files[0];
+        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); fd.append('target_pages', pCount); 
+        const r = await fetch(`/uploader?sid=${sid}`, {method:'POST', body:fd});
+        if(r.ok) interval = setInterval(checkStatus, 2000);
+        else { alert("Upload failed"); location.reload(); }
+    }
+    
+    async function checkStatus() {
+        try {
+            const r = await fetch(`/status?sid=${sid}`); const d = await r.json();
+            document.getElementById('status-text').innerText = d.message;
+            if(d.progress >= 100) { clearInterval(interval); document.getElementById('upload-container').style.display='none'; document.getElementById('editor-container').style.display='block'; loadNewComic(); }
+            else if (d.progress < 0) { clearInterval(interval); document.getElementById('status-text').textContent = "Error: " + d.message; document.querySelector('.loader').style.display = 'none'; }
+        } catch(e) {}
+    }
+    
+    function loadNewComic() {
+        fetch(`/output/pages.json?sid=${sid}`).then(r=>r.json()).then(data => {
+            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: (p.bubbles[j].type || 'speech'),
+                        classes: `speech-bubble ${p.bubbles[j].type || 'speech'} tail-bottom`
+                    }] : []
+                }))
+            }));
+            renderFromState(cleanData); saveDraft(true);
+        });
+    }
+    
+    function createBubbleHTML(data) {
+        const b = document.createElement('div');
+        const type = data.type || 'speech';
+        b.className = data.classes || `speech-bubble ${type} tail-bottom`;
+        if (type === 'thought' && !b.className.includes('pos-')) b.className += ' pos-bl'; 
         
-        function editBubbleText(bubble) {
-            if (currentlyEditing) return; currentlyEditing = bubble;
-            const textSpan = bubble.querySelector('.bubble-text'); const textarea = document.createElement('textarea');
-            textarea.value = textSpan.textContent; bubble.appendChild(textarea); textSpan.style.display = 'none'; textarea.focus();
-            const finishEditing = () => { textSpan.textContent = textarea.value; textarea.remove(); textSpan.style.display = ''; currentlyEditing = null; saveDraft(); };
-            textarea.addEventListener('blur', finishEditing, { once: true });
-            textarea.addEventListener('keydown', e => { if (e.key === 'Enter' && !e.shiftKey) { e.preventDefault(); textarea.blur(); } });
-        }
+        b.dataset.type = type;
+        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);
         
-        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; 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'); 
-            document.getElementById('bubble-type-select').disabled = false;
-            document.getElementById('font-select').disabled = false;
-            document.getElementById('bubble-text-color').disabled = false;
-            document.getElementById('bubble-fill-color').disabled = false;
-            document.getElementById('tail-controls').style.display = 'block';
-            document.getElementById('bubble-type-select').value = el.dataset.type || 'speech';
-        }
+        const textSpan = document.createElement('span'); textSpan.className = 'bubble-text'; textSpan.textContent = data.text || ''; b.appendChild(textSpan);
         
-        function selectPanel(el) { 
-            if(selectedPanel) selectedPanel.classList.remove('selected'); 
-            if(selectedBubble) { selectedBubble.classList.remove('selected'); selectedBubble = null; } 
-            selectedPanel = el; el.classList.add('selected'); 
-            document.getElementById('zoom-slider').disabled = false;
-            const img = el.querySelector('img');
-            document.getElementById('zoom-slider').value = img.dataset.zoom || 100;
-            document.getElementById('bubble-type-select').disabled = true;
-            document.getElementById('font-select').disabled = true;
-            document.getElementById('tail-controls').style.display = 'none';
-        }
+        if(type === 'thought') { for(let i=1; i<=2; i++){ const d = document.createElement('div'); d.className = `thought-dot thought-dot-${i}`; b.appendChild(d); } }
         
-        function addBubble() { 
-            if(!selectedPanel) return alert("Select a panel first"); 
-            const b = createBubbleHTML({ text: "Text", left: "50px", top: "30px", type: 'speech', classes: "speech-bubble speech tail-bottom" }); 
-            selectedPanel.appendChild(b); selectBubble(b); saveDraft(); 
-        }
+        ['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); });
         
-        function deleteBubble() { 
-            if(!selectedBubble) return alert("Select a bubble"); 
-            selectedBubble.remove(); selectedBubble=null; saveDraft(); 
+        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.onclick = (e) => { e.stopPropagation(); };
+        b.ondblclick = (e) => { e.stopPropagation(); editBubbleText(b); };
+        return b;
+    }
+    
+    function editBubbleText(bubble) {
+        if (currentlyEditing) return; currentlyEditing = bubble;
+        const textSpan = bubble.querySelector('.bubble-text'); const textarea = document.createElement('textarea');
+        textarea.value = textSpan.textContent; bubble.appendChild(textarea); textSpan.style.display = 'none'; textarea.focus();
+        const finishEditing = () => { 
+            textSpan.textContent = textarea.value; textarea.remove(); textSpan.style.display = ''; currentlyEditing = null; 
+            saveDraft(true); 
+        };
+        textarea.addEventListener('blur', finishEditing, { once: true });
+        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); }
+    });
+    
+    // Save draft and add to history on mouseup ONLY if action occurred
+    document.addEventListener('mouseup', () => { 
+        if(isDragging || isResizing || isPanning) {
+            saveDraft(true);
         }
+        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; 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'); 
+        document.getElementById('bubble-type-select').disabled = false;
+        document.getElementById('font-select').disabled = false;
+        document.getElementById('bubble-text-color').disabled = false;
+        document.getElementById('bubble-fill-color').disabled = false;
+        document.getElementById('tail-controls').style.display = 'block';
+        document.getElementById('bubble-type-select').value = el.dataset.type || 'speech';
+    }
+    
+    function selectPanel(el) { 
+        if(selectedPanel) selectedPanel.classList.remove('selected'); 
+        if(selectedBubble) { selectedBubble.classList.remove('selected'); selectedBubble = null; } 
+        selectedPanel = el; el.classList.add('selected'); 
+        document.getElementById('zoom-slider').disabled = false;
+        const img = el.querySelector('img');
+        document.getElementById('zoom-slider').value = img.dataset.zoom || 100;
+        document.getElementById('bubble-type-select').disabled = true;
+        document.getElementById('font-select').disabled = true;
+        document.getElementById('tail-controls').style.display = 'none';
+    }
+    
+    function addBubble() { 
+        if(!selectedPanel) return alert("Select a panel first"); 
+        const b = createBubbleHTML({ text: "Text", left: "50px", top: "30px", type: 'speech', classes: "speech-bubble speech tail-bottom" }); 
+        selectedPanel.appendChild(b); selectBubble(b); saveDraft(true); 
+    }
+    
+    function deleteBubble() { 
+        if(!selectedBubble) return alert("Select a bubble"); 
+        selectedBubble.remove(); selectedBubble=null; saveDraft(true); 
+    }
+    
+    function changeBubbleType(type) {
+        if(!selectedBubble) return;
+        selectedBubble.dataset.type = type;
+        selectedBubble.className = 'speech-bubble ' + type + ' selected';
         
-        function changeBubbleType(type) {
-            if(!selectedBubble) return;
-            selectedBubble.dataset.type = type;
-            selectedBubble.className = 'speech-bubble ' + type + ' selected';
-            
-            // Default tail for thought is pos-bl if not set
-            if(type === 'thought') selectedBubble.classList.add('pos-bl');
-            else selectedBubble.classList.add('tail-bottom'); // Default for speech
-            
-            selectedBubble.querySelectorAll('.thought-dot').forEach(d=>d.remove());
-            if(type === 'thought') { for(let i=1; i<=2; i++){ const d = document.createElement('div'); d.className = `thought-dot thought-dot-${i}`; selectedBubble.appendChild(d); } }
-            saveDraft();
-        }
+        // Default tail for thought is pos-bl if not set
+        if(type === 'thought') selectedBubble.classList.add('pos-bl');
+        else selectedBubble.classList.add('tail-bottom'); // Default for speech
         
-        function changeFont(font) { if(!selectedBubble) return; selectedBubble.style.fontFamily = font; saveDraft(); }
+        selectedBubble.querySelectorAll('.thought-dot').forEach(d=>d.remove());
+        if(type === 'thought') { for(let i=1; i<=2; i++){ const d = document.createElement('div'); d.className = `thought-dot thought-dot-${i}`; selectedBubble.appendChild(d); } }
+        saveDraft(true);
+    }
+    
+    function changeFont(font) { if(!selectedBubble) return; selectedBubble.style.fontFamily = font; saveDraft(true); }
+    
+    function rotateTail() {
+        if(!selectedBubble) return;
+        const type = selectedBubble.dataset.type;
         
-        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') {
-                // Cycle specifically through the 4 CSS positions we defined
-                const positions = ['pos-bl', 'pos-br', 'pos-tr', 'pos-tl'];
-                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]);
-            }
-            saveDraft();
+        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 positions = ['pos-bl', 'pos-br', 'pos-tr', 'pos-tl'];
+            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]);
         }
+        saveDraft(true);
+    }
+    
+    function slideTail(v) { if(selectedBubble) { selectedBubble.style.setProperty('--tail-pos', v+'%'); saveDraft(true); } }
+    
+    document.getElementById('bubble-text-color').addEventListener('change', (e) => { if(selectedBubble) { selectedBubble.style.setProperty('--bubble-text-color', e.target.value); saveDraft(true); } });
+    document.getElementById('bubble-fill-color').addEventListener('change', (e) => { if(selectedBubble) { selectedBubble.style.setProperty('--bubble-fill-color', e.target.value); saveDraft(true); } });
+    
+    function handleZoom(el) { if(!selectedPanel) return; const img = selectedPanel.querySelector('img'); img.dataset.zoom = el.value; updateImageTransform(img); }
+    // Add history on zoom release
+    document.getElementById('zoom-slider').addEventListener('change', () => saveDraft(true));
+
+    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"); 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(true); }
+    
+    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(true); } inp.value = ''; }; inp.click(); }
+    async function adjustFrame(dir) { if(isProcessing) return; if(!selectedPanel) return alert("Select a panel"); const img = selectedPanel.querySelector('img'); let fname = img.src.split('/').pop().split('?')[0]; setProcessing(true); 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); saveDraft(true); }
+    async function gotoTimestamp() { if(isProcessing) return; 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); 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); saveDraft(true); }
+    
+    async function exportComic() { 
+        const pgs = document.querySelectorAll('.comic-page'); 
+        if(pgs.length === 0) return alert("No pages found"); 
         
-        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(); } });
+        // Remove selection highlights before export
+        if(selectedBubble) selectedBubble.classList.remove('selected');
+        if(selectedPanel) selectedPanel.classList.remove('selected');
+
+        alert(`Exporting ${pgs.length} page(s)...`); 
         
-        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"); 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(); }
+        // Fix for mismatch during export: Lock dimensions
+        const bubbles = document.querySelectorAll('.speech-bubble'); 
+        bubbles.forEach(b => { 
+            const rect = b.getBoundingClientRect(); 
+            // We use style directly to prevent layout shifts
+            b.style.width = rect.width + 'px'; 
+            b.style.height = rect.height + 'px'; 
+        }); 
         
-        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; if(!selectedPanel) return alert("Select a panel"); const img = selectedPanel.querySelector('img'); let fname = img.src.split('/').pop().split('?')[0]; setProcessing(true); 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); saveDraft(); }
-        async function gotoTimestamp() { if(isProcessing) return; 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); 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); 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)...`); 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'; }); 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}`); } } bubbles.forEach(b => { b.style.width = ''; b.style.height = ''; }); }
+        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>
-'''
+        // Clean up styles
+        bubbles.forEach(b => { 
+            // Only reset if they weren't manually sized, but since users resize manually,
+            // we should leave them be or reset only if they were auto-sized.
+            // In this specific app, we rely on the state re-render or just leaving them 
+            // as they are usually pixel defined anyway.
+        }); 
+    }
+    
+    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> '''
 
 @app.route('/')
 def index():
@@ -864,7 +943,7 @@ def upload():
     
     # GET PAGE COUNT FROM FORM
     target_pages = request.form.get('target_pages', 4)
-    
+
     f = request.files['file']
     gen = EnhancedComicGenerator(sid)
     gen.cleanup()
@@ -972,6 +1051,6 @@ def load_comic(code):
         return jsonify({'success': False, 'message': str(e)})
 
 if __name__ == '__main__':
-    try: gpu_warmup() 
+    try: gpu_warmup()
     except: pass
     app.run(host='0.0.0.0', port=7860)