Update app_enhanced.py

app_enhanced.py

@@ -11,10 +11,43 @@ import random
 from concurrent.futures import ThreadPoolExecutor
 from flask import Flask, render_template, request, jsonify, send_from_directory, send_file
 
+# --- HUGGING FACE ZEROGPU SETUP ---
+try:
+    import spaces
+    print("✅ Hugging Face Spaces 'spaces' module detected.")
+    HAS_ZEROGPU = True
+except ImportError:
+    print("⚠️ 'spaces' module not found. Running in standard mode (CPU/GPU without ZeroGPU allocator).")
+    HAS_ZEROGPU = False
+
+# Conditional Decorator for ZeroGPU
+def gpu_task(duration=120):
+    def decorator(func):
+        if HAS_ZEROGPU:
+            return spaces.GPU(duration=duration)(func)
+        return func
+    return decorator
+
 # --- 0. CONFIG & LOGGING ---
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
+# --- HUGGING FACE FILESYSTEM SETUP ---
+# Spaces are read-only in root. We must use /tmp or /data (persistent storage)
+if os.access('/data', os.W_OK):
+    BASE_PATH = '/data'
+    print("✅ Using Persistent Storage at /data")
+else:
+    BASE_PATH = '/tmp'
+    print("✅ Using Temporary Storage at /tmp")
+
+BASE_USER_DIR = os.path.join(BASE_PATH, "userdata")
+SAVED_COMICS_DIR = os.path.join(BASE_PATH, "saved_comics")
+
+# Create directories
+os.makedirs(BASE_USER_DIR, exist_ok=True)
+os.makedirs(SAVED_COMICS_DIR, exist_ok=True)
+
 # --- 1. CORE DEPENDENCY CHECKS ---
 try:
     import cv2
@@ -98,12 +131,6 @@ except Exception as e:
 
 # --- FLASK APP SETUP ---
 app = Flask(__name__)
-BASE_USER_DIR = "userdata"
-SAVED_COMICS_DIR = "saved_comics"
-
-# Create directories
-os.makedirs(BASE_USER_DIR, exist_ok=True)
-os.makedirs(SAVED_COMICS_DIR, exist_ok=True)
 
 def generate_save_code(length=8):
     """Generate a unique save code"""
@@ -1361,6 +1388,8 @@ class EnhancedComicGenerator:
             
             return pages
 
+    # --- APPLY ZEROGPU DECORATOR HERE ---
+    @gpu_task(duration=180) # Allocate GPU for 3 minutes for this task
     def generate_comic(self):
         start_time = time.time()
         try:
@@ -1451,6 +1480,7 @@ class EnhancedComicGenerator:
         except Exception as e:
             print(f"❌ Save results failed: {e}")
 
+    @gpu_task(duration=60)
     def regenerate_frame(self, fname, direction):
         try:
             if not os.path.exists(self.metadata_path):
@@ -1506,6 +1536,7 @@ class EnhancedComicGenerator:
             traceback.print_exc()
             return {"success": False, "message": str(e)}
 
+    @gpu_task(duration=60)
     def get_frame_at_timestamp(self, fname, ts):
         try:
             cap = cv2.VideoCapture(self.video_path)
@@ -1573,6 +1604,7 @@ def upload():
     f.save(gen.video_path)
     gen.update_status("Starting...", 5)
     
+    # We use a Thread to call the generated GPU function
     threading.Thread(target=gen.generate_comic).start()
     return jsonify({'success': True, 'message': 'Generation started.'})
 
@@ -1609,6 +1641,7 @@ def regen():
     
     d = request.get_json()
     gen = EnhancedComicGenerator(sid)
+    # This might use ZeroGPU if configured
     return jsonify(gen.regenerate_frame(d['filename'], d['direction']))
 
 @app.route('/goto_timestamp', methods=['POST'])
@@ -1730,8 +1763,7 @@ def get_saved_frame(code, filename):
 
 
 if __name__ == '__main__':
-    os.makedirs(BASE_USER_DIR, exist_ok=True)
-    os.makedirs(SAVED_COMICS_DIR, exist_ok=True)
+    # HF Spaces use port 7860 by default
     port = int(os.getenv("PORT", 7860))
     print(f"🚀 Starting Enhanced Comic Generator on host 0.0.0.0, port {port}")
     print(f"📁 User data directory: {BASE_USER_DIR}")