Update app_enhanced.py

app_enhanced.py

@@ -9,6 +9,7 @@ import string
 import random
 import shutil
 import cv2
+import math
 import numpy as np
 import srt
 from flask import Flask, jsonify, request, send_from_directory, send_file
@@ -23,273 +24,532 @@ def gpu_warmup():
     return True
 
 # ======================================================
-# 🧠 GLOBAL GPU FUNCTIONS
+# 💾 PERSISTENT STORAGE CONFIGURATION
+# ======================================================
+if os.path.exists('/data'):
+    BASE_STORAGE_PATH = '/data'
+    print("✅ Using Persistent Storage at /data")
+else:
+    BASE_STORAGE_PATH = '.'
+    print("⚠️ Using Ephemeral Storage")
+
+BASE_USER_DIR = os.path.join(BASE_STORAGE_PATH, "userdata")
+os.makedirs(BASE_USER_DIR, exist_ok=True)
+
+# ======================================================
+# 🔧 APP CONFIG
+# ======================================================
+app = Flask(__name__)
+
+# ======================================================
+# 🧠 CORE LOGIC (SIMPLIFIED FOR ROBUSTNESS)
 # ======================================================
 @spaces.GPU(duration=300)
-def generate_comic_gpu(video_path, user_dir, frames_dir, metadata_path, target_pages):
+def generate_comic_gpu(video_path, user_dir, frames_dir, metadata_path, target_pages, panels_per_page_req):
     import cv2
     import numpy as np
     
+    # 1. Setup
+    if os.path.exists(frames_dir): shutil.rmtree(frames_dir)
+    os.makedirs(frames_dir, exist_ok=True)
+    
+    # 2. Extract Frames
     cap = cv2.VideoCapture(video_path)
-    if not cap.isOpened(): raise Exception("Cannot open video")
     fps = cap.get(cv2.CAP_PROP_FPS) or 25
-    duration = int(cap.get(cv2.CAP_PROP_FRAME_COUNT)) / fps
-    cap.release()
-
-    # We force 2 panels per page for the vertical slant layout
+    total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
+    duration = total_frames / fps
+    
+    # Default to 2 panels per page (Split layout)
     panels_per_page = 2
-    total_panels_needed = int(target_pages) * panels_per_page
-    times = np.linspace(1, max(1, duration - 1), total_panels_needed)
+    total_panels = target_pages * panels_per_page
+    
+    # Pick timestamps
+    times = np.linspace(1, duration - 1, total_panels)
     
-    frame_metadata = {}
     frame_files_ordered = []
-    cap = cv2.VideoCapture(video_path)
+    frame_metadata = {}
+    
     for i, t in enumerate(times):
         cap.set(cv2.CAP_PROP_POS_MSEC, t * 1000)
         ret, frame = cap.read()
         if ret:
             fname = f"frame_{i:04d}.png"
+            # Resize for vertical quality
+            h, w = frame.shape[:2]
+            if w > h: # If landscape, crop center
+                center = w // 2
+                start = center - (h // 2)
+                frame = frame[:, start:start+h]
+            
             p = os.path.join(frames_dir, fname)
             cv2.imwrite(p, frame)
-            frame_metadata[fname] = {'dialogue': f"Moment {i+1}", 'time': t}
+            
+            frame_metadata[fname] = {'dialogue': "Edit Text", 'time': t}
             frame_files_ordered.append(fname)
-    cap.release()
-
-    with open(metadata_path, 'w') as f: json.dump(frame_metadata, f)
             
+    cap.release()
+    
+    # 3. Build Page Structure
     pages = []
-    for i in range(int(target_pages)):
-        start_idx = i * 2
-        p_frames = frame_files_ordered[start_idx:start_idx+2]
-        if len(p_frames) == 2:
-            pg_panels = [{'image': f} for f in p_frames]
-            pg_bubbles = [{'dialog': frame_metadata[f]['dialogue']} for f in p_frames]
-            pages.append({'panels': pg_panels, 'bubbles': pg_bubbles})
-    return pages
+    for i in range(target_pages):
+        start_idx = i * panels_per_page
+        current_frames = frame_files_ordered[start_idx : start_idx + panels_per_page]
+        
+        # Create Page Object
+        pg_panels = [{'image': f} for f in current_frames]
+        # Add a default bubble to the first panel of each page
+        pg_bubbles = []
+        if len(pg_panels) > 0:
+            pg_bubbles.append({
+                'dialog': "Your text here...", 
+                'type': 'speech', 
+                'layout_idx': 0 # Belongs to first panel
+            })
+            
+        pages.append({
+            'panels': pg_panels,
+            'bubbles': pg_bubbles,
+            'splitT': '45%', # Default top split position
+            'splitB': '55%'  # Default bottom split position
+        })
 
-@spaces.GPU
-def regen_frame_gpu(video_path, frames_dir, metadata_path, fname, direction):
-    import cv2
-    import json
-    with open(metadata_path, 'r') as f: meta = json.load(f)
-    t = meta[fname]['time']
-    cap = cv2.VideoCapture(video_path)
-    fps = cap.get(cv2.CAP_PROP_FPS) or 25
-    new_t = max(0, t + (1.0/fps * (1 if direction == 'forward' else -1)))
-    cap.set(cv2.CAP_PROP_POS_MSEC, new_t * 1000)
-    ret, frame = cap.read()
-    cap.release()
-    if ret:
-        cv2.imwrite(os.path.join(frames_dir, fname), frame)
-        meta[fname]['time'] = new_t
-        with open(metadata_path, 'w') as f: json.dump(meta, f)
-        return {"success": True}
-    return {"success": False}
+    return pages
 
 # ======================================================
-# 💾 STORAGE SETUP
+# 💻 BACKEND CLASS
 # ======================================================
-BASE_STORAGE_PATH = '/data' if os.path.exists('/data') else '.'
-BASE_USER_DIR = os.path.join(BASE_STORAGE_PATH, "userdata")
-os.makedirs(BASE_USER_DIR, exist_ok=True)
-
-app = Flask(__name__)
-
-class ComicBackend:
+class EnhancedComicGenerator:
     def __init__(self, sid):
         self.sid = sid
-        self.dir = os.path.join(BASE_USER_DIR, sid)
-        self.v_path = os.path.join(self.dir, 'uploaded.mp4')
-        self.f_dir = os.path.join(self.dir, 'frames')
-        self.o_dir = os.path.join(self.dir, 'output')
-        for d in [self.f_dir, self.o_dir]: os.makedirs(d, exist_ok=True)
-        self.meta_p = os.path.join(self.f_dir, 'metadata.json')
-
-    def write_status(self, msg, prog):
-        with open(os.path.join(self.o_dir, 'status.json'), 'w') as f:
-            json.dump({'message': msg, 'progress': prog}, f)
+        self.user_dir = os.path.join(BASE_USER_DIR, sid)
+        self.video_path = os.path.join(self.user_dir, 'uploaded.mp4')
+        self.frames_dir = os.path.join(self.user_dir, 'frames')
+        self.output_dir = os.path.join(self.user_dir, 'output')
+        os.makedirs(self.frames_dir, exist_ok=True)
+        os.makedirs(self.output_dir, exist_ok=True)
+        self.metadata_path = os.path.join(self.frames_dir, 'frame_metadata.json')
 
-    def run(self, target_pages):
+    def run(self, target_pages, panels_per_page):
         try:
-            self.write_status("Running ZeroGPU Extraction...", 20)
-            data = generate_comic_gpu(self.v_path, self.dir, self.f_dir, self.meta_p, target_pages)
-            with open(os.path.join(self.o_dir, 'pages.json'), 'w') as f: json.dump(data, f)
+            self.write_status("Generating Frames...", 10)
+            data = generate_comic_gpu(self.video_path, self.user_dir, self.frames_dir, self.metadata_path, int(target_pages), int(panels_per_page))
+            
+            with open(os.path.join(self.output_dir, 'pages.json'), 'w') as f:
+                json.dump(data, f, indent=2)
             self.write_status("Complete!", 100)
         except Exception as e:
+            traceback.print_exc()
             self.write_status(f"Error: {str(e)}", -1)
 
+    def write_status(self, msg, prog):
+        with open(os.path.join(self.output_dir, 'status.json'), 'w') as f:
+            json.dump({'message': msg, 'progress': prog}, f)
+
 # ======================================================
-# 🌐 UI & TEMPLATE (Vertical Tilt Feature)
+# 🌐 HTML FRONTEND
 # ======================================================
 INDEX_HTML = '''
-<!DOCTYPE html><html lang="en"> <head> <meta charset="UTF-8"> <title>🎬 Slanted Divider 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=Comic+Neue:wght@700&family=Lato&display=swap" rel="stylesheet"> <style>
-    body { background-color: #fdf6e3; font-family: 'Lato', sans-serif; margin: 0; padding: 20px; }
-    #upload-container { display: flex; justify-content: center; align-items: center; min-height: 80vh; }
-    .upload-box { background: white; padding: 40px; border-radius: 12px; box-shadow: 0 8px 30px rgba(0,0,0,0.1); text-align: center; width: 400px; }
-    
-    /* COMIC PAGE - VERTICAL SLANT */
-    .comic-page { 
-        background: white; width: 864px; height: 1080px; margin: 20px auto;
-        position: relative; overflow: hidden; border: 8px solid #000; box-shadow: 0 10px 30px rgba(0,0,0,0.2);
-    }
-    .comic-grid { width: 100%; height: 100%; position: relative; background: #000; }
-    .panel { position: absolute; top:0; left:0; width:100%; height:100%; overflow:hidden; }
-    
-    /* THE TILT LOGIC */
-    /* Panel 1 is clipped to show the left side */
-    .panel:nth-child(1) { 
-        z-index: 2; 
-        clip-path: polygon(0 0, var(--split-t, 45%) 0, var(--split-b, 55%) 100%, 0 100%); 
-    }
-    /* Panel 2 is behind and clipped to show the right side */
-    .panel:nth-child(2) { 
-        z-index: 1; 
-        clip-path: polygon(var(--split-t, 45%) 0, 100% 0, 100% 100%, var(--split-b, 55%) 100%); 
-    }
-    
-    .panel img { width: 100%; height: 100%; object-fit: cover; pointer-events: none; }
-    
-    /* DRAGGABLE HANDLES ON THE EDGES */
-    .split-handle {
-        position: absolute; width: 34px; height: 34px; 
-        background: #FF9800; border: 4px solid white; border-radius: 50%;
-        cursor: ew-resize; z-index: 100; box-shadow: 0 4px 10px rgba(0,0,0,0.4);
-    }
-    /* Top edge handle */
-    .split-handle.top { top: -17px; left: var(--split-t, 45%); transform: translateX(-50%); }
-    /* Bottom edge handle */
-    .split-handle.bottom { bottom: -17px; left: var(--split-b, 55%); transform: translateX(-50%); }
-
-    /* CONTROLS */
-    .edit-controls { position: fixed; right: 20px; top: 20px; width: 220px; background: #2c3e50; color: white; padding: 15px; border-radius: 10px; z-index: 999; }
-    button { width: 100%; padding: 10px; margin: 5px 0; border: none; border-radius: 5px; cursor: pointer; font-weight: bold; }
-    .btn-gen { background: #27ae60; color: white; }
-    .btn-exp { background: #2980b9; color: white; }
-    #status { color: #f1c40f; font-family: monospace; font-size: 12px; margin-top: 10px; }
-</style> </head> <body>
+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>Vertical Split 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&display=swap" rel="stylesheet">
+    <style>
+        body { background: #222; font-family: 'Comic Neue', cursive; margin: 0; color: white; }
+        
+        /* UPLOAD SCREEN */
+        #upload-container { display: flex; flex-direction: column; align-items: center; justify-content: center; height: 100vh; }
+        .box { background: #333; padding: 40px; border-radius: 10px; text-align: center; }
+        .btn { padding: 10px 20px; background: #e67e22; border: none; color: white; cursor: pointer; font-size: 18px; border-radius: 5px; margin-top: 10px; }
+        .btn:hover { background: #d35400; }
+        input[type=file] { margin-bottom: 20px; }
+
+        /* EDITOR SCREEN */
+        #editor-container { display: none; padding: 20px; text-align: center; }
+        
+        .comic-wrapper { 
+            display: flex; 
+            flex-direction: row; 
+            flex-wrap: wrap; 
+            justify-content: center; 
+            gap: 40px; 
+            padding-bottom: 100px;
+        }
+
+        /* PAGE CONTAINER */
+        .comic-page {
+            width: 400px; /* Reduced width for preview */
+            height: 600px;
+            background: black;
+            position: relative;
+            border: 4px solid white;
+            box-shadow: 0 0 20px rgba(0,0,0,0.5);
+            overflow: visible; /* Allow handles to stick out */
+            user-select: none;
+        }
+
+        /* THE GRID HOLDER */
+        .comic-grid {
+            width: 100%;
+            height: 100%;
+            position: relative;
+            /* Default split variables */
+            --split-t: 45%; 
+            --split-b: 55%;
+        }
 
+        /* PANELS - CLIPPED */
+        .panel {
+            position: absolute;
+            top: 0; left: 0;
+            width: 100%; height: 100%;
+            overflow: hidden;
+            pointer-events: none; /* Let clicks pass through to bubbles/handles if needed */
+        }
+        
+        .panel img {
+            width: 100%; height: 100%;
+            object-fit: cover;
+            pointer-events: auto; /* Re-enable pointer events for dragging image if needed */
+        }
+
+        /* Left/Top Panel */
+        .panel:nth-child(1) {
+            z-index: 2;
+            clip-path: polygon(0 0, var(--split-t) 0, var(--split-b) 100%, 0 100%);
+            border-right: 2px solid black; /* Fake border */
+        }
+
+        /* Right/Bottom Panel */
+        .panel:nth-child(2) {
+            z-index: 1;
+            clip-path: polygon(var(--split-t) 0, 100% 0, 100% 100%, var(--split-b) 100%);
+        }
+
+        /* DRAG HANDLES FOR SPLIT LINE */
+        .split-handle {
+            position: absolute;
+            width: 24px; height: 24px;
+            background: #2196F3;
+            border: 2px solid white;
+            border-radius: 50%;
+            transform: translate(-50%, -50%);
+            cursor: col-resize;
+            z-index: 999;
+            box-shadow: 0 2px 5px rgba(0,0,0,0.5);
+            pointer-events: auto;
+        }
+        
+        /* Top Handle */
+        .split-handle.top { top: 0%; left: var(--split-t); margin-top: -12px; }
+        /* Bottom Handle */
+        .split-handle.bottom { top: 100%; left: var(--split-b); margin-top: 12px; }
+
+        /* LINE VISUALIZATION */
+        .split-line-visual {
+            position: absolute; top:0; left:0; width:100%; height:100%;
+            pointer-events: none; z-index: 10;
+            background: transparent;
+        }
+
+        /* BUBBLES */
+        .speech-bubble {
+            position: absolute;
+            background: white; color: black;
+            padding: 10px; border-radius: 10px;
+            min-width: 80px; text-align: center;
+            font-family: 'Comic Neue'; font-weight: bold;
+            cursor: move; z-index: 100;
+            border: 2px solid black;
+            transform: translate(-50%, -50%);
+        }
+        
+        .speech-bubble:after {
+            content: ''; position: absolute; bottom: -10px; left: 50%;
+            border-width: 10px 10px 0; border-style: solid;
+            border-color: black transparent; display: block; width: 0;
+        }
+        
+        .speech-bubble:before {
+            content: ''; position: absolute; bottom: -6px; left: 50%;
+            border-width: 10px 10px 0; border-style: solid;
+            border-color: white transparent; display: block; width: 0; z-index: 1;
+        }
+
+        /* CONTROLS */
+        .controls {
+            position: fixed; bottom: 0; left: 0; width: 100%;
+            background: #333; padding: 15px; display: flex;
+            justify-content: center; gap: 20px; z-index: 1000;
+        }
+    </style>
+</head>
+<body>
+
+<!-- UPLOAD -->
 <div id="upload-container">
-    <div class="upload-box">
-        <h1>🎬 Vertical Slant Gen</h1>
-        <input type="file" id="vid-file" style="display:none" onchange="document.getElementById('fn').innerText=this.files[0].name">
-        <button onclick="document.getElementById('vid-file').click()" class="btn-gen">📁 Choose Video</button>
-        <p id="fn">No file selected</p>
-        <button class="btn-gen" onclick="upload()" style="background:#e67e22">🚀 Generate Comic</button>
-        <div id="status">Ready.</div>
+    <div class="box">
+        <h1>🎬 Vertical Split Comic</h1>
+        <input type="file" id="file-upload" accept="video/*"><br>
+        <label>Pages:</label> <input type="number" id="page-count" value="2" min="1" max="5" style="width:50px;">
+        <button class="btn" onclick="startUpload()">Generate Comic</button>
+        <p id="status" style="margin-top:10px; color:#aaa;"></p>
     </div>
 </div>
 
-<div id="editor-container" style="display:none">
-    <div id="comic-list"></div>
-    <div class="edit-controls">
-        <h4>Panel Editor</h4>
-        <p style="font-size: 11px; opacity: 0.8;">Drag the <b style="color:#FF9800">orange circles</b> at the top and bottom to tilt the middle line.</p>
-        <button class="btn-exp" onclick="exportPNG()">📥 Export PNG</button>
-        <button onclick="location.reload()" style="background:#c0392b; color:white;">Reset</button>
+<!-- EDITOR -->
+<div id="editor-container">
+    <h2>Adjust the Split Line by Dragging the Blue Dots</h2>
+    <div class="comic-wrapper" id="comic-container"></div>
+    
+    <div class="controls">
+        <button class="btn" onclick="addBubble()">+ Text Bubble</button>
+        <button class="btn" onclick="downloadAll()">Download Images</button>
+        <button class="btn" style="background:#c0392b" onclick="location.reload()">Reset</button>
     </div>
 </div>
 
 <script>
-    let sid = 'S' + Math.floor(Math.random()*1000000);
-    let isDragging = false, currentHandle = null;
-
-    async function upload() {
-        const file = document.getElementById('vid-file').files[0];
-        if(!file) return alert("Select a video");
-        const fd = new FormData(); fd.append('file', file);
-        document.getElementById('status').innerText = "Uploading to ZeroGPU...";
-        await fetch(`/uploader?sid=${sid}`, {method:'POST', body:fd});
+    let sid = 'S' + Math.floor(Math.random()*100000);
+    let isDraggingHandle = false;
+    let activeHandle = null; // { element: domNode, type: 'top'|'bottom', grid: domNode }
+    
+    let isDraggingBubble = false;
+    let activeBubble = null;
+    let offset = {x:0, y:0};
+
+    // 1. UPLOAD LOGIC
+    async function startUpload() {
+        const file = document.getElementById('file-upload').files[0];
+        if(!file) return alert("Please select a file");
+        
+        const pages = document.getElementById('page-count').value;
+        const fd = new FormData();
+        fd.append('file', file);
+        fd.append('target_pages', pages);
+        fd.append('panels_per_page', 2); // FORCE 2 PANELS
         
-        const timer = setInterval(async () => {
-            const r = await fetch(`/status?sid=${sid}`);
-            const d = await r.json();
-            document.getElementById('status').innerText = d.message;
-            if(d.progress >= 100) { clearInterval(timer); loadEditor(); }
-        }, 2000);
+        document.querySelector('.box').style.display = 'none';
+        document.getElementById('status').innerText = "Uploading...";
+        
+        const res = await fetch(`/uploader?sid=${sid}`, { method: 'POST', body: fd });
+        if(res.ok) {
+            checkStatus();
+        }
     }
 
-    async function loadEditor() {
-        const r = await fetch(`/output/pages.json?sid=${sid}`);
-        const pages = await r.json();
-        const list = document.getElementById('comic-list');
-        pages.forEach(p => {
-            const pgDiv = document.createElement('div');
-            pgDiv.className = 'comic-page';
-            const grid = document.createElement('div');
-            grid.className = 'comic-grid';
-            grid.style.setProperty('--split-t', '45%');
-            grid.style.setProperty('--split-b', '55%');
-
-            const hTop = document.createElement('div'); hTop.className = 'split-handle top';
-            const hBot = document.createElement('div'); hBot.className = 'split-handle bottom';
+    async function checkStatus() {
+        const interval = setInterval(async () => {
+            const res = await fetch(`/status?sid=${sid}`);
+            const data = await res.json();
+            document.getElementById('status').innerText = data.message;
             
-            hTop.onmousedown = (e) => { isDragging = true; currentHandle = { grid, key: '--split-t' }; };
-            hBot.onmousedown = (e) => { isDragging = true; currentHandle = { grid, key: '--split-b' }; };
-
-            p.panels.forEach(pan => {
-                const div = document.createElement('div');
-                div.className = 'panel';
-                div.innerHTML = `<img src="/frames/${pan.image}?sid=${sid}">`;
-                grid.appendChild(div);
-            });
-
-            grid.appendChild(hTop); grid.appendChild(hBot);
-            pgDiv.appendChild(grid);
-            list.appendChild(pgDiv);
+            if(data.progress >= 100) {
+                clearInterval(interval);
+                loadComic();
+            }
+        }, 1000);
+    }
+
+    // 2. RENDER COMIC
+    async function loadComic() {
+        document.getElementById('upload-container').style.display = 'none';
+        document.getElementById('editor-container').style.display = 'block';
+        
+        const res = await fetch(`/output/pages.json?sid=${sid}`);
+        const pages = await res.json();
+        
+        const container = document.getElementById('comic-container');
+        container.innerHTML = '';
+        
+        pages.forEach((pageData, idx) => {
+            renderPage(pageData, idx, container);
         });
-        document.getElementById('upload-container').style.display='none';
-        document.getElementById('editor-container').style.display='block';
     }
 
-    window.onmousemove = (e) => {
-        if(!isDragging || !currentHandle) return;
-        const rect = currentHandle.grid.getBoundingClientRect();
-        let x = ((e.clientX - rect.left) / rect.width) * 100;
-        x = Math.max(5, Math.min(95, x)); // Constraints
-        currentHandle.grid.style.setProperty(currentHandle.key, x + '%');
-    };
-    window.onmouseup = () => isDragging = false;
-
-    async function exportPNG() {
-        const pgs = document.querySelectorAll('.comic-page');
-        for(let i=0; i<pgs.length; i++){
-            const url = await htmlToImage.toPng(pgs[i]);
-            const a = document.createElement('a');
-            a.download = `comic-page-${i+1}.png`; a.href = url; a.click();
+    function renderPage(data, idx, container) {
+        const pageDiv = document.createElement('div');
+        pageDiv.className = 'comic-page';
+        
+        // Grid Container
+        const grid = document.createElement('div');
+        grid.className = 'comic-grid';
+        grid.style.setProperty('--split-t', data.splitT || '45%');
+        grid.style.setProperty('--split-b', data.splitB || '55%');
+        
+        // Panels (Assuming 2 panels always)
+        if(data.panels.length >= 2) {
+            // Panel 1 (Left)
+            const p1 = document.createElement('div');
+            p1.className = 'panel';
+            p1.innerHTML = `<img src="/frames/${data.panels[0].image}?sid=${sid}">`;
+            grid.appendChild(p1);
+
+            // Panel 2 (Right)
+            const p2 = document.createElement('div');
+            p2.className = 'panel';
+            p2.innerHTML = `<img src="/frames/${data.panels[1].image}?sid=${sid}">`;
+            grid.appendChild(p2);
+        }
+
+        // HANDLES
+        const hTop = document.createElement('div');
+        hTop.className = 'split-handle top';
+        hTop.onmousedown = (e) => startDragHandle(e, hTop, 'top', grid);
+        
+        const hBot = document.createElement('div');
+        hBot.className = 'split-handle bottom';
+        hBot.onmousedown = (e) => startDragHandle(e, hBot, 'bottom', grid);
+        
+        grid.appendChild(hTop);
+        grid.appendChild(hBot);
+
+        // BUBBLES
+        if(data.bubbles) {
+            data.bubbles.forEach(b => createBubble(b.dialog, grid));
+        }
+
+        pageDiv.appendChild(grid);
+        container.appendChild(pageDiv);
+    }
+
+    // 3. INTERACTION LOGIC (DRAG HANDLES)
+    function startDragHandle(e, el, type, grid) {
+        e.preventDefault();
+        e.stopPropagation();
+        isDraggingHandle = true;
+        activeHandle = { el, type, grid };
+    }
+
+    // 4. INTERACTION LOGIC (BUBBLES)
+    function createBubble(text, parent) {
+        const b = document.createElement('div');
+        b.className = 'speech-bubble';
+        b.contentEditable = true;
+        b.innerText = text;
+        b.style.left = '50%';
+        b.style.top = '50%';
+        
+        b.onmousedown = (e) => {
+            if(e.target !== b) return; // Allow text selection
+            e.preventDefault();
+            isDraggingBubble = true;
+            activeBubble = b;
+            offset.x = e.clientX - b.getBoundingClientRect().left - (b.offsetWidth/2);
+            offset.y = e.clientY - b.getBoundingClientRect().top - (b.offsetHeight/2);
+        };
+        
+        parent.appendChild(b);
+    }
+    
+    function addBubble() {
+        const grids = document.querySelectorAll('.comic-grid');
+        if(grids.length > 0) createBubble("New Text", grids[0]);
+    }
+
+    // GLOBAL MOUSE EVENTS
+    document.addEventListener('mousemove', (e) => {
+        // Handle Dragging
+        if(isDraggingHandle && activeHandle) {
+            const rect = activeHandle.grid.getBoundingClientRect();
+            let relativeX = e.clientX - rect.left;
+            
+            // Constrain
+            if(relativeX < 0) relativeX = 0;
+            if(relativeX > rect.width) relativeX = rect.width;
+            
+            const pct = (relativeX / rect.width) * 100;
+            
+            if(activeHandle.type === 'top') {
+                activeHandle.grid.style.setProperty('--split-t', pct + '%');
+            } else {
+                activeHandle.grid.style.setProperty('--split-b', pct + '%');
+            }
+        }
+
+        // Bubble Dragging
+        if(isDraggingBubble && activeBubble) {
+            const rect = activeBubble.parentElement.getBoundingClientRect();
+            let x = e.clientX - rect.left; // - offset.x;
+            let y = e.clientY - rect.top; // - offset.y;
+            
+            activeBubble.style.left = x + 'px';
+            activeBubble.style.top = y + 'px';
+        }
+    });
+
+    document.addEventListener('mouseup', () => {
+        isDraggingHandle = false;
+        activeHandle = null;
+        isDraggingBubble = false;
+        activeBubble = null;
+    });
+
+    // 5. EXPORT
+    async function downloadAll() {
+        const pages = document.querySelectorAll('.comic-page');
+        for(let i=0; i<pages.length; i++) {
+            // Hide handles for screenshot
+            const handles = pages[i].querySelectorAll('.split-handle');
+            handles.forEach(h => h.style.display = 'none');
+            
+            const dataUrl = await htmlToImage.toPng(pages[i]);
+            const link = document.createElement('a');
+            link.download = `comic_page_${i+1}.png`;
+            link.href = dataUrl;
+            link.click();
+            
+            // Show handles again
+            handles.forEach(h => h.style.display = 'block');
         }
     }
-</script> </body> </html>
+</script>
+</body>
+</html>
 '''
 
+# ======================================================
+# 🚀 FLASK ROUTES
+# ======================================================
 @app.route('/')
-def index(): return INDEX_HTML
+def index():
+    return INDEX_HTML
 
 @app.route('/uploader', methods=['POST'])
-def uploader():
+def upload():
     sid = request.args.get('sid')
+    target_pages = request.form.get('target_pages', 2)
+    panels_per_page = 2 # Fixed for this specific layout req
+    
     f = request.files['file']
-    gen = ComicBackend(sid)
-    f.save(gen.v_path)
-    threading.Thread(target=gen.run, args=(2,)).start()
-    return jsonify({'ok': True})
+    gen = EnhancedComicGenerator(sid)
+    f.save(gen.video_path)
+    
+    # Run in background
+    threading.Thread(target=gen.run, args=(target_pages, panels_per_page)).start()
+    return jsonify({'success': True})
 
 @app.route('/status')
 def get_status():
     sid = request.args.get('sid')
-    try:
-        with open(os.path.join(BASE_USER_DIR, sid, 'output', 'status.json'), 'r') as f:
-            return f.read()
-    except: return jsonify({'message': 'Starting...', 'progress': 0})
+    path = os.path.join(BASE_USER_DIR, sid, 'output', 'status.json')
+    if os.path.exists(path): return send_file(path)
+    return jsonify({'progress': 0, 'message': "Waiting..."})
 
-@app.route('/output/<path:filename>')
+@app.route('/output/path:filename')
 def get_output(filename):
     sid = request.args.get('sid')
     return send_from_directory(os.path.join(BASE_USER_DIR, sid, 'output'), filename)
 
-@app.route('/frames/<path:filename>')
+@app.route('/frames/path:filename')
 def get_frame(filename):
     sid = request.args.get('sid')
     return send_from_directory(os.path.join(BASE_USER_DIR, sid, 'frames'), filename)
 
 if __name__ == '__main__':
+    try: gpu_warmup()
+    except: pass
     app.run(host='0.0.0.0', port=7860)