RealComic

Configuration error

App Files Files Community

tester343 commited on Dec 3, 2025

Commit

b5ffd51

verified ·

1 Parent(s): f2379ad

Update app_enhanced.py

Browse files

Files changed (1) hide show

app_enhanced.py +54 -24

app_enhanced.py CHANGED Viewed

@@ -5,9 +5,16 @@ import uuid
 import shutil
 import json
 import traceback
 from concurrent.futures import ThreadPoolExecutor
 from flask import Flask, render_template, request, jsonify, send_from_directory, send_file
 # --- 1. CORE DEPENDENCY CHECKS ---
 try:
     import cv2
@@ -16,15 +23,12 @@ try:
     import srt
 except ImportError as e:
     print(f"❌ CRITICAL ERROR: Missing python library. {e}")
-    # Define dummies to allow app to start (will fail gracefully later)
     cv2 = None
     np = None
     Image = None
     srt = None
 # --- 2. BACKEND MODULE IMPORTS (WITH ROBUST FALLBACKS) ---
-# This section ensures the app loads even if specific backend files are missing.
 def dummy_black_bar_crop(): return 0, 0, None, None
 try:
@@ -51,7 +55,6 @@ try:
     from backend.class_def import bubble, panel, Page
     print("✅ Core class definitions loaded.")
 except Exception:
-    # Fallback definitions if backend/class_def.py is missing
     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': bubble_offset_x, 'bubble_offset_y': bubble_offset_y,
@@ -65,9 +68,9 @@ try:
     from backend.ai_enhanced_core import image_processor, comic_styler, face_detector, layout_optimizer
     from backend.ai_bubble_placement import ai_bubble_placer
     from backend.subtitles.subs_real import get_real_subtitles
     print("✅ Core utility modules loaded.")
 except Exception:
-    # Dummies for AI modules
     def get_real_subtitles(v): pass
     class DummyDetector:
         def detect_faces(self, p): return []
@@ -82,7 +85,7 @@ except Exception:
 app = Flask(__name__)
 BASE_USER_DIR = "userdata"
-# --- HTML INTERFACE (Upload + Editor + JS) ---
 INDEX_HTML = '''
 <!DOCTYPE html>
 <html lang="en">
@@ -90,10 +93,7 @@ INDEX_HTML = '''
     <meta charset="UTF-8">
     <meta name="viewport" content="width=device-width, initial-scale=1.0">
     <title>Movie to Comic Generator</title>
-    <!-- Export Library that supports CSS Masks/Gradients -->
     <script src="https://cdnjs.cloudflare.com/ajax/libs/html-to-image/1.11.11/html-to-image.min.js"></script>
-    <link rel="preconnect" href="https://fonts.googleapis.com">
-    <link rel="preconnect" href="https://fonts.gstatic.com" crossorigin>
     <link href="https://fonts.googleapis.com/css2?family=Bangers&family=Comic+Neue:wght@700&family=Lato&display=swap" rel="stylesheet">
     <style>
         /* GLOBAL STYLES */
@@ -570,8 +570,9 @@ INDEX_HTML = '''
             updateImageTransform(img);
         }
         function replacePanelImage() {
-            if(!currentlySelectedPanel) return alert("Select panel");
             const img = currentlySelectedPanel.querySelector('img');
             const inp = document.getElementById('image-uploader');
             inp.onchange = async (e) => {
@@ -649,18 +650,23 @@ class EnhancedComicGenerator:
                 json.dump({'message': message, 'progress': progress}, f)
         except: pass
-    def cleanup_generated(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)
-        os.makedirs(self.frames_dir, exist_ok=True)
-        os.makedirs(self.output_dir, exist_ok=True)
-        srt = os.path.join(self.user_dir, 'subs.srt')
-        if os.path.exists(srt): os.remove(srt)
     def generate_comic(self):
         try:
             if cv2 is None: raise Exception("OpenCV missing on server.")
-            self.cleanup_generated()
             self.update_status("Processing Video...", 5)
             cap = cv2.VideoCapture(self.video_path)
@@ -676,7 +682,7 @@ class EnhancedComicGenerator:
                 get_real_subtitles(self.video_path)
                 if os.path.exists('test1.srt'): shutil.move('test1.srt', user_srt)
             except:
-                with open(user_srt, 'w') as f: f.write("1\n00:00:00,000 --> 00:00:05,000\nStart\n")
             # 2. Keyframes
             self.update_status("Generating Panels...", 40)
@@ -686,7 +692,7 @@ class EnhancedComicGenerator:
             frame_files = []
             bubbles = []
-            limit_subs = subs[:12] # Limit to 12 panels for demo speed
             for i, sub in enumerate(limit_subs):
                 mid = (sub.start.total_seconds() + sub.end.total_seconds()) / 2
@@ -696,7 +702,6 @@ class EnhancedComicGenerator:
                     fname = f"frame_{i}.png"
                     cv2.imwrite(os.path.join(self.frames_dir, fname), frame)
                     frame_files.append(fname)
-                    # Map file to time for regeneration
                     self.frame_metadata[fname] = mid
                     bubbles.append(bubble(
@@ -706,13 +711,13 @@ class EnhancedComicGenerator:
                     ))
             cap.release()
-            # Save metadata for features like "Next Frame"
             with open(os.path.join(self.frames_dir, 'frame_metadata.json'), 'w') as f:
                 json.dump(self.frame_metadata, f)
             # 3. Enhance
             self.update_status("Enhancing...", 70)
-            # (Call enhancers here if needed)
             # 4. Assemble
             self.update_status("Finalizing...", 90)
@@ -758,6 +763,8 @@ class EnhancedComicGenerator:
                 cv2.imwrite(os.path.join(self.frames_dir, fname), frame)
                 meta[fname] = new_time
                 with open(meta_path,'w') as f: json.dump(meta, f)
                 return {"success":True}
             return {"success":False, "message":"End of video"}
         except Exception as e: return {"success":False, "message":str(e)}
@@ -770,15 +777,34 @@ class EnhancedComicGenerator:
             cap.release()
             if ret:
                 cv2.imwrite(os.path.join(self.frames_dir, fname), frame)
-                # Update meta
                 meta_path = os.path.join(self.frames_dir, 'frame_metadata.json')
                 if os.path.exists(meta_path):
                     with open(meta_path,'r') as f: meta = json.load(f)
                     meta[fname] = float(ts)
                     with open(meta_path,'w') as f: json.dump(meta, f)
                 return {"success":True}
             return {"success":False, "message":"Invalid time"}
         except Exception as e: return {"success":False, "message":str(e)}
 # --- ROUTES ---
 @app.route('/')
@@ -790,6 +816,10 @@ def upload():
     if not sid: return "Missing SID", 400
     f = request.files['file']
     gen = EnhancedComicGenerator(sid)
     f.save(gen.video_path)
     gen.update_status("Starting...", 5)
     threading.Thread(target=gen.generate_comic).start()

 import shutil
 import json
 import traceback
+import logging
 from concurrent.futures import ThreadPoolExecutor
 from flask import Flask, render_template, request, jsonify, send_from_directory, send_file
+# --- 0. SUPPRESS HUGGING FACE WARNINGS ---
+import warnings
+warnings.filterwarnings("ignore", category=UserWarning)
+os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3"
+logging.getLogger("transformers").setLevel(logging.ERROR)
 # --- 1. CORE DEPENDENCY CHECKS ---
 try:
     import cv2
     import srt
 except ImportError as e:
     print(f"❌ CRITICAL ERROR: Missing python library. {e}")
     cv2 = None
     np = None
     Image = None
     srt = None
 # --- 2. BACKEND MODULE IMPORTS (WITH ROBUST FALLBACKS) ---
 def dummy_black_bar_crop(): return 0, 0, None, None
 try:
     from backend.class_def import bubble, panel, Page
     print("✅ Core class definitions loaded.")
 except Exception:
     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': bubble_offset_x, 'bubble_offset_y': bubble_offset_y,
     from backend.ai_enhanced_core import image_processor, comic_styler, face_detector, layout_optimizer
     from backend.ai_bubble_placement import ai_bubble_placer
     from backend.subtitles.subs_real import get_real_subtitles
+    from backend.keyframes.keyframes_simple import generate_keyframes_simple
     print("✅ Core utility modules loaded.")
 except Exception:
     def get_real_subtitles(v): pass
     class DummyDetector:
         def detect_faces(self, p): return []
 app = Flask(__name__)
 BASE_USER_DIR = "userdata"
+# --- HTML INTERFACE ---
 INDEX_HTML = '''
 <!DOCTYPE html>
 <html lang="en">
     <meta charset="UTF-8">
     <meta name="viewport" content="width=device-width, initial-scale=1.0">
     <title>Movie to Comic Generator</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&family=Lato&display=swap" rel="stylesheet">
     <style>
         /* GLOBAL STYLES */
             updateImageTransform(img);
         }
+        // --- API Calls ---
         function replacePanelImage() {
+            if (!currentlySelectedPanel) return alert("Select panel");
             const img = currentlySelectedPanel.querySelector('img');
             const inp = document.getElementById('image-uploader');
             inp.onchange = async (e) => {
                 json.dump({'message': message, 'progress': progress}, f)
         except: pass
+    # --- CLEANUP: Deletes OLD processing files on NEW upload ---
+    def cleanup_previous_run(self):
+        print(f"[{self.sid}] 🧹 Cleaning previous run...")
+        if os.path.exists(self.frames_dir):
+            for f in os.listdir(self.frames_dir):
+                try: os.remove(os.path.join(self.frames_dir, f))
+                except: pass
+        if os.path.exists(self.output_dir):
+            for f in os.listdir(self.output_dir):
+                try: os.remove(os.path.join(self.output_dir, f))
+                except: pass
+        srt_file = os.path.join(self.user_dir, 'subs.srt')
+        if os.path.exists(srt_file): os.remove(srt_file)
     def generate_comic(self):
         try:
             if cv2 is None: raise Exception("OpenCV missing on server.")
             self.update_status("Processing Video...", 5)
             cap = cv2.VideoCapture(self.video_path)
                 get_real_subtitles(self.video_path)
                 if os.path.exists('test1.srt'): shutil.move('test1.srt', user_srt)
             except:
+                with open(user_srt, 'w') as f: f.write("1\n00:00:00,000 --> 00:00:05,000\n...\n")
             # 2. Keyframes
             self.update_status("Generating Panels...", 40)
             frame_files = []
             bubbles = []
+            limit_subs = subs[:12] # Limit to 12 panels for demo
             for i, sub in enumerate(limit_subs):
                 mid = (sub.start.total_seconds() + sub.end.total_seconds()) / 2
                     fname = f"frame_{i}.png"
                     cv2.imwrite(os.path.join(self.frames_dir, fname), frame)
                     frame_files.append(fname)
                     self.frame_metadata[fname] = mid
                     bubbles.append(bubble(
                     ))
             cap.release()
             with open(os.path.join(self.frames_dir, 'frame_metadata.json'), 'w') as f:
                 json.dump(self.frame_metadata, f)
             # 3. Enhance
             self.update_status("Enhancing...", 70)
+            self._enhance_all_images()
+            self._enhance_quality_colors()
             # 4. Assemble
             self.update_status("Finalizing...", 90)
                 cv2.imwrite(os.path.join(self.frames_dir, fname), frame)
                 meta[fname] = new_time
                 with open(meta_path,'w') as f: json.dump(meta, f)
+                self._enhance_all_images(single_image_path=os.path.join(self.frames_dir, fname))
+                self._enhance_quality_colors(single_image_path=os.path.join(self.frames_dir, fname))
                 return {"success":True}
             return {"success":False, "message":"End of video"}
         except Exception as e: return {"success":False, "message":str(e)}
             cap.release()
             if ret:
                 cv2.imwrite(os.path.join(self.frames_dir, fname), frame)
                 meta_path = os.path.join(self.frames_dir, 'frame_metadata.json')
                 if os.path.exists(meta_path):
                     with open(meta_path,'r') as f: meta = json.load(f)
                     meta[fname] = float(ts)
                     with open(meta_path,'w') as f: json.dump(meta, f)
+                self._enhance_all_images(single_image_path=os.path.join(self.frames_dir, fname))
+                self._enhance_quality_colors(single_image_path=os.path.join(self.frames_dir, fname))
                 return {"success":True}
             return {"success":False, "message":"Invalid time"}
         except Exception as e: return {"success":False, "message":str(e)}
+    def _enhance_all_images(self, single_image_path=None):
+        try:
+            enhancer = SimpleColorEnhancer()
+            if single_image_path: enhancer.enhance_single(single_image_path)
+            else:
+                paths = [os.path.join(self.frames_dir, f) for f in os.listdir(self.frames_dir) if f.endswith('.png')]
+                with ThreadPoolExecutor() as ex: list(ex.map(enhancer.enhance_single, paths))
+        except: pass
+    def _enhance_quality_colors(self, single_image_path=None):
+        try:
+            enhancer = QualityColorEnhancer()
+            if single_image_path: enhancer.enhance_single(single_image_path)
+            else:
+                paths = [os.path.join(self.frames_dir, f) for f in os.listdir(self.frames_dir) if f.endswith('.png')]
+                with ThreadPoolExecutor() as ex: list(ex.map(enhancer.enhance_single, paths))
+        except: pass
 # --- ROUTES ---
 @app.route('/')
     if not sid: return "Missing SID", 400
     f = request.files['file']
     gen = EnhancedComicGenerator(sid)
+    # CLEAN OLD FILES
+    gen.cleanup_previous_run()
     f.save(gen.video_path)
     gen.update_status("Starting...", 5)
     threading.Thread(target=gen.generate_comic).start()