Update app_enhanced.py

app_enhanced.py

@@ -1,4 +1,4 @@
-import spaces
+import spaces  # <--- CRITICAL: MUST BE THE FIRST IMPORT
 import os
 import time
 import threading
@@ -28,8 +28,10 @@ def gpu_warmup():
 # ======================================================
 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")
@@ -70,24 +72,32 @@ def generate_comic_gpu(video_path, user_dir, frames_dir, metadata_path, target_p
     from backend.subtitles.subs_real import get_real_subtitles
     
     cap = cv2.VideoCapture(video_path)
-    if not cap.isOpened(): raise Exception("Cannot open video")
+    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
     cap.release()
 
-    # Subtitles
+    # Subtitles logic with FIXED SYNTAX
     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)
-        elif not os.path.exists(user_srt): with open(user_srt, 'w') as f: f.write("1\n00:00:01,000 --> 00:00:04,000\n...\n")
+        if os.path.exists('test1.srt'):
+            shutil.move('test1.srt', user_srt)
+        elif not os.path.exists(user_srt):
+            with open(user_srt, 'w') as f:
+                f.write("1\n00:00:01,000 --> 00:00:04,000\n...\n")
     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, '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 = []
+        try:
+            all_subs = list(srt.parse(f.read()))
+        except:
+            all_subs = []
 
     valid_subs = [s for s in all_subs if s.content.strip()]
     if valid_subs:
@@ -102,7 +112,8 @@ def generate_comic_gpu(video_path, user_dir, frames_dir, metadata_path, target_p
     selected_moments = []
     if not raw_moments:
         times = np.linspace(1, max(1, duration-1), total_panels_needed)
-        for t in times: selected_moments.append({'text': '', 'start': t, 'end': t+1})
+        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:
@@ -120,7 +131,6 @@ def generate_comic_gpu(video_path, user_dir, frames_dir, metadata_path, target_p
         ret, frame = cap.read()
         if ret:
             # 🎯 EXTRACT FULL HD (1280x720) - NO CROP
-            # Frontend handles Zoom/Fit
             frame = cv2.resize(frame, (1280, 720))
             fname = f"frame_{count:04d}.png"
             p = os.path.join(frames_dir, fname)
@@ -130,7 +140,8 @@ def generate_comic_gpu(video_path, user_dir, frames_dir, metadata_path, target_p
             count += 1
     cap.release()
     
-    with open(metadata_path, 'w') as f: json.dump(frame_metadata, f, indent=2)
+    with open(metadata_path, 'w') as f:
+        json.dump(frame_metadata, f, indent=2)
         
     bubbles_list = []
     for f in frame_files_ordered:
@@ -149,13 +160,13 @@ def generate_comic_gpu(video_path, user_dir, frames_dir, metadata_path, target_p
         
         while len(p_frames) < 4:
              fname = f"empty_{i}_{len(p_frames)}.png"
-             img = np.zeros((720, 1280, 3), dtype=np.uint8); img[:] = (30,30,30)
+             img = np.zeros((720, 1280, 3), dtype=np.uint8)
+             img[:] = (30,30,30)
              cv2.imwrite(os.path.join(frames_dir, fname), img)
              p_frames.append(fname)
              p_bubbles.append(bubble(dialog="", type='speech'))
 
         if p_frames:
-            # Create Page Object
             pg_panels = [{'image': f} for f in p_frames]
             pages.append({'panels': pg_panels, 'bubbles': p_bubbles})
 
@@ -165,8 +176,11 @@ def generate_comic_gpu(video_path, user_dir, frames_dir, metadata_path, target_p
 def regen_frame_gpu(video_path, frames_dir, metadata_path, fname, direction):
     import cv2
     import json
-    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 not os.path.exists(metadata_path):
+        return {"success": False, "message": "No metadata"}
+    
+    with open(metadata_path, 'r') as f:
+        meta = json.load(f)
     
     t = meta[fname]['time'] if isinstance(meta[fname], dict) else meta[fname]
     cap = cv2.VideoCapture(video_path)
@@ -181,9 +195,12 @@ def regen_frame_gpu(video_path, frames_dir, metadata_path, fname, direction):
     if ret:
         frame = cv2.resize(frame, (1280, 720))
         cv2.imwrite(os.path.join(frames_dir, fname), frame)
-        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)
+        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"Time: {new_t:.2f}s"}
     return {"success": False}
 
@@ -202,8 +219,10 @@ class EnhancedComicGenerator:
         self.metadata_path = os.path.join(self.frames_dir, 'frame_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)
+        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)
 
@@ -240,19 +259,20 @@ INDEX_HTML = '''
     
     .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: #ccc; }
-    .page-input-group input { width: 100%; padding: 12px; border: 2px solid #555; background: #2c3e50; color: white; border-radius: 8px; }
+    .page-input-group input { width: 100%; padding: 12px; border: 2px solid #555; background: #2c3e50; color: white; border-radius: 8px; font-size: 16px; box-sizing: border-box; }
     
     .submit-btn { width: 100%; padding: 15px; background: #2980b9; color: white; border: none; border-radius: 8px; font-size: 18px; font-weight: bold; cursor: pointer; }
     .loader { width: 100px; height: 10px; background: #e67e22; margin: 20px auto; animation: load 1s infinite alternate; }
     @keyframes load { from { width: 20px; } to { width: 100px; } }
     
-    /* === SQUARE LAYOUT === */
+    /* === SQUARE COMIC LAYOUT (800x800) === */
     .comic-wrapper { max-width: 1000px; margin: 0 auto; display: flex; flex-direction: column; align-items: center; gap: 40px; }
     .page-wrapper { display: flex; flex-direction: column; align-items: center; }
     .page-title { text-align: center; color: #eee; margin-bottom: 10px; font-size: 20px; font-weight: bold; }
     
     .comic-page { 
-        width: 800px; height: 800px; 
+        width: 800px; 
+        height: 800px; 
         background: white; 
         box-shadow: 0 5px 30px rgba(0,0,0,0.6); 
         position: relative; overflow: hidden; 
@@ -292,7 +312,6 @@ INDEX_HTML = '''
     .bubble-text { padding: 0.8em; width: 100%; height: 100%; display: flex; align-items: center; justify-content: center; overflow: hidden; white-space: pre-wrap; pointer-events: none; border-radius: inherit; }
     .speech-bubble.selected { outline: 2px dashed #4CAF50; z-index: 100; }
     
-    /* Bubble Styles */
     .speech-bubble.speech { background: var(--bubble-fill, #fff); color: var(--bubble-text, #000); border: 2px solid #000; border-radius: 50%; width: 180px; height: 100px; }
     .speech-bubble.speech::after { content: ''; position: absolute; bottom: -12px; left: var(--tail-pos); border: 12px solid transparent; border-top-color: #000; border-bottom: 0; margin-left: -12px; }
     
@@ -316,6 +335,7 @@ INDEX_HTML = '''
     .color-grid { display: grid; grid-template-columns: 1fr 1fr; gap: 5px; }
     .action-btn { background: #27ae60; color: white; }
     .reset-btn { background: #c0392b; color: white; }
+    .secondary-btn { background: #f39c12; color: white; }
     .save-btn { background: #8e44ad; color: white; }
     
     .tip { text-align:center; padding:10px; background:#e74c3c; color:white; font-weight:bold; margin-bottom:20px; border-radius:5px; }