Update app_enhanced.py

app_enhanced.py

@@ -9,7 +9,6 @@ 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,592 +22,264 @@ def gpu_warmup():
     print(f"✅ ZeroGPU Warmup: CUDA Available: {torch.cuda.is_available()}")
     return True
 
-# ======================================================
-# 💾 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")
-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)
-
-# ======================================================
-# 🧱 DATA CLASSES
-# ======================================================
-def bubble(dialog="", bubble_offset_x=-1, bubble_offset_y=-1, lip_x=-1, lip_y=-1, emotion='normal', type='speech'):
-    return {
-        '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),
-        'emotion': emotion,
-        'type': type,
-        'tail_pos': '50%',
-        'classes': f'speech-bubble {type} tail-bottom'
-    }
-
-def panel(image=""):
-    return {'image': image}
-
-class Page:
-    def __init__(self, panels, bubbles):
-        self.panels = panels
-        self.bubbles = bubbles
-
-# ======================================================
-# 🔧 APP CONFIG
-# ======================================================
-logging.basicConfig(level=logging.INFO)
-logger = logging.getLogger(__name__)
-
-app = Flask(__name__)
-
-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
-
 # ======================================================
 # 🧠 GLOBAL GPU FUNCTIONS
 # ======================================================
 @spaces.GPU(duration=300)
-def generate_comic_gpu(video_path, user_dir, frames_dir, metadata_path, target_pages, panels_per_page_req):
-    print(f"🚀 GPU Task Started: {video_path} | Pages: {target_pages} | Panels/Page: {panels_per_page_req}")
-    
+def generate_comic_gpu(video_path, user_dir, frames_dir, metadata_path, target_pages):
     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
-    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
-
+    
     cap = cv2.VideoCapture(video_path)
     if not cap.isOpened(): raise Exception("Cannot open video")
     fps = cap.get(cv2.CAP_PROP_FPS) or 25
-    total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
-    duration = total_frames / fps
+    duration = int(cap.get(cv2.CAP_PROP_FRAME_COUNT)) / fps
     cap.release()
 
-    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:
-        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:
-        try: all_subs = list(srt.parse(f.read()))
-        except: all_subs = []
-
-    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 = int(panels_per_page_req)
-    if panels_per_page <= 0: panels_per_page = 2
+    # We force 2 panels per page for the vertical slant 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_needed = target_pages * panels_per_page
-
-    selected_moments = []
-    if not raw_moments:
-        times = np.linspace(1, duration-1, total_panels_needed)
-        for t in times: selected_moments.append({'text': '', 'start': t, 'end': t+1})
-    elif len(raw_moments) <= total_panels_needed:
-        selected_moments = raw_moments
-    else:
-        indices = np.linspace(0, len(raw_moments) - 1, total_panels_needed, dtype=int)
-        selected_moments = [raw_moments[i] for i in indices]
-        
     frame_metadata = {}
-    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
-        cap.set(cv2.CAP_PROP_POS_FRAMES, int(mid * fps))
+    cap = cv2.VideoCapture(video_path)
+    for i, t in enumerate(times):
+        cap.set(cv2.CAP_PROP_POS_MSEC, t * 1000)
         ret, frame = cap.read()
         if ret:
-            fname = f"frame_{count:04d}.png"
+            fname = f"frame_{i:04d}.png"
             p = os.path.join(frames_dir, fname)
             cv2.imwrite(p, frame)
-            os.sync() 
-            frame_metadata[fname] = {'dialogue': moment['text'], 'time': mid}
+            frame_metadata[fname] = {'dialogue': f"Moment {i+1}", 'time': t}
             frame_files_ordered.append(fname)
-            count += 1
     cap.release()
 
-    with open(metadata_path, 'w') as f: json.dump(frame_metadata, f, indent=2)
-        
-    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)
-        except: pass
-        try: qe.enhance_single(p, p)
-        except: pass
-        
-    bubbles_list = []
-    for f in frame_files_ordered:
-        p = os.path.join(frames_dir, f)
-        dialogue = frame_metadata.get(f, {}).get('dialogue', '')
-        
-        b_type = 'speech'
-        if '(' in dialogue and ')' in dialogue: b_type = 'narration'
-        elif '!' in dialogue and dialogue.isupper(): b_type = 'reaction'
-        elif '?' in dialogue: b_type = 'speech'
-        
-        bubbles_list.append(bubble(dialog=dialogue, bubble_offset_x=-1, bubble_offset_y=-1, type=b_type))
+    with open(metadata_path, 'w') as f: json.dump(frame_metadata, f)
             
     pages = []
-    for i in range(target_pages):
-        start_idx = i * panels_per_page
-        end_idx = start_idx + panels_per_page
-        p_frames = frame_files_ordered[start_idx:end_idx]
-        p_bubbles = bubbles_list[start_idx:end_idx]
-        if p_frames:
-            pg_panels = [panel(image=f) for f in p_frames]
-            pages.append(Page(panels=pg_panels, bubbles=p_bubbles))
-
-    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
+    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
 
 @spaces.GPU
 def regen_frame_gpu(video_path, frames_dir, metadata_path, fname, direction):
     import cv2
     import json
-    from backend.simple_color_enhancer import SimpleColorEnhancer
-    
-    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]
+    t = meta[fname]['time']
     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)
-
+    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:
-        p = os.path.join(frames_dir, fname)
-        cv2.imwrite(p, frame)
-        os.sync()
-        try: SimpleColorEnhancer().enhance_single(p, p)
-        except: pass
-        
-        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"Adjusted to {new_t:.2f}s"}
-    return {"success": False, "message": "End of video"}
-
-@spaces.GPU
-def get_frame_at_ts_gpu(video_path, frames_dir, metadata_path, fname, ts):
-    import cv2
-    import json
-    from backend.simple_color_enhancer import SimpleColorEnhancer
-    
-    cap = cv2.VideoCapture(video_path)
-    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)
-        os.sync()
-        try: SimpleColorEnhancer().enhance_single(p, p)
-        except: pass
-        
-        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"}
+        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}
 
 # ======================================================
-# 💻 BACKEND CLASS
+# 💾 STORAGE SETUP
 # ======================================================
-class EnhancedComicGenerator:
+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:
     def __init__(self, sid):
         self.sid = sid
-        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')
+        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 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 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)
 
-    def run(self, target_pages, panels_per_page):
+    def run(self, target_pages):
         try:
-            self.write_status("Waiting for GPU...", 5)
-            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("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("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)
-
 # ======================================================
-# 🌐 ROUTES & FULL UI
+# 🌐 UI & TEMPLATE (Vertical Tilt Feature)
 # ======================================================
 INDEX_HTML = '''
-<!DOCTYPE html><html lang="en"> <head> <meta charset="UTF-8"> <meta name="viewport" content="width=device-width, initial-scale=1.0"> <title>🎬 Vertical 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; }
-    
-    .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; }
+<!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; 
-        box-shadow: 0 4px 10px rgba(0,0,0,0.1); 
-        position: relative; 
-        overflow: hidden; 
-        border: 2px solid #000; 
-        padding: 10px; 
+        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; }
     
-    /* === VERTICAL TILT LAYOUT (Split Left/Right) === */
-    .comic-grid { width: 100%; height: 100%; position: relative; background: #000; display: block; }
-    
-    .panel { position: absolute; overflow: hidden; background: #000; cursor: pointer; border: 2px solid #000; }
-    .comic-grid.layout-custom-slant .panel { border: none; background: transparent; width: 100%; height: 100%; }
-    
-    /* Panel 1 (Left) */
-    .comic-grid.layout-custom-slant .panel:nth-child(1) { 
-        top: 0; left: 0;
-        z-index:2; 
+    /* 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 (Right) */
-    .comic-grid.layout-custom-slant .panel:nth-child(2) { 
-        top: 0; left: 0;
-        z-index:1; 
+    /* 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.selected { z-index: 20; border: 3px solid #2196F3; }
+    .panel img { width: 100%; height: 100%; object-fit: cover; pointer-events: none; }
     
-    /* DRAG POINTS FOR VERTICAL TILT (Top and Bottom edges) */
+    /* DRAGGABLE HANDLES ON THE EDGES */
     .split-handle {
-        position: absolute; width: 26px; height: 26px; 
-        background: #2196F3; border: 3px solid white; border-radius: 50%;
-        cursor: ew-resize; z-index: 1000; box-shadow: 0 2px 5px rgba(0,0,0,0.4); 
-    }
-    .split-handle:hover { transform: scale(1.2); }
-    .split-handle.top { top: -13px; left: var(--split-t, 45%); transform: translateX(-50%); }
-    .split-handle.bottom { bottom: -13px; left: var(--split-b, 55%); transform: translateX(-50%); }
-    
-    .panel img { 
-        width: 100%; height: 100%; 
-        object-fit: cover;
-        transition: transform 0.1s ease-out; 
-        transform-origin: center center; 
-        pointer-events: auto; 
+        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);
     }
-    
-    .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; 
-        line-height: 1.2;
-        --tail-pos: 50%;
-        left: 50%; top: 50%; transform: translate(-50%, -50%);
-    }
-    
-    .bubble-text { 
-        padding: 0.5em; word-wrap: break-word; white-space: pre-wrap; position: relative; 
-        z-index: 5; pointer-events: none; user-select: none; width: 100%; height: 100%;
-        overflow: hidden; display: flex; align-items: center; justify-content: center; border-radius: inherit;
-    }
-    
-    .speech-bubble.selected { outline: 2px dashed #4CAF50; z-index: 100; }
-    
-    .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: 0;
-        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))); }
-    
+    /* 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; 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; }
-    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; }
-    .action-btn { background: #4CAF50; color: white; }
-    .reset-btn { background: #e74c3c; color: white; }
-    .export-btn { background: #2196F3; color: white; }
-    
-</style>
-</head> <body> <div id="upload-container"> <div class="upload-box"> <h1>🎬 Vertical Tilt 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="button-grid" style="display:grid; grid-template-columns: 1fr 1fr; gap:10px;">
-            <div class="page-input-group" style="margin:5px 0;">
-                <label>📚 Panels / Page:</label>
-                <input type="number" id="panels-per-page" value="2" min="1" max="4">
-            </div>
-            <div class="page-input-group" style="margin:5px 0;">
-                <label>📄 Total Pages:</label>
-                <input type="number" id="page-count" value="4" min="1" max="15">
-            </div>
-        </div>
-        
-        <button class="submit-btn" onclick="upload()">🚀 Generate Comic</button>
-        
-        <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>
+    .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>
+
+<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>
 </div>
-<div id="editor-container">
-    <div class="comic-wrapper" id="comic-container"></div>
+
+<div id="editor-container" style="display:none">
+    <div id="comic-list"></div>
     <div class="edit-controls">
-        <h4>✏️ Vertical Editor</h4>
-        
-        <div class="control-group">
-            <button onclick="addBubble()" class="action-btn">Add Bubble</button>
-            <button onclick="deleteBubble()" class="reset-btn">Delete Bubble</button>
-        </div>
-        
-        <div class="control-group">
-            <label>🖼️ Panel Tools:</label>
-            <div style="display:grid; grid-template-columns: 1fr 1fr; gap:5px;">
-                <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>
-        
-        <div class="control-group">
-            <button onclick="exportComic()" class="export-btn">📥 Export as PNG</button>
-            <button onclick="location.reload()" class="reset-btn" style="margin-top:10px;">🏠 Home</button>
-        </div>
+        <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>
     </div>
 </div>
 
 <script>
     let sid = 'S' + Math.floor(Math.random()*1000000);
-    let selectedBubble = null, selectedPanel = null;
-    let isDragging = false, isPanning = false, isDraggingSplit = false;
-    let startX, startY, initX, initY, panStartX, panStartY, panStartTx, panStartTy;
-    let selectedSplitHandle = null;
-    
+    let isDragging = false, currentHandle = null;
+
     async function upload() {
-        const f = document.getElementById('file-upload').files[0];
-        const pCount = document.getElementById('page-count').value;
-        const panelCount = document.getElementById('panels-per-page').value;
-        if(!f) return alert("Select a video");
-        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); fd.append('panels_per_page', panelCount);
-        const r = await fetch(`/uploader?sid=${sid}`, {method:'POST', body:fd});
-        if(r.ok) interval = setInterval(checkStatus, 2000);
-    }
-    
-    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(); }
-        } catch(e) {}
-    }
-    
-    function loadNewComic() {
-        fetch(`/output/pages.json?sid=${sid}`).then(r=>r.json()).then(data => {
-            const cleanData = data.map(p => ({
-                layout: 'layout-custom-slant',
-                splitT: '45%', splitB: '55%',
-                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: '50%', top: '50%', type: 'speech', classes: 'speech-bubble speech tail-bottom'
-                    }] : []
-                }))
-            }));
-            renderFromState(cleanData);
-        });
+        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});
+        
+        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);
     }
-    
-    function renderFromState(pagesData) {
-        const con = document.getElementById('comic-container'); con.innerHTML = '';
-        pagesData.forEach((page, pi) => {
-            const wrap = document.createElement('div'); wrap.className = 'page-wrapper';
-            wrap.innerHTML = `<h2 class="page-title">Page ${pi + 1}</h2>`;
-            const div = document.createElement('div'); div.className = 'comic-page';
-            const grid = document.createElement('div'); grid.className = 'comic-grid layout-custom-slant';
-            
-            // Apply Vertical Split variables
-            grid.style.setProperty('--split-t', page.splitT || '45%');
-            grid.style.setProperty('--split-b', page.splitB || '55%');
-            
-            // Vertical Handles
+
+    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';
-            hTop.onmousedown = (e) => { e.preventDefault(); isDraggingSplit = true; selectedSplitHandle = { grid, side: 'top' }; };
             const hBot = document.createElement('div'); hBot.className = 'split-handle bottom';
-            hBot.onmousedown = (e) => { e.preventDefault(); isDraggingSplit = true; selectedSplitHandle = { grid, side: 'bottom' }; };
-            
-            grid.appendChild(hTop); grid.appendChild(hBot);
             
-            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;
-                pDiv.appendChild(img);
-                (pan.bubbles || []).forEach(bData => pDiv.appendChild(createBubbleHTML(bData)));
-                grid.appendChild(pDiv);
+            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);
             });
-            div.appendChild(grid); wrap.appendChild(div); con.appendChild(wrap);
+
+            grid.appendChild(hTop); grid.appendChild(hBot);
+            pgDiv.appendChild(grid);
+            list.appendChild(pgDiv);
         });
+        document.getElementById('upload-container').style.display='none';
+        document.getElementById('editor-container').style.display='block';
     }
-    
-    function createBubbleHTML(data) {
-        const b = document.createElement('div'); b.className = data.classes || 'speech-bubble speech tail-bottom';
-        b.style.left = data.left; b.style.top = data.top;
-        const txt = document.createElement('span'); txt.className = 'bubble-text'; txt.textContent = data.text;
-        b.appendChild(txt);
-        b.onmousedown = (e) => { e.stopPropagation(); selectBubble(b); isDragging = true; startX = e.clientX; startY = e.clientY; initX = b.offsetLeft; initY = b.offsetTop; };
-        return b;
-    }
-    
-    function selectBubble(el) { if(selectedBubble) selectedBubble.classList.remove('selected'); selectedBubble = el; el.classList.add('selected'); }
-    function selectPanel(el) { if(selectedPanel) selectedPanel.classList.remove('selected'); selectedPanel = el; el.classList.add('selected'); }
 
-    document.addEventListener('mousemove', (e) => {
-        if(isDragging && selectedBubble) {
-            selectedBubble.style.left = (initX + e.clientX - startX) + 'px';
-            selectedBubble.style.top = (initY + e.clientY - startY) + 'px';
+    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();
         }
-        if(isDraggingSplit && selectedSplitHandle) {
-            const rect = selectedSplitHandle.grid.getBoundingClientRect();
-            let xPos = e.clientX - rect.left;
-            let percent = (xPos / rect.width) * 100;
-            percent = Math.max(10, Math.min(90, percent));
-            
-            if(selectedSplitHandle.side === 'top') {
-                selectedSplitHandle.grid.style.setProperty('--split-t', percent + '%');
-            } else {
-                selectedSplitHandle.grid.style.setProperty('--split-b', percent + '%');
-            }
-        }
-    });
-    
-    document.addEventListener('mouseup', () => { isDragging = false; isDraggingSplit = false; selectedSplitHandle = null; });
-    
-    function addBubble() { if(!selectedPanel) return; const b = createBubbleHTML({text:'Text', left:'50%', top:'50%'}); selectedPanel.appendChild(b); selectBubble(b); }
-    function deleteBubble() { if(selectedBubble) selectedBubble.remove(); }
-    
-    async function exportComic() {
-        const node = document.querySelector('.comic-page');
-        const dataUrl = await htmlToImage.toPng(node);
-        const link = document.createElement('a');
-        link.download = 'comic.png'; link.href = dataUrl; link.click();
     }
-</script>
-</body> </html> '''
-
-# ... (Include all backend Flask routes and Generation logic from your original code) ...
+</script> </body> </html>
+'''
 
 @app.route('/')
-def index():
-    return INDEX_HTML
+def index(): return INDEX_HTML
 
 @app.route('/uploader', methods=['POST'])
-def upload():
+def uploader():
     sid = request.args.get('sid')
-    target_pages = request.form.get('target_pages', 4)
-    panels_per_page = request.form.get('panels_per_page', 2)
     f = request.files['file']
-    gen = EnhancedComicGenerator(sid)
-    f.save(gen.video_path)
-    threading.Thread(target=gen.run, args=(target_pages, panels_per_page)).start()
-    return jsonify({'success': True})
+    gen = ComicBackend(sid)
+    f.save(gen.v_path)
+    threading.Thread(target=gen.run, args=(2,)).start()
+    return jsonify({'ok': True})
 
 @app.route('/status')
 def get_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..."})
+    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})
 
 @app.route('/output/<path:filename>')
 def get_output(filename):
@@ -621,6 +292,4 @@ def get_frame(filename):
     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)