Update app_enhanced.py

app_enhanced.py

@@ -24,7 +24,7 @@ def gpu_warmup():
     return True
 
 # ======================================================
-# 💾 PERSISTENT STORAGE
+# 💾 STORAGE SETUP
 # ======================================================
 if os.path.exists('/data'):
     BASE_STORAGE_PATH = '/data'
@@ -39,11 +39,8 @@ SAVED_COMICS_DIR = os.path.join(BASE_STORAGE_PATH, "saved_comics")
 os.makedirs(BASE_USER_DIR, exist_ok=True)
 os.makedirs(SAVED_COMICS_DIR, exist_ok=True)
 
-# ======================================================
-# 🔧 APP CONFIG
-# ======================================================
 app = Flask(__name__)
-app.config['MAX_CONTENT_LENGTH'] = 500 * 1024 * 1024
+app.config['MAX_CONTENT_LENGTH'] = 500 * 1024 * 1024  # 500MB Upload Limit
 
 def generate_save_code(length=8):
     chars = string.ascii_uppercase + string.digits
@@ -55,21 +52,14 @@ def generate_save_code(length=8):
 # ======================================================
 # 🧱 DATA CLASSES
 # ======================================================
-def bubble(dialog="", x=50, y=50, type='speech'):
-    # Determine CSS classes based on type
-    classes = f"speech-bubble {type}"
-    if type == 'speech':
-        classes += " tail-bottom"
-    elif type == 'thought':
-        classes += " pos-bl"
-        
+def bubble(dialog="", x=50, y=20, type='speech'):
     return {
         'dialog': dialog,
         'bubble_offset_x': int(x),
         'bubble_offset_y': int(y),
         'type': type,
         'tail_pos': '50%',
-        'classes': classes,
+        'classes': f'speech-bubble {type} tail-bottom',
         'colors': {'fill': '#ffffff', 'text': '#000000'},
         'font': "'Comic Neue', cursive"
     }
@@ -83,11 +73,12 @@ class Page:
         self.bubbles = bubbles
 
 # ======================================================
-# 🧠 GPU GENERATION (1920x1080 Source -> 864x1080 Layout)
+# 🧠 GPU GENERATION (16:9 Aspect Ratio Preservation)
 # ======================================================
 @spaces.GPU(duration=300)
 def generate_comic_gpu(video_path, user_dir, frames_dir, metadata_path, target_pages):
-    print(f"🚀 Generating Comic (864x1080): {video_path}")
+    print(f"🚀 Generating Comic (16:9 Aspect Ratio): {video_path}")
+    
     import cv2
     import srt
     import numpy as np
@@ -100,7 +91,6 @@ def generate_comic_gpu(video_path, user_dir, frames_dir, metadata_path, target_p
     duration = total_frames / fps
     cap.release()
 
-    # Subtitles
     user_srt = os.path.join(user_dir, 'subs.srt')
     try:
         get_real_subtitles(video_path)
@@ -116,14 +106,11 @@ def generate_comic_gpu(video_path, user_dir, frames_dir, metadata_path, target_p
         except: all_subs = []
 
     valid_subs = [s for s in all_subs if s.content.strip()]
-    if valid_subs:
-        raw_moments = [{'text': s.content.strip(), 'start': s.start.total_seconds(), 'end': s.end.total_seconds()} for s in valid_subs]
-    else:
-        raw_moments = []
+    raw_moments = [{'text': s.content, 'start': s.start.total_seconds(), 'end': s.end.total_seconds()} for s in valid_subs]
 
-    # 4 Panels Per Page (Logic)
+    if target_pages <= 0: target_pages = 1
     panels_per_page = 4
-    total_panels_needed = int(target_pages) * panels_per_page
+    total_panels_needed = target_pages * panels_per_page
 
     selected_moments = []
     if not raw_moments:
@@ -142,11 +129,16 @@ def generate_comic_gpu(video_path, user_dir, frames_dir, metadata_path, target_p
 
     for i, moment in enumerate(selected_moments):
         mid = (moment['start'] + moment['end']) / 2
-        cap.set(cv2.CAP_PROP_POS_MSEC, mid * 1000)
+        if mid > duration: mid = duration - 1
+        cap.set(cv2.CAP_PROP_POS_FRAMES, int(mid * fps))
         ret, frame = cap.read()
         if ret:
-            # 🎯 EXTRACT AT 1920x1080 (Matches Page Height, High Quality)
-            frame = cv2.resize(frame, (1920, 1080))
+            # ----------------------------------------------------
+            # 🎯 EXTRACT AT 1280x720 (16:9 Aspect Ratio)
+            # This preserves full width and height.
+            # Frontend handles fitting into square panels.
+            # ----------------------------------------------------
+            frame = cv2.resize(frame, (1280, 720))
             
             fname = f"frame_{count:04d}.png"
             p = os.path.join(frames_dir, fname)
@@ -161,32 +153,26 @@ def generate_comic_gpu(video_path, user_dir, frames_dir, metadata_path, target_p
         
     bubbles_list = []
     for i, f in enumerate(frame_files_ordered):
+        p = os.path.join(frames_dir, f)
         dialogue = frame_metadata.get(f, {}).get('dialogue', '')
         
         b_type = 'speech'
-        if '(' in dialogue: b_type = 'narration'
-        elif '!' in dialogue: b_type = 'reaction'
+        if '(' in dialogue and ')' in dialogue: b_type = 'narration'
+        elif '!' in dialogue and dialogue.isupper(): b_type = 'reaction'
         elif '?' in dialogue: b_type = 'speech'
         
-        # 🎯 SMART PLACEMENT FOR 864x1080 GRID
-        # Top-Left (0), Top-Right (1), Bot-Left (2), Bot-Right (3)
-        pos_idx = i % 4
+        # Position bubbles centrally in panels
+        pos_idx = i % 4 # Determine which of the 4 panels this frame is for
+        if pos_idx == 0: bx, by = 150, 80  # Top-Left
+        elif pos_idx == 1: bx, by = 600, 80 # Top-Right
+        elif pos_idx == 2: bx, by = 150, 600 # Bottom-Left
+        elif pos_idx == 3: bx, by = 600, 600 # Bottom-Right
+        else: bx, by = 50, 50 # Default
         
-        if pos_idx == 0:   # Top-Left Panel
-            bx, by = 150, 80
-        elif pos_idx == 1: # Top-Right Panel (Hidden by default)
-            bx, by = 600, 80
-        elif pos_idx == 2: # Bot-Left Panel
-            bx, by = 150, 600
-        elif pos_idx == 3: # Bot-Right Panel (Hidden by default)
-            bx, by = 600, 600
-        else:
-            bx, by = 50, 50
-            
         bubbles_list.append(bubble(dialog=dialogue, x=bx, y=by, type=b_type))
             
     pages = []
-    for i in range(int(target_pages)):
+    for i in range(target_pages):
         start_idx = i * 4
         end_idx = start_idx + 4
         p_frames = frame_files_ordered[start_idx:end_idx]
@@ -194,17 +180,22 @@ def generate_comic_gpu(video_path, user_dir, frames_dir, metadata_path, target_p
         
         while len(p_frames) < 4:
              fname = f"empty_{i}_{len(p_frames)}.png"
-             img = np.zeros((1080, 1920, 3), dtype=np.uint8); img[:] = (30,30,30)
+             img = np.zeros((1024, 1024, 3), dtype=np.uint8); img[:] = (30,30,30)
              cv2.imwrite(os.path.join(frames_dir, fname), img)
              p_frames.append(fname)
-             # Place empty bubble offscreen
-             p_bubbles.append(bubble(dialog="", x=-999, y=-999, type='speech'))
+             p_bubbles.append(bubble(dialog="", type='speech'))
 
         if p_frames:
-            pg_panels = [{'image': f} for f in p_frames]
+            pg_panels = [panel(image=f) for f in p_frames]
             pages.append({'panels': pg_panels, 'bubbles': p_bubbles})
 
-    return pages
+    result = []
+    for pg in pages:
+        p_data = [p if isinstance(p, dict) else p.__dict__ for p in pg.panels]
+        b_data = [b if isinstance(b, dict) else b.__dict__ for b in pg.bubbles]
+        result.append({'panels': p_data, 'bubbles': b_data})
+        
+    return result
 
 @spaces.GPU
 def regen_frame_gpu(video_path, frames_dir, metadata_path, fname, direction):
@@ -224,35 +215,15 @@ def regen_frame_gpu(video_path, frames_dir, metadata_path, fname, direction):
     cap.release()
 
     if ret:
-        frame = cv2.resize(frame, (1920, 1080))
-        cv2.imwrite(os.path.join(frames_dir, fname), frame)
+        frame = cv2.resize(frame, (1280, 720)) # Keep 16:9 Aspect Ratio
+        p = os.path.join(frames_dir, fname)
+        cv2.imwrite(p, frame)
+        
         if isinstance(meta[fname], dict): meta[fname]['time'] = new_t
         else: meta[fname] = new_t
         with open(metadata_path, 'w') as f: json.dump(meta, f, indent=2)
         return {"success": True, "message": f"Time: {new_t:.2f}s"}
-    return {"success": False}
-
-@spaces.GPU
-def get_frame_at_ts_gpu(video_path, frames_dir, metadata_path, fname, ts):
-    import cv2
-    import json
-    cap = cv2.VideoCapture(video_path)
-    cap.set(cv2.CAP_PROP_POS_MSEC, float(ts) * 1000)
-    ret, frame = cap.read()
-    cap.release()
-
-    if ret:
-        frame = cv2.resize(frame, (1920, 1080))
-        p = os.path.join(frames_dir, fname)
-        cv2.imwrite(p, frame)
-        if os.path.exists(metadata_path):
-            with open(metadata_path, 'r') as f: meta = json.load(f)
-            if fname in meta:
-                if isinstance(meta[fname], dict): meta[fname]['time'] = float(ts)
-                else: meta[fname] = float(ts)
-                with open(metadata_path, 'w') as f: json.dump(meta, f, indent=2)
-        return {"success": True, "message": f"Jumped to {ts}s"}
-    return {"success": False, "message": "Invalid timestamp"}
+    return {"success": False, "message": "End of video"}
 
 # ======================================================
 # 💻 BACKEND CLASS
@@ -290,10 +261,10 @@ class EnhancedComicGenerator:
             json.dump({'message': msg, 'progress': prog}, f)
 
 # ======================================================
-# 🌐 FRONTEND
+# 🌐 ROUTES & FRONTEND
 # ======================================================
 INDEX_HTML = '''
-<!DOCTYPE html><html lang="en"> <head> <meta charset="UTF-8"> <meta name="viewport" content="width=device-width, initial-scale=1.0"> <title>864x1080 Comic Gen</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: #2c3e50; font-family: 'Lato', sans-serif; color: #eee; margin: 0; min-height: 100vh; }
+<!DOCTYPE html><html lang="en"> <head> <meta charset="UTF-8"> <meta name="viewport" content="width=device-width, initial-scale=1.0"> <title>Square HD Comic</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: #2c3e50; font-family: 'Lato', sans-serif; color: #eee; 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: #34495e; border-radius: 12px; box-shadow: 0 10px 30px rgba(0,0,0,0.5); text-align: center; }
@@ -313,14 +284,14 @@ INDEX_HTML = '''
     .loader { width: 100px; height: 10px; background: #e67e22; margin: 20px auto; animation: load 1s infinite alternate; }
     @keyframes load { from { width: 20px; } to { width: 100px; } }
     
-    /* === 864x1080 PAGE LAYOUT === */
+    /* === SQUARE COMIC LAYOUT (800x800) === */
     .comic-wrapper { max-width: 1000px; margin: 0 auto; display: flex; flex-direction: column; align-items: center; gap: 40px; }
     .page-wrapper { display: flex; flex-direction: column; align-items: center; }
     .page-title { text-align: center; color: #eee; margin-bottom: 10px; font-size: 20px; font-weight: bold; }
     
     .comic-page { 
-        width: 864px; 
-        height: 1080px; 
+        width: 800px; 
+        height: 800px; 
         background: white; 
         box-shadow: 0 5px 30px rgba(0,0,0,0.6); 
         position: relative; overflow: hidden; 
@@ -333,25 +304,17 @@ INDEX_HTML = '''
     }
 
     .panel { position: absolute; top: 0; left: 0; width: 100%; height: 100%; overflow: hidden; background: #1a1a1a; cursor: grab; }
-    
-    /* IMAGE HANDLING */
-    .panel img { 
-        width: 100%; height: 100%; 
-        object-fit: cover; 
-        transform-origin: center; 
-        transition: transform 0.05s ease-out; 
-        display: block; 
-    }
+    .panel img { width: 100%; height: 100%; object-fit: cover; transform-origin: center; transition: transform 0.05s ease-out; display: block; }
     .panel img.panning { cursor: grabbing; transition: none; }
     .panel.selected { outline: 4px solid #3498db; z-index: 5; }
 
-    /* === CLIP PATHS FOR 864x1080 (100% t1/b1 = 864px width) === */
+    /* === CLIP PATHS === */
     .panel:nth-child(1) { clip-path: polygon(0 0, calc(var(--t1) - var(--gap)) 0, calc(var(--t2) - var(--gap)) calc(var(--y) - var(--gap)), 0 calc(var(--y) - var(--gap))); z-index: 1; }
     .panel:nth-child(2) { clip-path: polygon(calc(var(--t1) + var(--gap)) 0, 100% 0, 100% calc(var(--y) - var(--gap)), calc(var(--t2) + var(--gap)) calc(var(--y) - var(--gap))); z-index: 1; }
     .panel:nth-child(3) { clip-path: polygon(0 calc(var(--y) + var(--gap)), calc(var(--b1) - var(--gap)) calc(var(--y) + var(--gap)), calc(var(--b2) - var(--gap)) 100%, 0 100%); z-index: 1; }
     .panel:nth-child(4) { clip-path: polygon(calc(var(--b1) + var(--gap)) calc(var(--y) + var(--gap)), 100% calc(var(--y) + var(--gap)), 100% 100%, calc(var(--b2) + var(--gap)) 100%); z-index: 1; }
 
-    /* Handles */
+    /* === HANDLES === */
     .handle { position: absolute; width: 26px; height: 26px; border: 3px solid white; border-radius: 50%; transform: translate(-50%, -50%); z-index: 101; cursor: ew-resize; box-shadow: 0 2px 5px rgba(0,0,0,0.8); }
     .handle:hover { transform: translate(-50%, -50%) scale(1.3); }
     .h-t1 { background: #3498db; left: var(--t1); top: 0%; margin-top: 15px; }
@@ -359,18 +322,14 @@ INDEX_HTML = '''
     .h-b1 { background: #2ecc71; left: var(--b1); top: 50%; margin-top: 15px; }
     .h-b2 { background: #2ecc71; left: var(--b2); top: 100%; margin-top: -15px; }
     
-    /* === SPEECH BUBBLES === */
+    /* SPEECH BUBBLES */
     .speech-bubble { 
         position: absolute; display: flex; justify-content: center; align-items: center; 
         min-width: 60px; min-height: 40px; box-sizing: border-box; 
         z-index: 10; cursor: move; font-weight: bold; text-align: center; 
         overflow: visible; line-height: 1.2; --tail-pos: 50%;
     }
-    .bubble-text { 
-        padding: 0.8em; word-wrap: break-word; white-space: pre-wrap; 
-        width: 100%; height: 100%; display: flex; align-items: center; justify-content: center; 
-        border-radius: inherit; pointer-events: none;
-    }
+    .bubble-text { padding: 0.8em; width: 100%; height: 100%; display: flex; align-items: center; justify-content: center; overflow: hidden; white-space: pre-wrap; pointer-events: none; border-radius: inherit; }
     .speech-bubble.selected { outline: 2px dashed #4CAF50; z-index: 100; }
     
     /* SPEECH */
@@ -422,120 +381,142 @@ INDEX_HTML = '''
     button, input, select { width: 100%; margin-top: 5px; padding: 8px; border-radius: 4px; border: 1px solid #ddd; cursor: pointer; font-weight: bold; font-size: 13px; }
     .button-grid { display: grid; grid-template-columns: 1fr 1fr; gap: 5px; }
     .color-grid { display: grid; grid-template-columns: 1fr 1fr; gap: 5px; }
-    .action-btn { background: #27ae60; color: white; }
-    .reset-btn { background: #c0392b; color: white; }
+    .action-btn { background: #4CAF50; color: white; }
+    .reset-btn { background: #e74c3c; color: white; }
     .secondary-btn { background: #f39c12; color: white; }
-    .save-btn { background: #8e44ad; color: white; }
+    .save-btn { background: #9b59b6; color: white; }
+    .undo-btn { background: #7f8c8d; color: white; margin-bottom: 5px; }
     
-    .tip { text-align:center; padding:10px; background:#e74c3c; color:white; font-weight:bold; margin-bottom:20px; border-radius:5px; }
-    .modal-overlay { position: fixed; top: 0; left: 0; width: 100%; height: 100%; background: rgba(0,0,0,0.8); display: none; justify-content: center; align-items: center; z-index: 2000; }
-    .modal-content { background: white; padding: 30px; border-radius: 12px; width: 90%; max-width: 400px; text-align: center; color: #333; }
-    .code { font-size: 24px; font-weight: bold; letter-spacing: 3px; background: #eee; padding: 10px; margin: 15px 0; display: inline-block; font-family: monospace; }
+    /* 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; }
 </style>
 </head> <body> 
 
 <div id="upload-container"> 
     <div class="upload-box"> 
-        <h1>⚡ 864x1080 Comic Gen</h1> 
+        <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</label> 
-        <span id="fn" style="margin-bottom:10px; display:block; color:#aaa;">No file selected</span>
+        <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 Pages:</label>
-            <input type="number" id="page-count" value="4" min="1" max="15">
+            <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</button>
-        <button id="restore-draft-btn" class="reset-btn" style="display:none; margin-top:10px;" onclick="restoreDraft()">📂 Restore Draft</button>
+        <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 style="margin-top:20px; border-top:1px solid #555; padding-top:10px;">
-            <input type="text" id="load-code" placeholder="ENTER SAVE CODE" style="width:70%; display:inline-block;">
-            <button onclick="loadComic()" style="width:25%; display:inline-block; background:#9b59b6; color:white;">Load</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"></div>
-            <p id="status-text" style="margin-top:10px;">Analyzing Video...</p>
+            <div class="loader" style="margin:0 auto;"></div>
+            <p id="status-text" style="margin-top:10px;">Starting...</p>
         </div>
     </div>
 </div>
 
 <div id="editor-container">
-    <div class="tip">👉 Drag Right-Side Dots (Blue/Green) to reveal 4 panels! | 📜 Scroll to Zoom/Pan</div>
+    <div class="tip">👉 Drag Right-Side Dots to reveal 4 panels! | 📜 Scroll to Zoom/Pan</div>
     <div class="comic-wrapper" id="comic-container"></div>
     <input type="file" id="image-uploader" style="display: none;" accept="image/*">
     
     <div class="edit-controls">
-        <h4>✏️ Editor</h4>
+        <h4>✏️ Interactive Editor</h4>
         
         <div class="control-group">
-            <button onclick="undo()" style="background:#7f8c8d; color:white;">↩️ Undo</button>
+            <button onclick="undoLastAction()" class="undo-btn">↩️ Undo</button>
             <button onclick="saveComic()" class="save-btn">💾 Save Comic</button>
         </div>
-
+        
         <div class="control-group">
             <label>💬 Bubble Styling:</label>
-            <select id="bubble-type" onchange="updateBubbleType()">
-                <option value="speech">Speech 💬</option>
-                <option value="thought">Thought 💭</option>
-                <option value="reaction">Reaction 💥</option>
-                <option value="narration">Narration ⬜</option>
+            <select id="bubble-type" 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="updateFont()">
+            <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">Handwritten</option>
-                <option value="'Lato', sans-serif">Modern</option>
+                <option value="'Gloria Hallelujah', cursive">Gloria</option>
+                <option value="'Lato', sans-serif">Lato</option>
             </select>
             <div class="color-grid">
-                <input type="color" id="bub-fill" value="#ffffff" onchange="updateColors()" title="Fill">
-                <input type="color" id="bub-text" value="#000000" onchange="updateColors()" title="Text">
+                <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>
-            <input type="range" min="10" max="90" value="50" oninput="slideTail(this.value)" title="Tail Pos">
+            
+            <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>
-
+        
         <div class="control-group">
-            <label>🖼️ Image Control:</label>
-            <button onclick="replaceImage()" class="action-btn">Replace Image</button>
+            <label>🖼️ Panel Tools:</label>
+            <button onclick="replacePanelImage()" class="action-btn">🖼️ Replace Image</button>
             <div class="button-grid">
-                <button onclick="adjustFrame('backward')" class="secondary-btn">⬅️ Frame</button>
-                <button onclick="adjustFrame('forward')" class="action-btn">Frame ➡️</button>
+                <button onclick="adjustFrame('backward')" class="secondary-btn">⬅️ Prev</button>
+                <button onclick="adjustFrame('forward')" class="action-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 (Scroll Wheel):</label>
-            <input type="range" id="zoom-slider" min="20" max="300" value="100" step="5" oninput="handleZoom(this.value)" disabled>
-            <button onclick="resetPanelTransform()" class="secondary-btn">Reset View</button>
+            <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="action-btn" style="background:#3498db;">📥 Export PNG</button>
+            <button onclick="exportComic()" class="export-btn">📥 Export as PNG</button>
             <button onclick="location.reload()" 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-code">XXXX</div>
-        <button onclick="closeModal()">Close</button>
+        <div class="code" id="modal-save-code">XXXX</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);
+    
+    let currentSaveCode = null;
+    let isProcessing = false;
     let interval, selectedBubble = null, selectedPanel = null;
     let dragType = null, activeObj = null, dragStart = {x:0, y:0};
     let historyStack = [];
     
+    // HISTORY & UNDO
     function saveState() {
         const state = [];
         document.querySelectorAll('.comic-page').forEach(pg => {
@@ -547,9 +528,9 @@ INDEX_HTML = '''
                     text: b.querySelector('.bubble-text').textContent,
                     left: b.style.left, top: b.style.top, width: b.style.width, height: b.style.height,
                     type: b.dataset.type, font: b.style.fontFamily,
-                    colors: { fill: b.style.getPropertyValue('--bubble-fill'), text: b.style.getPropertyValue('--bubble-text') },
+                    classes: b.className, // Save the full class list
                     tailPos: b.style.getPropertyValue('--tail-pos'),
-                    classes: b.className
+                    colors: { fill: b.style.getPropertyValue('--bubble-fill'), text: b.style.getPropertyValue('--bubble-text') }
                 });
             });
             const panels = [];
@@ -633,7 +614,6 @@ INDEX_HTML = '''
         const data = await r.json();
         const cleanData = data.map(p => ({
             panels: p.panels.map(pan => ({ src: `/frames/${pan.image}?sid=${sid}` })),
-            // Map backend bubbles to strict list structure
             bubbles: p.bubbles.map(b => ({
                 text: b.dialog, left: (b.bubble_offset_x||50)+'px', top: (b.bubble_offset_y||20)+'px', type: b.type,
                 colors: b.colors, font: b.font, classes: b.classes, tailPos: b.tail_pos
@@ -687,8 +667,8 @@ INDEX_HTML = '''
         const type = data.type || 'speech';
         let className = data.classes || `speech-bubble ${type} tail-bottom`;
         if (type === 'thought' && !className.includes('pos-')) className += ' pos-bl';
-        b.className = className;
         
+        b.className = className;
         b.dataset.type = type;
         b.style.left = data.left; b.style.top = data.top;
         if(data.width) b.style.width = data.width;
@@ -699,12 +679,9 @@ INDEX_HTML = '''
         
         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); } }
         
-        const textSpan = document.createElement('span'); textSpan.className = 'bubble-text'; textSpan.textContent = data.text || 'Text'; b.appendChild(textSpan);
-        const resizer = document.createElement('div'); resizer.className = 'resize-handle'; 
-        resizer.onmousedown = (e) => { e.stopPropagation(); dragType='resize'; activeObj={b:b, startW:b.offsetWidth, startH:b.offsetHeight, mx:e.clientX, my:e.clientY}; };
-        b.appendChild(resizer);
+        ['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 === resizer) return; e.stopPropagation(); selectBubble(b); dragType = 'bubble'; activeObj = b; dragStart = {x: e.clientX, y: e.clientY}; };
+        b.onmousedown = (e) => { if(e.target.classList.contains('resize-handle')) return; e.stopPropagation(); selectBubble(b); dragType = 'bubble'; activeObj = b; dragStart = {x: e.clientX, y: e.clientY}; };
         b.ondblclick = (e) => { e.stopPropagation(); editBubbleText(b); };
         return b;
     }
@@ -768,7 +745,8 @@ INDEX_HTML = '''
             left: oldB.style.left, top: oldB.style.top, width: oldB.style.width, height: oldB.style.height,
             type: type, font: oldB.style.fontFamily,
             colors: { fill: oldB.style.getPropertyValue('--bubble-fill'), text: oldB.style.getPropertyValue('--bubble-text') },
-            tailPos: oldB.style.getPropertyValue('--tail-pos')
+            tailPos: oldB.style.getPropertyValue('--tail-pos'),
+            classes: oldB.className // Pass existing classes
         };
         const newB = createBubbleHTML(data);
         oldB.parentElement.replaceChild(newB, oldB);
@@ -841,7 +819,7 @@ INDEX_HTML = '''
             const layout = { t1: grid.style.getPropertyValue('--t1')||'100%', t2: grid.style.getPropertyValue('--t2')||'100%', b1: grid.style.getPropertyValue('--b1')||'100%', b2: grid.style.getPropertyValue('--b2')||'100%' };
             const bubbles = [];
             grid.querySelectorAll('.speech-bubble').forEach(b => {
-                bubbles.push({ text: b.querySelector('.bubble-text').textContent, left: b.style.left, top: b.style.top, width: b.style.width, height: b.style.height, type: b.dataset.type, colors: { fill: b.style.getPropertyValue('--bubble-fill'), text: b.style.getPropertyValue('--bubble-text') }, tailPos: b.style.getPropertyValue('--tail-pos') });
+                bubbles.push({ text: b.querySelector('.bubble-text').textContent, left: b.style.left, top: b.style.top, width: b.style.width, height: b.style.height, type: b.dataset.type, colors: { fill: b.style.getPropertyValue('--bubble-fill'), text: b.style.getPropertyValue('--bubble-text') }, tailPos: b.style.getPropertyValue('--tail-pos'), classes: b.className });
             });
             const panels = [];
             grid.querySelectorAll('.panel').forEach(pan => {