Update app_enhanced.py

app_enhanced.py

@@ -13,7 +13,6 @@ from flask import Flask, jsonify, request, send_from_directory, send_file
 # ======================================================
 # 🚀 ZEROGPU DETECTION FUNCTION
 # ======================================================
-# This function exists solely to be detected by Hugging Face
 @spaces.GPU
 def gpu_warmup():
     import torch
@@ -80,8 +79,8 @@ def generate_comic_gpu(video_path, user_dir, frames_dir, metadata_path):
     cap = cv2.VideoCapture(video_path)
     count = 0
     
-    # Limit to 12 frames for speed/demo purposes on ZeroGPU, increase if needed
-    for i, moment in enumerate(key_moments[:24]): 
+    # Process frames
+    for i, moment in enumerate(key_moments[:32]): # Limit frames for stability
         mid_time = (moment['start'] + moment['end']) / 2
         if mid_time > duration: continue
         
@@ -105,12 +104,18 @@ def generate_comic_gpu(video_path, user_dir, frames_dir, metadata_path):
     qe = QualityColorEnhancer()
     frame_files = sorted([f for f in os.listdir(frames_dir) if f.endswith('.png')])
     
+    print(f"🎨 Enhancing {len(frame_files)} frames...")
     for f in frame_files:
         p = os.path.join(frames_dir, f)
-        se.enhance_single(p)
-        qe.enhance_single(p)
+        # FIX: Pass 'p' as both input and output path to overwrite
+        try: se.enhance_single(p, p)
+        except Exception as e: print(f"SimpleEnhance error on {f}: {e}")
+            
+        try: qe.enhance_single(p, p)
+        except Exception as e: print(f"QualityEnhance error on {f}: {e}")
         
     # 5. Bubbles
+    print("💬 Placing bubbles...")
     bubbles_list = []
     for f in frame_files:
         p = os.path.join(frames_dir, f)
@@ -124,6 +129,7 @@ def generate_comic_gpu(video_path, user_dir, frames_dir, metadata_path):
             bubbles_list.append(bubble(dialog=dialogue))
             
     # 6. Layout
+    print("📄 Generating layout...")
     try:
         from backend.fixed_12_pages_800x1080 import generate_12_pages_800x1080
         pages = generate_12_pages_800x1080(frame_files, bubbles_list)
@@ -148,6 +154,8 @@ def generate_comic_gpu(video_path, user_dir, frames_dir, metadata_path):
 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 None
     with open(metadata_path, 'r') as f: meta = json.load(f)
     if fname not in meta: return None
@@ -165,9 +173,9 @@ def regen_frame_gpu(video_path, frames_dir, metadata_path, fname, direction):
     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)
+        # FIX: Pass output path
+        try: SimpleColorEnhancer().enhance_single(path, path)
+        except: pass
         
         if isinstance(meta[fname], dict): meta[fname]['time'] = new_t
         else: meta[fname] = new_t