Update app_enhanced.py

app_enhanced.py

@@ -1,35 +1,49 @@
+import spaces  # <--- MUST BE THE FIRST IMPORT
 import os
 import time
 import threading
-import uuid
-import shutil
 import json
 import traceback
 import logging
 import string
 import random
-from concurrent.futures import ThreadPoolExecutor
-from flask import Flask, render_template, request, jsonify, send_from_directory, send_file
+import shutil
+from flask import Flask, jsonify, request, send_from_directory, send_file
 
-# ==========================================
-# 🚀 ZEROGPU IMPORTS & SETUP
-# ==========================================
-import spaces  # Must be at the top level
-import torch   # Import torch explicitly
+# ======================================================
+# 🚀 ZEROGPU DETECTION FUNCTION
+# ======================================================
+# This function exists solely to be detected by Hugging Face
+@spaces.GPU
+def gpu_warmup():
+    import torch
+    print(f"✅ ZeroGPU Warmup: CUDA Available: {torch.cuda.is_available()}")
+    return True
 
-# We define GPU functions GLOBALLY so the scanner finds them.
-# We pass only necessary paths/data to avoid pickling the whole class.
+# ======================================================
+# 🔧 CONFIGURATION
+# ======================================================
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+app = Flask(__name__)
+BASE_USER_DIR = "userdata" 
+SAVED_COMICS_DIR = "saved_comics"
 
-@spaces.GPU(duration=120)
+os.makedirs(BASE_USER_DIR, exist_ok=True)
+os.makedirs(SAVED_COMICS_DIR, exist_ok=True)
+
+# ======================================================
+# 🧠 MAIN GPU LOGIC
+# ======================================================
+@spaces.GPU(duration=180)
 def generate_comic_gpu(video_path, user_dir, frames_dir, metadata_path):
-    """
-    Standalone GPU function to handle the heavy lifting.
-    """
     print(f"🚀 GPU Task Started for {video_path}")
     
-    # 1. Imports inside to ensure they run in the GPU environment
+    # Imports inside function to ensure they use the GPU environment
     import cv2
     import srt
+    import numpy as np
     from backend.keyframes.keyframes import black_bar_crop
     from backend.simple_color_enhancer import SimpleColorEnhancer
     from backend.quality_color_enhancer import QualityColorEnhancer
@@ -38,202 +52,132 @@ def generate_comic_gpu(video_path, user_dir, frames_dir, metadata_path):
     from backend.ai_enhanced_core import face_detector
     from backend.class_def import bubble, panel, Page
     
-    # 2. Analyze Video
+    # 1. Analyze Video
     cap = cv2.VideoCapture(video_path)
-    if not cap.isOpened():
-        raise Exception("Cannot open video")
-    video_fps = cap.get(cv2.CAP_PROP_FPS) or 25
+    fps = cap.get(cv2.CAP_PROP_FPS) or 25
     total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
-    duration = total_frames / video_fps
+    duration = total_frames / fps
     cap.release()
     
-    # 3. Subtitles
+    # 2. Subtitles
     user_srt = os.path.join(user_dir, 'subs.srt')
     try:
         get_real_subtitles(video_path)
         if os.path.exists('test1.srt'):
             shutil.move('test1.srt', user_srt)
     except Exception as e:
-        print(f"⚠️ Subtitle generation failed: {e}. Creating fallback.")
-        with open(user_srt, 'w') as f:
-            f.write("1\n00:00:01,000 --> 00:00:04,000\nHello\n")
+        print(f"Subtitle fallback used: {e}")
+        with open(user_srt, 'w') as f: f.write("1\n00:00:01,000 --> 00:00:04,000\nHello\n")
             
     with open(user_srt, 'r', encoding='utf-8') as f:
         all_subs = list(srt.parse(f.read()))
     
-    key_moments = [{
-        'index': s.index,
-        'text': s.content,
-        'start': s.start.total_seconds(),
-        'end': s.end.total_seconds()
-    } for s in all_subs]
-    
-    # 4. Extract Keyframes
+    key_moments = [{'index':s.index, 'text':s.content, 'start':s.start.total_seconds(), 'end':s.end.total_seconds()} for s in all_subs]
     key_moments.sort(key=lambda x: x['start'])
+    
+    # 3. Extract Frames
     frame_metadata = {}
-    frame_count = 0
     cap = cv2.VideoCapture(video_path)
+    count = 0
     
-    max_frames = 48
-    for i, moment in enumerate(key_moments[:max_frames]):
-        frame_time = (moment['start'] + moment['end']) / 2
-        if frame_time > duration: continue
+    # Limit to 12 frames for speed/demo purposes on ZeroGPU, increase if needed
+    for i, moment in enumerate(key_moments[:24]): 
+        mid_time = (moment['start'] + moment['end']) / 2
+        if mid_time > duration: continue
         
-        frame_number = int(frame_time * video_fps)
-        cap.set(cv2.CAP_PROP_POS_FRAMES, frame_number)
+        cap.set(cv2.CAP_PROP_POS_FRAMES, int(mid_time * fps))
         ret, frame = cap.read()
-        
         if ret:
-            fname = f"frame_{frame_count:04d}.png"
-            fpath = os.path.join(frames_dir, fname)
-            cv2.imwrite(fpath, frame)
-            frame_metadata[fname] = {
-                'time': frame_time,
-                'dialogue': moment['text'],
-                'start': moment['start'],
-                'end': moment['end']
-            }
-            frame_count += 1
+            fname = f"frame_{count:04d}.png"
+            full_p = os.path.join(frames_dir, fname)
+            cv2.imwrite(full_p, frame)
+            frame_metadata[fname] = {'dialogue': moment['text'], 'time': mid_time}
+            count += 1
     cap.release()
     
-    with open(metadata_path, 'w') as f:
-        json.dump(frame_metadata, f, indent=2)
-        
-    # 5. Enhance & Crop
-    try:
-        black_bar_crop() # Assuming it processes files in place or globally configured
-    except: pass
+    with open(metadata_path, 'w') as f: json.dump(frame_metadata, f)
 
-    simple_enhancer = SimpleColorEnhancer()
-    quality_enhancer = QualityColorEnhancer()
+    # 4. Enhance
+    try: black_bar_crop() 
+    except: pass
     
+    se = SimpleColorEnhancer()
+    qe = QualityColorEnhancer()
     frame_files = sorted([f for f in os.listdir(frames_dir) if f.endswith('.png')])
+    
     for f in frame_files:
-        full_p = os.path.join(frames_dir, f)
-        simple_enhancer.enhance_single(full_p)
-        quality_enhancer.enhance_single(full_p)
+        p = os.path.join(frames_dir, f)
+        se.enhance_single(p)
+        qe.enhance_single(p)
         
-    # 6. Bubbles & Layout
+    # 5. Bubbles
     bubbles_list = []
     for f in frame_files:
-        full_p = os.path.join(frames_dir, f)
-        meta = frame_metadata.get(f, {})
-        dialogue = meta.get('dialogue', '')
-        
+        p = os.path.join(frames_dir, f)
+        dialogue = frame_metadata.get(f, {}).get('dialogue', '')
         try:
-            faces = face_detector.detect_faces(full_p)
-            lip = face_detector.get_lip_position(full_p, faces[0]) if faces else (-1, -1)
-            bx, by = ai_bubble_placer.place_bubble_ai(full_p, lip)
+            faces = face_detector.detect_faces(p)
+            lip = face_detector.get_lip_position(p, faces[0]) if faces else (-1, -1)
+            bx, by = ai_bubble_placer.place_bubble_ai(p, lip)
             bubbles_list.append(bubble(dialog=dialogue, bubble_offset_x=bx, bubble_offset_y=by, lip_x=lip[0], lip_y=lip[1]))
         except:
             bubbles_list.append(bubble(dialog=dialogue))
             
-    # 7. Generate Pages
+    # 6. Layout
     try:
         from backend.fixed_12_pages_800x1080 import generate_12_pages_800x1080
         pages = generate_12_pages_800x1080(frame_files, bubbles_list)
     except:
-        # Fallback layout
         pages = []
-        num_pages = (len(frame_files) + 3) // 4
-        for i in range(num_pages):
-            start, end = i * 4, (i + 1) * 4
-            p_panels = [panel(image=f) for f in frame_files[start:end]]
-            p_bubbles = bubbles_list[start:end]
-            if p_panels:
-                pages.append(Page(panels=p_panels, bubbles=p_bubbles))
-                
-    # Return simple data structure to main thread
-    pages_data = []
-    for page in pages:
-        panels = [p.__dict__ if hasattr(p, '__dict__') else p for p in page.panels]
-        bubbles_data = [b.__dict__ if hasattr(b, '__dict__') else b for b in page.bubbles]
-        pages_data.append({'panels': panels, 'bubbles': bubbles_data})
+        for i in range((len(frame_files)+3)//4):
+            start = i*4
+            pg_panels = [panel(image=f) for f in frame_files[start:start+4]]
+            pg_bubbles = bubbles_list[start:start+4]
+            if pg_panels: pages.append(Page(panels=pg_panels, bubbles=pg_bubbles))
+
+    # Serialize
+    result = []
+    for pg in pages:
+        p_data = [p.__dict__ if hasattr(p,'__dict__') else p for p in pg.panels]
+        b_data = [b.__dict__ if hasattr(b,'__dict__') else b for b in pg.bubbles]
+        result.append({'panels': p_data, 'bubbles': b_data})
         
-    return pages_data
+    return result
 
 @spaces.GPU
-def regenerate_frame_gpu(video_path, frames_dir, metadata_path, fname, direction):
+def regen_frame_gpu(video_path, frames_dir, metadata_path, fname, direction):
     import cv2
-    from backend.simple_color_enhancer import SimpleColorEnhancer
-    from backend.quality_color_enhancer import QualityColorEnhancer
-    
+    import json
     if not os.path.exists(metadata_path): return None
     with open(metadata_path, 'r') as f: meta = json.load(f)
     if fname not in meta: return None
     
-    current_data = meta[fname]
-    curr_time = current_data['time'] if isinstance(current_data, dict) else current_data
-    
+    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_time = max(0, curr_time + offset)
+    offset = (1.0/fps) * (1 if direction == 'forward' else -1)
+    new_t = max(0, t + offset)
     
-    cap.set(cv2.CAP_PROP_POS_MSEC, new_time * 1000)
+    cap.set(cv2.CAP_PROP_POS_MSEC, new_t * 1000)
     ret, frame = cap.read()
     cap.release()
     
     if ret:
         path = os.path.join(frames_dir, fname)
         cv2.imwrite(path, frame)
+        # Re-enhance
+        from backend.simple_color_enhancer import SimpleColorEnhancer
         SimpleColorEnhancer().enhance_single(path)
-        QualityColorEnhancer().enhance_single(path)
         
-        if isinstance(meta[fname], dict): meta[fname]['time'] = new_time
-        else: meta[fname] = new_time
-        with open(metadata_path, 'w') as f: json.dump(meta, f, indent=2)
-        return new_time
+        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)
+        return new_t
     return None
 
-@spaces.GPU
-def get_frame_at_ts_gpu(video_path, frames_dir, metadata_path, fname, ts):
-    import cv2
-    from backend.simple_color_enhancer import SimpleColorEnhancer
-    from backend.quality_color_enhancer import QualityColorEnhancer
-    
-    cap = cv2.VideoCapture(video_path)
-    cap.set(cv2.CAP_PROP_POS_MSEC, float(ts) * 1000)
-    ret, frame = cap.read()
-    cap.release()
-    
-    if ret:
-        path = os.path.join(frames_dir, fname)
-        cv2.imwrite(path, frame)
-        SimpleColorEnhancer().enhance_single(path)
-        QualityColorEnhancer().enhance_single(path)
-        
-        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 True
-    return False
-
-# ==========================================
-# ⚙️ CONFIG & DEPENDENCIES
-# ==========================================
-logging.basicConfig(level=logging.INFO)
-logger = logging.getLogger(__name__)
-
-# --- FLASK APP SETUP ---
-app = Flask(__name__)
-BASE_USER_DIR = "userdata" 
-SAVED_COMICS_DIR = "saved_comics"
-
-os.makedirs(BASE_USER_DIR, exist_ok=True)
-os.makedirs(SAVED_COMICS_DIR, exist_ok=True)
-
-def generate_save_code(length=8):
-    chars = string.ascii_uppercase + string.digits
-    while True:
-        code = ''.join(random.choices(chars, k=length))
-        if not os.path.exists(os.path.join(SAVED_COMICS_DIR, code)):
-            return code
-
-# --- 3. ENHANCED COMIC GENERATOR CLASS (Wrapper) ---
+# ======================================================
+# 💻 BACKEND CLASS
+# ======================================================
 class EnhancedComicGenerator:
     def __init__(self, sid):
         self.sid = sid
@@ -241,179 +185,149 @@ class EnhancedComicGenerator:
         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')
-        self.status_file = os.path.join(self.output_dir, 'status.json')
-        self.metadata_path = os.path.join(self.frames_dir, 'frame_metadata.json')
         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 update_status(self, message, progress):
-        try:
-            with open(self.status_file, 'w') as f: 
-                json.dump({'message': message, 'progress': progress}, f)
-        except: pass
-
-    def cleanup_previous_run(self):
-        if os.path.exists(self.frames_dir):
-            for f in os.listdir(self.frames_dir):
-                try: os.remove(os.path.join(self.frames_dir, f))
-                except: pass
-        if os.path.exists(self.output_dir):
-            for f in os.listdir(self.output_dir):
-                if f != 'status.json':
-                    try: os.remove(os.path.join(self.output_dir, f))
-                    except: pass
+    def cleanup(self):
+        if os.path.exists(self.frames_dir): shutil.rmtree(self.frames_dir)
+        if os.path.exists(self.output_dir): shutil.rmtree(self.output_dir)
+        os.makedirs(self.frames_dir, exist_ok=True)
+        os.makedirs(self.output_dir, exist_ok=True)
 
-    def run_process(self):
+    def run(self):
         try:
-            self.update_status("Starting GPU Process...", 10)
-            
-            # CALL GLOBAL GPU FUNCTION
-            pages_data = generate_comic_gpu(
-                self.video_path, 
-                self.user_dir, 
-                self.frames_dir, 
-                self.metadata_path
-            )
-            
-            # SAVE RESULTS
-            with open(os.path.join(self.output_dir, 'pages.json'), 'w', encoding='utf-8') as f:
-                json.dump(pages_data, f, indent=2)
-                
-            self.update_status("Complete!", 100)
-            print("✅ Comic generated successfully")
+            self.write_status("Processing on GPU...", 10)
+            data = generate_comic_gpu(self.video_path, self.user_dir, self.frames_dir, self.metadata_path)
+            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.update_status(f"Error: {str(e)}", -1)
+            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)
 
-# --- ROUTES ---
+# ======================================================
+# 🌐 ROUTES
+# ======================================================
 INDEX_HTML = '''
 <!DOCTYPE html>
 <html lang="en">
 <head>
-    <meta charset="UTF-8">
-    <meta name="viewport" content="width=device-width, initial-scale=1.0">
-    <title>Movie to Comic Generator (ZeroGPU)</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; }
-        .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; }
-        .file-label { display: block; padding: 15px; background: #2c3e50; color: white; border-radius: 8px; cursor: pointer; font-weight: bold; margin-bottom: 10px; transition:0.2s; }
-        .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; margin: 20px auto;}
-        .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 img { width: 100%; height: 100%; object-fit: cover; }
-        .speech-bubble { position: absolute; display: flex; justify-content: center; align-items: center; width: 150px; height: 80px; z-index: 10; cursor: move; font-family: 'Comic Neue', cursive; font-weight: bold; font-size: 13px; text-align: center; background: #4ECDC4; color: white; border-radius: 20px; padding: 10px;}
-        .edit-controls { position: fixed; bottom: 20px; right: 20px; width: 260px; background: rgba(44, 62, 80, 0.95); color: white; padding: 15px; border-radius: 8px; }
-        button { width: 100%; margin-top: 5px; padding: 6px; border-radius: 4px; border: 1px solid #ddd; cursor: pointer; }
-    </style>
+<meta charset="UTF-8"><meta name="viewport" content="width=device-width, initial-scale=1.0">
+<title>ComicGen ZeroGPU</title>
+<style>
+body{font-family:sans-serif;background:#f0f2f5;display:flex;flex-direction:column;align-items:center;padding:20px;}
+.box{background:white;padding:30px;border-radius:10px;box-shadow:0 4px 12px rgba(0,0,0,0.1);text-align:center;max-width:500px;width:100%;}
+button{background:#007bff;color:white;border:none;padding:10px 20px;border-radius:5px;cursor:pointer;font-size:16px;margin-top:10px;}
+button:hover{background:#0056b3;}
+.comic-page{background:white;border:2px solid #333;margin:20px 0;padding:10px;width:100%;max-width:600px;}
+.panel{position:relative;margin-bottom:10px;border:2px solid #000;}
+.panel img{width:100%;display:block;}
+.bubble{position:absolute;background:#fff;border:2px solid #000;border-radius:20px;padding:10px;font-size:12px;text-align:center;min-width:50px;}
+</style>
 </head>
 <body>
-    <div id="upload-container">
-        <div class="upload-box">
-            <h1>🎬 Comic Generator</h1>
-            <input type="file" id="file-upload" style="display:none" onchange="document.getElementById('fn').innerText=this.files[0].name">
-            <label for="file-upload" class="file-label">📁 Choose Video File</label>
-            <span id="fn" style="display:block; margin-bottom:10px;">No file selected</span>
-            <button class="submit-btn" onclick="upload()">🚀 Generate Comic</button>
-            <div id="loading" style="display:none; margin-top:20px;">Processing... <span id="prog">0</span>%</div>
-        </div>
-    </div>
-    <div id="editor-container">
-        <div id="comic-container"></div>
-        <div class="edit-controls">
-             <h4>Controls</h4>
-             <button onclick="saveComic()">💾 Save Comic</button>
-             <button onclick="location.reload()">🏠 New Comic</button>
-        </div>
-    </div>
-    <script>
-        let sid = Math.random().toString(36).substring(7);
-        async function upload() {
-            const f = document.getElementById('file-upload').files[0];
-            if(!f) return alert("Select file");
-            document.getElementById('loading').style.display='block';
-            const fd = new FormData(); fd.append('file', f);
-            await fetch(`/uploader?sid=${sid}`, {method:'POST', body:fd});
-            setInterval(checkStatus, 2000);
-        }
-        async function checkStatus() {
-            const r = await fetch(`/status?sid=${sid}`);
-            const d = await r.json();
-            document.getElementById('prog').innerText = d.progress;
-            if(d.progress >= 100) {
-                document.getElementById('upload-container').style.display='none';
-                document.getElementById('editor-container').style.display='block';
-                loadComic();
-            }
-        }
-        async function loadComic() {
-            const r = await fetch(`/output/pages.json?sid=${sid}`);
-            const data = await r.json();
-            const con = document.getElementById('comic-container');
-            data.forEach((page, i) => {
-                const div = document.createElement('div'); div.className='comic-page';
-                const grid = document.createElement('div'); grid.className='comic-grid';
-                page.panels.forEach(pan => {
-                    const p = document.createElement('div'); p.className='panel';
-                    p.innerHTML = `<img src="/frames/${pan.image}?sid=${sid}">`;
-                    grid.appendChild(p);
-                    (pan.bubbles||[]).forEach(b => {
-                        const bub = document.createElement('div'); bub.className='speech-bubble';
-                        bub.innerText = b.dialog || "...";
-                        bub.style.left = (b.bubble_offset_x||50)+'px';
-                        bub.style.top = (b.bubble_offset_y||20)+'px';
-                        p.appendChild(bub);
-                    });
-                });
-                div.appendChild(grid); con.appendChild(div);
+<div class="box" id="upload-box">
+    <h1>🎬 Video to Comic</h1>
+    <input type="file" id="vid" accept="video/*"><br>
+    <button onclick="start()">Generate Comic</button>
+    <div id="status" style="margin-top:15px;color:#666;"></div>
+</div>
+<div id="result"></div>
+<script>
+let sid = Math.random().toString(36).substring(7);
+async function start() {
+    let f = document.getElementById('vid').files[0];
+    if(!f) return alert("Please select a video");
+    document.getElementById('status').innerText = "Uploading...";
+    let fd = new FormData(); fd.append('file', f);
+    let r = await fetch(`/uploader?sid=${sid}`, {method:'POST', body:fd});
+    if(r.ok) {
+        document.getElementById('status').innerText = "Queued for GPU...";
+        setInterval(check, 2000);
+    }
+}
+async function check() {
+    let r = await fetch(`/status?sid=${sid}`);
+    let d = await r.json();
+    document.getElementById('status').innerText = d.message;
+    if(d.progress >= 100) {
+        document.getElementById('upload-box').style.display = 'none';
+        loadPages();
+    }
+}
+async function loadPages() {
+    let r = await fetch(`/output/pages.json?sid=${sid}`);
+    let pages = await r.json();
+    let res = document.getElementById('result');
+    pages.forEach((pg, i) => {
+        let div = document.createElement('div'); div.className='comic-page';
+        div.innerHTML = `<h3>Page ${i+1}</h3>`;
+        pg.panels.forEach(pan => {
+            let pDiv = document.createElement('div'); pDiv.className='panel';
+            pDiv.innerHTML = `<img src="/frames/${pan.image}?sid=${sid}">`;
+            (pan.bubbles||[]).forEach(b => {
+                let bub = document.createElement('div'); bub.className='bubble';
+                bub.innerText = b.dialog;
+                bub.style.left = (b.bubble_offset_x||10)+'px';
+                bub.style.top = (b.bubble_offset_y||10)+'px';
+                pDiv.appendChild(bub);
             });
-        }
-        async function saveComic() { alert("Save not implemented in simple view"); }
-    </script>
+            div.appendChild(pDiv);
+        });
+        res.appendChild(div);
+    });
+}
+</script>
 </body>
 </html>
 '''
 
 @app.route('/')
-def index():
-    return INDEX_HTML
+def home(): return INDEX_HTML
 
 @app.route('/uploader', methods=['POST'])
 def upload():
     sid = request.args.get('sid')
-    if 'file' not in request.files: return "No file", 400
     f = request.files['file']
     gen = EnhancedComicGenerator(sid)
-    gen.cleanup_previous_run()
+    gen.cleanup()
     f.save(gen.video_path)
-    gen.update_status("Starting...", 5)
-    threading.Thread(target=gen.run_process).start()
+    gen.write_status("Starting...", 0)
+    threading.Thread(target=gen.run).start()
     return jsonify({'success': True})
 
 @app.route('/status')
-def get_status():
+def status():
     sid = request.args.get('sid')
     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..."})
+    return jsonify({'progress': 0, 'message': 'Waiting...'})
 
 @app.route('/output/<path:filename>')
-def get_output(filename):
+def files_out(filename):
     sid = request.args.get('sid')
     return send_from_directory(os.path.join(BASE_USER_DIR, sid, 'output'), filename)
 
 @app.route('/frames/<path:filename>')
-def get_frame(filename):
+def files_frames(filename):
     sid = request.args.get('sid')
     return send_from_directory(os.path.join(BASE_USER_DIR, sid, 'frames'), filename)
 
+@app.route('/regenerate_frame', methods=['POST'])
+def regen_frame():
+    d = request.json
+    sid = request.args.get('sid')
+    gen = EnhancedComicGenerator(sid)
+    res = regen_frame_gpu(gen.video_path, gen.frames_dir, gen.metadata_path, d['filename'], d['direction'])
+    return jsonify({'success': res is not None})
+
 if __name__ == '__main__':
-    port = int(os.getenv("PORT", 7860))
-    print(f"🚀 Starting on port {port}")
-    app.run(host='0.0.0.0', port=port)
+    # Force detection of gpu_warmup
+    try: gpu_warmup() 
+    except: pass
+    app.run(host='0.0.0.0', port=7860)