Update app_enhanced.py

app_enhanced.py

@@ -39,6 +39,22 @@ SAVED_COMICS_DIR = os.path.join(BASE_STORAGE_PATH, "saved_comics")
 os.makedirs(BASE_USER_DIR, exist_ok=True)
 os.makedirs(SAVED_COMICS_DIR, exist_ok=True)
 
+# ======================================================
+# 🔧 APP CONFIG
+# ======================================================
+logging.basicConfig(level=logging.INFO)
+app = Flask(__name__)
+
+# ALLOW LARGE UPLOADS (500MB)
+app.config['MAX_CONTENT_LENGTH'] = 500 * 1024 * 1024  
+
+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
+
 # ======================================================
 # 🧱 DATA CLASSES
 # ======================================================
@@ -61,30 +77,17 @@ class Page:
         self.bubbles = bubbles
 
 # ======================================================
-# 🔧 APP CONFIG
-# ======================================================
-logging.basicConfig(level=logging.INFO)
-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
-
-# ======================================================
-# 🧠 GPU FUNCTIONS (OPTIMIZED FOR SPEED)
+# 🧠 GPU GENERATION
 # ======================================================
 @spaces.GPU(duration=120)
 def generate_comic_gpu(video_path, user_dir, frames_dir, metadata_path, target_pages):
-    print(f"🚀 Fast Generation Started: {video_path}")
+    print(f"🚀 Generation Started for {video_path}...")
     
     import cv2
     import srt
     import numpy as np
     
-    # 1. Video Setup
+    # 1. Video Prep
     cap = cv2.VideoCapture(video_path)
     if not cap.isOpened(): raise Exception("Cannot open video")
     fps = cap.get(cv2.CAP_PROP_FPS) or 25
@@ -92,10 +95,9 @@ def generate_comic_gpu(video_path, user_dir, frames_dir, metadata_path, target_p
     duration = total_frames / fps
     cap.release()
 
-    # 2. Subtitle Processing
+    # 2. Subtitles (Dummy if missing)
     user_srt = os.path.join(user_dir, 'subs.srt')
     if not os.path.exists(user_srt):
-        # Create dummy SRT if missing to avoid crashes
         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:
@@ -105,7 +107,7 @@ def generate_comic_gpu(video_path, user_dir, frames_dir, metadata_path, target_p
     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]
 
-    # 3. Time Selection
+    # 3. Calculate Intervals
     if target_pages <= 0: target_pages = 1
     panels_per_page = 4
     total_panels_needed = target_pages * panels_per_page
@@ -120,7 +122,7 @@ def generate_comic_gpu(video_path, user_dir, frames_dir, metadata_path, target_p
         indices = np.linspace(0, len(raw_moments) - 1, total_panels_needed, dtype=int)
         selected_moments = [raw_moments[i] for i in indices]
         
-    # 4. Fast Extraction Loop
+    # 4. Extract Frames (1280x720 16:9)
     frame_metadata = {}
     cap = cv2.VideoCapture(video_path)
     count = 0
@@ -133,10 +135,7 @@ def generate_comic_gpu(video_path, user_dir, frames_dir, metadata_path, target_p
         ret, frame = cap.read()
         
         if ret:
-            # ------------------------------------------------
-            # ⚡ OPTIMIZATION: 16:9 RESIZE ONLY (NO AI)
-            # 1280x720 preserves the "Wide Space"
-            # ------------------------------------------------
+            # FORCE LANDSCAPE 16:9 for Wide View
             frame = cv2.resize(frame, (1280, 720))
             
             fname = f"frame_{count:04d}.png"
@@ -150,18 +149,16 @@ def generate_comic_gpu(video_path, user_dir, frames_dir, metadata_path, target_p
     cap.release()
     with open(metadata_path, 'w') as f: json.dump(frame_metadata, f, indent=2)
         
-    # 5. Fast Bubble Creation (Skipped Face Detect for Speed)
+    # 5. Bubbles
     bubbles_list = []
     for f in frame_files_ordered:
         dialogue = frame_metadata.get(f, {}).get('dialogue', '')
         b_type = 'speech'
         if '(' in dialogue: b_type = 'narration'
         elif '!' in dialogue: b_type = 'reaction'
-        
-        # Default to Center (50, 50) to save processing time
         bubbles_list.append(bubble(dialog=dialogue, bubble_offset_x=50, bubble_offset_y=50, type=b_type))
             
-    # 6. Pagination
+    # 6. Pages
     pages = []
     for i in range(target_pages):
         start_idx = i * 4
@@ -169,7 +166,7 @@ def generate_comic_gpu(video_path, user_dir, frames_dir, metadata_path, target_p
         p_frames = frame_files_ordered[start_idx:end_idx]
         p_bubbles = bubbles_list[start_idx:end_idx]
         
-        # Pad with empty frames if needed
+        # Fill empty
         while len(p_frames) < 4:
              fname = f"empty_{i}_{len(p_frames)}.png"
              img = np.zeros((720, 1280, 3), dtype=np.uint8); img[:] = (40,40,40)
@@ -181,7 +178,6 @@ def generate_comic_gpu(video_path, user_dir, frames_dir, metadata_path, target_p
             pg_panels = [panel(image=f) for f in p_frames]
             pages.append(Page(panels=pg_panels, bubbles=p_bubbles))
 
-    # 7. Convert to Dict
     result = []
     for pg in pages:
         p_data = [p if isinstance(p, dict) else p.__dict__ for p in pg.panels]
@@ -202,7 +198,7 @@ def regen_frame_gpu(video_path, frames_dir, metadata_path, fname, direction):
     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) # 1 Frame jump
+    offset = (1.0/fps) * (1 if direction == 'forward' else -1)
     new_t = max(0, t + offset)
 
     cap.set(cv2.CAP_PROP_POS_MSEC, new_t * 1000)
@@ -220,30 +216,6 @@ def regen_frame_gpu(video_path, frames_dir, metadata_path, fname, direction):
         return {"success": True, "message": f"Time: {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
-    
-    cap = cv2.VideoCapture(video_path)
-    cap.set(cv2.CAP_PROP_POS_MSEC, float(ts) * 1000)
-    ret, frame = cap.read()
-    cap.release()
-
-    if ret:
-        frame = cv2.resize(frame, (1280, 720)) # Keep 16:9
-        p = os.path.join(frames_dir, fname)
-        cv2.imwrite(p, frame)
-        
-        if os.path.exists(metadata_path):
-            with open(metadata_path, 'r') as f: meta = json.load(f)
-            if fname in meta:
-                if isinstance(meta[fname], dict): meta[fname]['time'] = float(ts)
-                else: meta[fname] = float(ts)
-                with open(metadata_path, 'w') as f: json.dump(meta, f, indent=2)
-        return {"success": True, "message": f"Jumped to {ts}s"}
-    return {"success": False, "message": "Invalid timestamp"}
-
 # ======================================================
 # 💻 BACKEND CLASS
 # ======================================================
@@ -266,7 +238,7 @@ class EnhancedComicGenerator:
 
     def run(self, target_pages):
         try:
-            self.write_status("Waiting for GPU...", 5)
+            self.write_status("Processing...", 10)
             data = generate_comic_gpu(self.video_path, self.user_dir, self.frames_dir, self.metadata_path, int(target_pages))
             with open(os.path.join(self.output_dir, 'pages.json'), 'w') as f:
                 json.dump(data, f, indent=2)
@@ -280,10 +252,10 @@ class EnhancedComicGenerator:
             json.dump({'message': msg, 'progress': prog}, f)
 
 # ======================================================
-# 🌐 ROUTES & FULL UI
+# 🌐 FRONTEND
 # ======================================================
 INDEX_HTML = '''
-<!DOCTYPE html><html lang="en"> <head> <meta charset="UTF-8"> <meta name="viewport" content="width=device-width, initial-scale=1.0"> <title>Fast Comic Gen (16:9)</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> * { box-sizing: border-box; } body { background-color: #222; font-family: 'Lato', sans-serif; color: #eee; margin: 0; min-height: 100vh; }
+<!DOCTYPE html><html lang="en"> <head> <meta charset="UTF-8"> <meta name="viewport" content="width=device-width, initial-scale=1.0"> <title>Comic Generator (16:9)</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> * { box-sizing: border-box; } body { background-color: #222; font-family: 'Lato', sans-serif; color: #eee; margin: 0; min-height: 100vh; }
     
     #upload-container { display: flex; flex-direction:column; justify-content: center; align-items: center; min-height: 100vh; width: 100%; padding: 20px; }
     .upload-box { max-width: 500px; width: 100%; padding: 40px; background: #333; border-radius: 12px; box-shadow: 0 8px 30px rgba(0,0,0,0.5); text-align: center; }
@@ -399,9 +371,7 @@ INDEX_HTML = '''
     .reset-btn { background: #e74c3c; color: white; }
     .secondary-btn { background: #f39c12; color: white; }
     
-    .tip {
-        text-align:center; padding:10px; background:#e74c3c; color:white; font-weight:bold; margin-bottom:20px; border-radius:5px;
-    }
+    .tip { text-align:center; padding:10px; background:#e74c3c; color:white; font-weight:bold; margin-bottom:20px; border-radius:5px; }
 </style>
 </head> <body> 
 
@@ -469,12 +439,6 @@ INDEX_HTML = '''
     let interval, selectedBubble = null, selectedPanel = null;
     let dragType = null, activeObj = null, dragStart = {x:0, y:0};
     
-    // STATE MANAGEMENT
-    function getCurrentState() {
-        // Simple state grabber for saves (not fully implemented in this fast version, but structure is here)
-        return {}; 
-    }
-    
     async function upload() {
         const f = document.getElementById('file-upload').files[0];
         const pCount = document.getElementById('page-count').value;
@@ -489,8 +453,12 @@ INDEX_HTML = '''
         fd.append('target_pages', pCount);
         
         const r = await fetch(`/uploader?sid=${sid}`, {method:'POST', body:fd});
-        if(r.ok) interval = setInterval(checkStatus, 1500); // Faster polling
-        else { alert("Upload failed"); location.reload(); }
+        if(r.ok) interval = setInterval(checkStatus, 1500); 
+        else { 
+            const d = await r.json();
+            alert("Upload failed: " + (d.message || "Server Error")); 
+            location.reload(); 
+        }
     }
     
     async function checkStatus() {
@@ -674,15 +642,24 @@ def index():
 
 @app.route('/uploader', methods=['POST'])
 def upload():
-    sid = request.args.get('sid')
+    # Robust checks
+    sid = request.args.get('sid') or request.form.get('sid')
     if not sid: return jsonify({'success': False, 'message': 'Missing session ID'}), 400
-    if 'file' not in request.files: return jsonify({'success': False, 'message': 'No file'}), 400
+    
+    file = None
+    if 'file' in request.files:
+        file = request.files['file']
+    
+    if not file or file.filename == '':
+         # Debugging aid
+         print(f"Files keys: {list(request.files.keys())}")
+         return jsonify({'success': False, 'message': 'No file uploaded'}), 400
     
     target_pages = request.form.get('target_pages', 4)
-    f = request.files['file']
+    
     gen = EnhancedComicGenerator(sid)
     gen.cleanup()
-    f.save(gen.video_path)
+    file.save(gen.video_path)
     gen.write_status("Starting...", 5)
     
     threading.Thread(target=gen.run, args=(target_pages,)).start()