Update app_enhanced.py

app_enhanced.py

@@ -8,10 +8,13 @@ import logging
 import string
 import random
 import shutil
+import cv2
+import numpy as np
+import srt
 from flask import Flask, jsonify, request, send_from_directory, send_file
 
 # ======================================================
-# 🚀 ZEROGPU DETECTION FUNCTION
+# 🚀 ZEROGPU DETECTION
 # ======================================================
 @spaces.GPU
 def gpu_warmup():
@@ -19,6 +22,27 @@ def gpu_warmup():
     print(f"✅ ZeroGPU Warmup: CUDA Available: {torch.cuda.is_available()}")
     return True
 
+# ======================================================
+# 🧱 ROBUST DATA CLASSES (Defined locally to prevent crashes)
+# ======================================================
+def bubble(dialog="", bubble_offset_x=50, bubble_offset_y=20, lip_x=-1, lip_y=-1, emotion='normal'):
+    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
+    }
+
+def panel(image=""): 
+    return {'image': image}
+
+class Page:
+    def __init__(self, panels, bubbles): 
+        self.panels = panels
+        self.bubbles = bubbles
+
 # ======================================================
 # 🔧 CONFIGURATION
 # ======================================================
@@ -49,7 +73,6 @@ def generate_comic_gpu(video_path, user_dir, frames_dir, metadata_path):
     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
-    from backend.class_def import bubble, panel, Page
     
     # 1. Analyze Video
     cap = cv2.VideoCapture(video_path)
@@ -79,8 +102,8 @@ def generate_comic_gpu(video_path, user_dir, frames_dir, metadata_path):
     cap = cv2.VideoCapture(video_path)
     count = 0
     
-    # Process frames
-    for i, moment in enumerate(key_moments[:32]): # Limit frames for stability
+    # Limit frames to prevent timeouts
+    for i, moment in enumerate(key_moments[:32]): 
         mid_time = (moment['start'] + moment['end']) / 2
         if mid_time > duration: continue
         
@@ -107,12 +130,11 @@ def generate_comic_gpu(video_path, user_dir, frames_dir, metadata_path):
     print(f"🎨 Enhancing {len(frame_files)} frames...")
     for f in frame_files:
         p = os.path.join(frames_dir, f)
-        # FIX: Pass 'p' as both input and output path to overwrite
+        # Pass path twice (Input, Output)
         try: se.enhance_single(p, p)
-        except Exception as e: print(f"SimpleEnhance error on {f}: {e}")
-            
+        except: pass
         try: qe.enhance_single(p, p)
-        except Exception as e: print(f"QualityEnhance error on {f}: {e}")
+        except: pass
         
     # 5. Bubbles
     print("💬 Placing bubbles...")
@@ -124,16 +146,29 @@ def generate_comic_gpu(video_path, user_dir, frames_dir, metadata_path):
             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:
+            # Use local bubble function which is safe
+            b = bubble(
+                dialog=dialogue, 
+                bubble_offset_x=bx, 
+                bubble_offset_y=by, 
+                lip_x=lip[0], 
+                lip_y=lip[1], 
+                emotion='normal'
+            )
+            bubbles_list.append(b)
+        except Exception as e:
+            print(f"Bubble placement error: {e}")
+            # Safe fallback
             bubbles_list.append(bubble(dialog=dialogue))
             
     # 6. Layout
     print("📄 Generating layout...")
     try:
         from backend.fixed_12_pages_800x1080 import generate_12_pages_800x1080
+        # We need to ensure the backend accepts our dicts, or we use fallback
         pages = generate_12_pages_800x1080(frame_files, bubbles_list)
-    except:
+    except Exception as e:
+        print(f"Layout engine error: {e}. Using simple layout.")
         pages = []
         for i in range((len(frame_files)+3)//4):
             start = i*4
@@ -144,8 +179,9 @@ def generate_comic_gpu(video_path, user_dir, frames_dir, metadata_path):
     # 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]
+        # Handle if backend returned objects or dicts
+        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
@@ -173,7 +209,6 @@ 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)
-        # FIX: Pass output path
         try: SimpleColorEnhancer().enhance_single(path, path)
         except: pass