Update app_enhanced.py

app_enhanced.py

@@ -15,15 +15,21 @@ import torch
 from flask import Flask, jsonify, request, send_from_directory, send_file
 
 # ======================================================
-# 🚀 CUDA CONFIGURATION
+# 🚀 H100 / CUDA CONFIGURATION
 # ======================================================
-def check_cuda():
-    """Checks for CUDA availability without ZeroGPU decorators"""
+def initialize_h100():
+    """Checks and initializes the H100 GPU environment"""
     if torch.cuda.is_available():
-        print(f"✅ CUDA Engine Active: {torch.cuda.get_device_name(0)}")
+        gpu_name = torch.cuda.get_device_name(0)
+        vram = torch.cuda.get_device_properties(0).total_memory / 1e9
+        print(f"\n{'='*40}")
+        print(f"✅ H100 DETECTED: {gpu_name}")
+        print(f"🚀 VRAM AVAILABLE: {vram:.2f} GB")
+        print(f"✅ CUDA Version: {torch.version.cuda}")
+        print(f"{'='*40}\n")
         return True
     else:
-        print("⚠️ CUDA Not Detected. Running on CPU (Expect slower generation)")
+        print("\n⚠️ WARNING: CUDA NOT DETECTED. Running on CPU (H100 not active?)\n")
         return False
 
 # ======================================================
@@ -60,8 +66,9 @@ app = Flask(__name__)
 BASE_USER_DIR = "userdata"
 SAVED_COMICS_DIR = "saved_comics"
 
-os.makedirs(BASE_USER_DIR, exist_ok=True)
-os.makedirs(SAVED_COMICS_DIR, exist_ok=True)
+# Ensure write permissions for Docker environment
+os.makedirs(BASE_USER_DIR, mode=0o777, exist_ok=True)
+os.makedirs(SAVED_COMICS_DIR, mode=0o777, exist_ok=True)
 
 def generate_save_code(length=8):
     chars = string.ascii_uppercase + string.digits
@@ -71,16 +78,13 @@ def generate_save_code(length=8):
             return code
 
 # ======================================================
-# 🧠 CORE PROCESSING FUNCTIONS (Standard CUDA)
-# ======================================================
-# Removed @spaces.GPU decorators. These functions now run 
-# directly in the thread/process invoking them.
+# 🧠 CORE LOGIC (Native CUDA)
 # ======================================================
 
 def generate_comic_core(video_path, user_dir, frames_dir, metadata_path, target_pages):
-    print(f"🚀 Processing Task Started: {video_path} | Pages: {target_pages}")
+    print(f"🚀 Processing Started on H100: {video_path}")
     
-    # Import backend logic here to ensure they load into the correct context
+    # Imports inside function to ensure they use the initialized CUDA context
     from backend.keyframes.keyframes import black_bar_crop
     from backend.simple_color_enhancer import SimpleColorEnhancer
     from backend.quality_color_enhancer import QualityColorEnhancer
@@ -99,18 +103,17 @@ def generate_comic_core(video_path, user_dir, frames_dir, metadata_path, target_
     # 2. Subtitles Generation
     user_srt = os.path.join(user_dir, 'subs.srt')
     try:
+        # This will use the GPU for Whisper/ASR if implemented in backend
         get_real_subtitles(video_path)
         if os.path.exists('test1.srt'):
             shutil.move('test1.srt', user_srt)
     except Exception as e:
-        print(f"Subtitle error (using placeholder): {e}")
+        print(f"Subtitle generation skipped/failed: {e}")
         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 = []
 
     # 3. Smart Keyframe Selection
     valid_subs = [s for s in all_subs if s.content.strip()]
@@ -152,11 +155,11 @@ def generate_comic_core(video_path, user_dir, frames_dir, metadata_path, target_
 
     with open(metadata_path, 'w') as f: json.dump(frame_metadata, f, indent=2)
         
-    # 5. Image Enhancement
+    # 5. Image Enhancement (GPU Accelerated)
     try: black_bar_crop() 
     except: pass
 
-    # Initialize enhancers (they usually load models to CUDA if available)
+    # Initialize enhancers - these should load to GPU automatically if backend supports it
     se = SimpleColorEnhancer()
     qe = QualityColorEnhancer()
 
@@ -167,7 +170,7 @@ def generate_comic_core(video_path, user_dir, frames_dir, metadata_path, target_
         try: qe.enhance_single(p, p)
         except: pass
         
-    # 6. Bubble Placement
+    # 6. Bubble Placement (Uses Face Detection on GPU)
     bubbles_list = []
     for f in frame_files_ordered:
         p = os.path.join(frames_dir, f)
@@ -280,8 +283,8 @@ class EnhancedComicGenerator:
 
     def run(self, target_pages):
         try:
-            self.write_status("Waiting for GPU/Processor...", 5)
-            # Directly call the core function (standard CUDA)
+            self.write_status("Waiting for H100...", 5)
+            # Direct call to core function (no decorator)
             data = generate_comic_core(self.video_path, self.user_dir, self.frames_dir, self.metadata_path, int(target_pages))
             with open(os.path.join(self.output_dir, 'pages.json'), 'w') as f:
                 json.dump(data, f, indent=2)
@@ -303,7 +306,7 @@ INDEX_HTML = '''
 <head> 
 <meta charset="UTF-8"> 
 <meta name="viewport" content="width=device-width, initial-scale=1.0"> 
-<title>🎬 Enhanced Comic Generator</title> 
+<title>🎬 H100 Comic Studio</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=Gloria+Hallelujah&family=Lato&display=swap" rel="stylesheet"> 
 <style> 
@@ -446,7 +449,7 @@ INDEX_HTML = '''
 <body> 
 <div id="upload-container"> 
     <div class="upload-box"> 
-        <h1>🎬 Enhanced Comic Generator</h1> 
+        <h1>🎬 H100 Comic Studio</h1> 
         <input type="file" id="file-upload" class="file-input" onchange="document.getElementById('fn').innerText=this.files[0].name"> 
         <label for="file-upload" class="file-label">📁 Choose Video File</label> 
         <span id="fn" style="margin-bottom:10px; display:block; color:#666;">No file selected</span>
@@ -455,7 +458,7 @@ INDEX_HTML = '''
             <input type="number" id="page-count" value="4" min="1" max="15" placeholder="e.g. 4">
             <small style="color:#666; font-size:11px; display:block; margin-top:5px;">System calculates ~4 panels per page.</small>
         </div>
-        <button class="submit-btn" onclick="upload()">🚀 Generate Comic</button>
+        <button class="submit-btn" onclick="upload()">🚀 Generate Comic (H100 Speed)</button>
         <button id="restore-draft-btn" class="restore-btn" style="display:none; margin-top:15px;" onclick="restoreDraft()">📂 Restore Unsaved Draft</button>
         <div class="load-section">
             <h3>📥 Load Saved Comic</h3>
@@ -466,7 +469,7 @@ INDEX_HTML = '''
         </div>
         <div class="loading-view" id="loading-view" style="display:none; margin-top:20px;">
             <div class="loader" style="margin:0 auto;"></div>
-            <p id="status-text" style="margin-top:10px;">Starting...</p>
+            <p id="status-text" style="margin-top:10px;">Firing up the H100...</p>
         </div>
     </div>
 </div>
@@ -903,7 +906,7 @@ def upload():
     f.save(gen.video_path)
     gen.write_status("Starting...", 5)
     
-    # Run in thread - Standard processing
+    # Run in thread - Standard H100 processing
     threading.Thread(target=gen.run, args=(target_pages,)).start()
     return jsonify({'success': True, 'message': 'Generation started.'})
 
@@ -929,7 +932,7 @@ def regen():
     sid = request.args.get('sid')
     d = request.get_json()
     gen = EnhancedComicGenerator(sid)
-    # Direct function call
+    # Direct function call - H100
     return jsonify(regen_frame_core(gen.video_path, gen.frames_dir, gen.metadata_path, d['filename'], d['direction']))
 
 @app.route('/goto_timestamp', methods=['POST'])
@@ -937,7 +940,7 @@ def go_time():
     sid = request.args.get('sid')
     d = request.get_json()
     gen = EnhancedComicGenerator(sid)
-    # Direct function call
+    # Direct function call - H100
     return jsonify(get_frame_at_ts_core(gen.video_path, gen.frames_dir, gen.metadata_path, d['filename'], float(d['timestamp'])))
 
 @app.route('/replace_panel', methods=['POST'])
@@ -1004,5 +1007,5 @@ def load_comic(code):
         return jsonify({'success': False, 'message': str(e)})
 
 if __name__ == '__main__':
-    check_cuda()
+    initialize_h100()
     app.run(host='0.0.0.0', port=7860)