Update app_enhanced.py

app_enhanced.py

@@ -81,9 +81,18 @@ def generate_save_code(length=8):
 # ======================================================
 # 🧠 GLOBAL GPU FUNCTIONS
 # ======================================================
+def update_status_file(path, msg, progress):
+    """Helper to write status from within GPU function"""
+    try:
+        with open(path, 'w') as f:
+            json.dump({'message': msg, 'progress': progress}, f)
+    except:
+        pass
+
 @spaces.GPU(duration=300)
-def generate_comic_gpu(video_path, user_dir, frames_dir, metadata_path, target_pages):
+def generate_comic_gpu(video_path, user_dir, frames_dir, metadata_path, status_path, target_pages):
     print(f"🚀 GPU Task Started: {video_path} | Pages: {target_pages}")
+    update_status_file(status_path, "Initializing GPU Models...", 10)
     
     import cv2
     import srt
@@ -103,6 +112,8 @@ def generate_comic_gpu(video_path, user_dir, frames_dir, metadata_path, target_p
     cap.release()
 
     user_srt = os.path.join(user_dir, 'subs.srt')
+    update_status_file(status_path, "Extracting Audio & Subtitles...", 20)
+    
     try:
         get_real_subtitles(video_path)
         if os.path.exists('test1.srt'):
@@ -119,7 +130,7 @@ def generate_comic_gpu(video_path, user_dir, frames_dir, metadata_path, target_p
 
     if target_pages <= 0: target_pages = 1
     
-    # === UPDATED: 5 PANELS PER PAGE FOR NEW LAYOUT ===
+    # === 5 PANELS PER PAGE ===
     panels_per_page = 5 
     total_panels_needed = target_pages * panels_per_page
 
@@ -138,6 +149,8 @@ def generate_comic_gpu(video_path, user_dir, frames_dir, metadata_path, target_p
     count = 0
     frame_files_ordered = []
 
+    update_status_file(status_path, f"Extracting {total_panels_needed} Frames...", 30)
+
     for i, moment in enumerate(selected_moments):
         mid = (moment['start'] + moment['end']) / 2
         if mid > duration: mid = duration - 1
@@ -151,17 +164,24 @@ def generate_comic_gpu(video_path, user_dir, frames_dir, metadata_path, target_p
             frame_metadata[fname] = {'dialogue': moment['text'], 'time': mid}
             frame_files_ordered.append(fname)
             count += 1
+            # Update progress periodically
+            if count % 2 == 0:
+                prog = 30 + int((count / total_panels_needed) * 20)
+                update_status_file(status_path, f"Extracted Frame {count}/{total_panels_needed}", prog)
+                
     cap.release()
 
     with open(metadata_path, 'w') as f: json.dump(frame_metadata, f, indent=2)
-        
+    
+    update_status_file(status_path, "Auto-Cropping...", 55)
     try: black_bar_crop() 
     except: pass
 
     se = SimpleColorEnhancer()
     qe = QualityColorEnhancer()
 
-    for f in frame_files_ordered:
+    for idx, f in enumerate(frame_files_ordered):
+        update_status_file(status_path, f"Enhancing Image {idx+1}/{len(frame_files_ordered)}", 60 + int((idx / len(frame_files_ordered)) * 20))
         p = os.path.join(frames_dir, f)
         try: se.enhance_single(p, p)
         except: pass
@@ -169,6 +189,8 @@ def generate_comic_gpu(video_path, user_dir, frames_dir, metadata_path, target_p
         except: pass
         
     bubbles_list = []
+    update_status_file(status_path, "Placing Bubbles...", 85)
+    
     for f in frame_files_ordered:
         p = os.path.join(frames_dir, f)
         dialogue = frame_metadata.get(f, {}).get('dialogue', '')
@@ -287,8 +309,10 @@ class EnhancedComicGenerator:
 
     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))
+            self.write_status("Queued for GPU...", 5)
+            status_file = os.path.join(self.output_dir, 'status.json')
+            # Pass status_file path to GPU function for real-time updates
+            data = generate_comic_gpu(self.video_path, self.user_dir, self.frames_dir, self.metadata_path, status_file, int(target_pages))
             with open(os.path.join(self.output_dir, 'pages.json'), 'w') as f:
                 json.dump(data, f, indent=2)
             self.write_status("Complete!", 100)