RealComic

Configuration error

App Files Files Community

tester343 commited on Dec 3, 2025

Commit

d828036

verified ·

1 Parent(s): 2918181

Update app_enhanced.py

Browse files

Files changed (1) hide show

app_enhanced.py +111 -93

app_enhanced.py CHANGED Viewed

@@ -6,7 +6,7 @@ import shutil
 import json
 import traceback
 from concurrent.futures import ThreadPoolExecutor
-from flask import Flask, render_template, request, jsonify, send_from_directory, send_file, session
 # --- 1. CORE DEPENDENCY CHECKS ---
 try:
@@ -16,17 +16,15 @@ try:
     import srt
 except ImportError as e:
     print(f"❌ CRITICAL ERROR: Missing python library. {e}")
-    # We define dummy variables to let the app start, but generation will fail with a clear message
     cv2 = None
     np = None
     Image = None
     srt = None
-# --- 2. BACKEND MODULE IMPORT WITH FALLBACKS ---
-# This ensures the app doesn't crash if a specific backend file is missing
-def dummy_function(*args, **kwargs):
-    print("⚠️ Warning: Function not loaded correctly.")
-    return 0, 0, None, None
 try:
     from backend.keyframes.keyframes import black_bar_crop
@@ -54,31 +52,27 @@ except Exception:
         def __init__(self, panels, bubbles): self.panels, self.bubbles = panels, bubbles
 try:
-    from backend.ai_enhanced_core import image_processor, face_detector
     from backend.ai_bubble_placement import ai_bubble_placer
     from backend.subtitles.subs_real import get_real_subtitles
-except Exception as e:
-    print(f"⚠️ Backend Import Error: {e}")
-    # Define dummy get_real_subtitles to prevent NameError
-    def get_real_subtitles(video_path):
-        raise Exception("Subtitle module failed to load. Check backend/subtitles/subs_real.py")
-    # Define dummy face detector
     class DummyDetector:
         def detect_faces(self, p): return []
         def get_lip_position(self, p, f): return -1, -1
     face_detector = DummyDetector()
-    # Define dummy bubble placer
     class DummyPlacer:
         def place_bubble_ai(self, p, l): return 50, 20
     ai_bubble_placer = DummyPlacer()
 # --- FLASK APP SETUP ---
 app = Flask(__name__)
-app.secret_key = "HF_SPACE_SECRET_KEY_555"
 BASE_USER_DIR = "userdata"
 INDEX_HTML = '''
 <!DOCTYPE html>
 <html lang="en">
@@ -99,31 +93,32 @@ INDEX_HTML = '''
         @keyframes ball { 0%{background-position:0 50%} 100%{background-position:100px 50%} }
         /* COMIC STYLES */
-        .comic-page { background: white; width: 600px; height: 400px; position: relative; overflow: hidden; border: 2px solid #000; margin: 0 auto 20px; }
         .comic-grid { display: grid; grid-template-columns: 1fr 1fr; grid-template-rows: 1fr 1fr; gap: 10px; width: 100%; height: 100%; padding: 10px; box-sizing: border-box; }
         .panel { position: relative; overflow: hidden; border: 2px solid #000; background: #eee; }
         .panel img { width: 100%; height: 100%; object-fit: cover; }
-        /* SPEECH BUBBLE - EXPORT SAFE GRADIENT */
         .speech-bubble.speech {
-            --b: 3em; --h: 1.8em; --t: 0.6; --p: var(--tail-pos, 50%); --r: 1.2em;
-            --c: var(--bubble-fill-color, #4ECDC4);
-            background: var(--c); color: var(--bubble-text-color, #fff); padding: 1em; position: absolute;
             border-radius: var(--r) var(--r) min(var(--r), calc(100% - var(--p) - (1 - var(--t)) * var(--b) / 2)) min(var(--r), calc(var(--p) - (1 - var(--t)) * var(--b) / 2)) / var(--r);
-            font-family: 'Comic Neue', cursive; font-weight: bold; font-size: 13px; text-align: center;
-            min-width: 60px; min-height: 40px; display: flex; align-items: center; justify-content: center;
             cursor: move; z-index: 10;
         }
         .speech-bubble.speech:before {
             content: ""; position: absolute; width: var(--b); height: var(--h);
             background: radial-gradient(100% 100% at 100% 0, transparent calc(var(--t) * 100% - 1px), var(--c) calc(var(--t) * 100%));
             border-bottom-left-radius: 100%; pointer-events: none; z-index: 1;
         }
-        .speech-bubble.speech.tail-bottom:before { top: 99%; left: clamp(0%, calc(var(--p) - (1 - var(--t)) * var(--b) / 2), calc(100% - (1 - var(--t)) * var(--b))); }
-        .speech-bubble.selected { outline: 2px dashed #333; }
         .edit-controls { position: fixed; bottom: 20px; right: 20px; background: white; padding: 15px; border-radius: 8px; box-shadow: 0 5px 15px rgba(0,0,0,0.2); width: 200px; }
-        .edit-controls button { width: 100%; margin-top: 5px; padding: 8px; cursor: pointer; }
     </style>
 </head>
 <body>
@@ -132,7 +127,7 @@ INDEX_HTML = '''
             <h1>🎬 Comic Generator</h1>
             <input type="file" id="file-upload" class="file-input" onchange="document.getElementById('fname').innerText=this.files[0].name">
             <label for="file-upload" class="file-label">Choose Video</label>
-            <span id="fname">No file</span>
             <button class="submit-btn" onclick="upload()">Generate</button>
             <div id="loading" style="display:none;">
                 <div class="loader"></div>
@@ -144,13 +139,31 @@ INDEX_HTML = '''
     <div id="editor-container">
         <div id="comic-pages"></div>
         <div class="edit-controls">
-            <h4>Editor</h4>
             <button onclick="exportToPng()">💾 Export PNG</button>
         </div>
     </div>
     <script>
         let interval;
         async function upload() {
             const file = document.getElementById('file-upload').files[0];
             if(!file) return alert("Select file");
@@ -161,13 +174,14 @@ INDEX_HTML = '''
             document.querySelector('.upload-box').style.display='none';
             document.getElementById('loading').style.display='block';
-            const res = await fetch('/uploader', {method:'POST', body:fd});
             if(res.ok) interval = setInterval(checkStatus, 2000);
             else { alert("Upload failed"); location.reload(); }
         }
         async function checkStatus() {
-            const res = await fetch('/status');
             const data = await res.json();
             document.getElementById('status').innerText = data.message;
@@ -184,8 +198,10 @@ INDEX_HTML = '''
         }
         function loadComic() {
-            fetch('output/pages.json').then(r=>r.json()).then(data => {
                 const c = document.getElementById('comic-pages');
                 data.forEach((p, i) => {
                     const div = document.createElement('div');
                     div.className = 'comic-page';
@@ -197,15 +213,14 @@ INDEX_HTML = '''
                         const pDiv = document.createElement('div');
                         pDiv.className = 'panel';
                         const img = document.createElement('img');
-                        img.src = 'frames/' + pan.image;
                         pDiv.appendChild(img);
-                        // Add bubble
                         if(p.bubbles && p.bubbles[j] && p.bubbles[j].dialog) {
                             const b = document.createElement('div');
-                            b.className = 'speech-bubble speech tail-bottom';
                             b.innerText = p.bubbles[j].dialog;
-                            // Safe defaults
                             b.style.left = (p.bubbles[j].bubble_offset_x || 50) + 'px';
                             b.style.top = (p.bubbles[j].bubble_offset_y || 20) + 'px';
                             pDiv.appendChild(b);
@@ -216,13 +231,12 @@ INDEX_HTML = '''
                     div.appendChild(grid);
                     c.appendChild(div);
                 });
-            });
         }
         function makeInteractive(el) {
             el.onmousedown = function(e) {
                 e.stopPropagation();
-                // Simple drag logic
                 const offX = e.clientX - el.offsetLeft;
                 const offY = e.clientY - el.offsetTop;
                 document.onmousemove = function(ev) {
@@ -235,12 +249,14 @@ INDEX_HTML = '''
         async function exportToPng() {
             const pages = document.querySelectorAll('.comic-page');
-            for(let p of pages) {
-                const url = await htmlToImage.toPng(p, {pixelRatio: 2});
-                const a = document.createElement('a');
-                a.download = 'comic.png';
-                a.href = url;
-                a.click();
             }
         }
     </script>
@@ -268,37 +284,38 @@ class EnhancedComicGenerator:
                 json.dump({'message': message, 'progress': progress}, f)
         except: pass
     def generate_comic(self):
         try:
-            if cv2 is None: raise Exception("OpenCV not installed on server.")
             self.update_status("Processing Video...", 10)
             cap = cv2.VideoCapture(self.video_path)
-            if not cap.isOpened(): raise Exception("Invalid Video File")
-            self.video_fps = cap.get(cv2.CAP_PROP_FPS)
-            total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
-            duration = total_frames / self.video_fps
             cap.release()
-            # --- GENERATE SUBTITLES ---
             self.update_status("Extracting Dialogue...", 30)
             try:
-                get_real_subtitles(self.video_path) # Generates 'test1.srt' in cwd usually
-                # Move it to user dir
                 if os.path.exists('test1.srt'):
                     shutil.move('test1.srt', os.path.join(self.user_dir, 'subs.srt'))
-            except Exception as e:
-                print(f"Subtitle Error: {e}")
-                # Create dummy srt if failed
                 with open(os.path.join(self.user_dir, 'subs.srt'), 'w') as f:
-                    f.write("1\n00:00:01,000 --> 00:00:04,000\nHello World!\n")
-            # --- EXTRACT FRAMES ---
-            self.update_status("Creating Panels...", 50)
             cap = cv2.VideoCapture(self.video_path)
-            # Parse SRT
             subs_path = os.path.join(self.user_dir, 'subs.srt')
             with open(subs_path, 'r', encoding='utf-8') as f:
                 subs = list(srt.parse(f.read()))
@@ -306,84 +323,85 @@ class EnhancedComicGenerator:
             frame_files = []
             bubbles = []
-            # Limit to 12 frames for speed
-            step = max(1, len(subs) // 12)
-            for i, sub in enumerate(subs[::step]):
-                mid_time = (sub.start.total_seconds() + sub.end.total_seconds()) / 2
-                cap.set(cv2.CAP_PROP_POS_MSEC, mid_time * 1000)
                 ret, frame = cap.read()
                 if ret:
                     fname = f"frame_{i}.png"
                     cv2.imwrite(os.path.join(self.frames_dir, fname), frame)
                     frame_files.append(fname)
-                    # Simple bubble placement
                     bubbles.append(bubble(dialog=sub.content, bubble_offset_x=50, bubble_offset_y=20))
             cap.release()
-            # --- GENERATE PAGES ---
-            self.update_status("Assembling...", 80)
             pages_data = []
-            # Create 4-panel pages
             for i in range(0, len(frame_files), 4):
-                batch_files = frame_files[i:i+4]
-                batch_bubbles = bubbles[i:i+4]
-                panels = [{'image': f} for f in batch_files]
-                b_data = [b.__dict__ if hasattr(b, '__dict__') else b for b in batch_bubbles]
                 pages_data.append({'panels': panels, 'bubbles': b_data})
             with open(os.path.join(self.output_dir, 'pages.json'), 'w') as f:
                 json.dump(pages_data, f)
-            # Save HTML for direct viewing
-            with open(os.path.join(self.output_dir, 'page.html'), 'w') as f:
-                f.write(INDEX_HTML)
             self.update_status("Done!", 100)
         except Exception as e:
             traceback.print_exc()
-            self.update_status(f"Failed: {str(e)}", -1)
 @app.route('/')
 def index():
-    if 'sid' not in session: session['sid'] = uuid.uuid4().hex
     return INDEX_HTML
 @app.route('/uploader', methods=['POST'])
 def upload():
-    if 'sid' not in session: session['sid'] = uuid.uuid4().hex
-    sid = session['sid']
     if 'file' not in request.files: return "No file", 400
     gen = EnhancedComicGenerator(sid)
     request.files['file'].save(gen.video_path)
-    # Initial status
     gen.update_status("Starting...", 5)
     threading.Thread(target=gen.generate_comic).start()
     return jsonify({'success': True})
 @app.route('/status')
 def get_status():
-    if 'sid' not in session: return jsonify({'progress': -1, 'message': "Session Lost"})
-    sid = session['sid']
     status_path = os.path.join(BASE_USER_DIR, sid, 'output', 'status.json')
     if os.path.exists(status_path):
         return send_file(status_path)
-    return jsonify({'progress': 0, 'message': "Waiting..."})
 @app.route('/output/<path:filename>')
 def get_output(filename):
-    return send_from_directory(os.path.join(BASE_USER_DIR, session['sid'], 'output'), filename)
 @app.route('/frames/<path:filename>')
 def get_frame(filename):
-    return send_from_directory(os.path.join(BASE_USER_DIR, session['sid'], 'frames'), filename)
 if __name__ == '__main__':
     os.makedirs(BASE_USER_DIR, exist_ok=True)
-    app.run(host='0.0.0.0', port=7860)

 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 srt
 except ImportError as e:
     print(f"❌ CRITICAL ERROR: Missing python library. {e}")
+    # Dummy definitions to allow app to start and show error in UI
     cv2 = None
     np = None
     Image = None
     srt = None
+# --- 2. BACKEND MODULE IMPORTS (WITH DUMMY FALLBACKS) ---
+# This allows the app to run even if complex backend modules fail to load
+def dummy_function(*args, **kwargs): return 0, 0, None, None
 try:
     from backend.keyframes.keyframes import black_bar_crop
         def __init__(self, panels, bubbles): self.panels, self.bubbles = panels, bubbles
 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
+    from backend.keyframes.keyframes_simple import generate_keyframes_simple
+except Exception:
+    # If backend fails, define minimal dummies to prevent crash
+    def get_real_subtitles(v): pass
     class DummyDetector:
         def detect_faces(self, p): return []
         def get_lip_position(self, p, f): return -1, -1
     face_detector = DummyDetector()
     class DummyPlacer:
         def place_bubble_ai(self, p, l): return 50, 20
     ai_bubble_placer = DummyPlacer()
 # --- FLASK APP SETUP ---
 app = Flask(__name__)
 BASE_USER_DIR = "userdata"
+# --- HTML ---
 INDEX_HTML = '''
 <!DOCTYPE html>
 <html lang="en">
         @keyframes ball { 0%{background-position:0 50%} 100%{background-position:100px 50%} }
         /* COMIC STYLES */
+        .comic-page { background: white; width: 600px; height: 400px; position: relative; overflow: hidden; border: 2px solid #000; margin: 0 auto 20px; box-shadow: 0 4px 10px rgba(0,0,0,0.1); }
         .comic-grid { display: grid; grid-template-columns: 1fr 1fr; grid-template-rows: 1fr 1fr; gap: 10px; width: 100%; height: 100%; padding: 10px; box-sizing: border-box; }
         .panel { position: relative; overflow: hidden; border: 2px solid #000; background: #eee; }
         .panel img { width: 100%; height: 100%; object-fit: cover; }
+        /* EXACT SHARK FIN SPEECH BUBBLE CSS */
         .speech-bubble.speech {
+            --b: 3em; --h: 1.8em; --t: 0.6; --p: 50%; --r: 1.2em;
+            --c: #4ECDC4;
+            background: var(--c); color: #fff; padding: 1em; position: absolute;
             border-radius: var(--r) var(--r) min(var(--r), calc(100% - var(--p) - (1 - var(--t)) * var(--b) / 2)) min(var(--r), calc(var(--p) - (1 - var(--t)) * var(--b) / 2)) / var(--r);
+            font-family: 'Comic Neue', cursive; font-weight: bold; font-size: 14px; text-align: center;
+            min-width: 80px; min-height: 40px; display: flex; align-items: center; justify-content: center;
             cursor: move; z-index: 10;
         }
+        /* Export-Safe Gradient instead of Mask */
         .speech-bubble.speech:before {
             content: ""; position: absolute; width: var(--b); height: var(--h);
             background: radial-gradient(100% 100% at 100% 0, transparent calc(var(--t) * 100% - 1px), var(--c) calc(var(--t) * 100%));
             border-bottom-left-radius: 100%; pointer-events: none; z-index: 1;
+            top: 99%; left: clamp(0%, calc(var(--p) - (1 - var(--t)) * var(--b) / 2), calc(100% - (1 - var(--t)) * var(--b)));
         }
         .edit-controls { position: fixed; bottom: 20px; right: 20px; background: white; padding: 15px; border-radius: 8px; box-shadow: 0 5px 15px rgba(0,0,0,0.2); width: 200px; }
+        .edit-controls button { width: 100%; margin-top: 5px; padding: 8px; cursor: pointer; background: #f0f0f0; border: 1px solid #ccc; border-radius: 4px; }
+        .edit-controls button:hover { background: #e0e0e0; }
     </style>
 </head>
 <body>
             <h1>🎬 Comic Generator</h1>
             <input type="file" id="file-upload" class="file-input" onchange="document.getElementById('fname').innerText=this.files[0].name">
             <label for="file-upload" class="file-label">Choose Video</label>
+            <span id="fname">No file selected</span>
             <button class="submit-btn" onclick="upload()">Generate</button>
             <div id="loading" style="display:none;">
                 <div class="loader"></div>
     <div id="editor-container">
         <div id="comic-pages"></div>
         <div class="edit-controls">
+            <h4>Editor Tools</h4>
             <button onclick="exportToPng()">💾 Export PNG</button>
+            <button onclick="location.reload()" style="color:red;">↺ Start Over</button>
         </div>
     </div>
     <script>
+        // CLIENT-SIDE SESSION ID GENERATION (Fixes "Session Lost" on Hugging Face)
+        function generateUUID() {
+            return 'xxxxxxxx-xxxx-4xxx-yxxx-xxxxxxxxxxxx'.replace(/[xy]/g, function(c) {
+                var r = Math.random() * 16 | 0, v = c == 'x' ? r : (r & 0x3 | 0x8);
+                return v.toString(16);
+            });
+        }
+        // Get or create SID
+        let sid = localStorage.getItem('comic_sid');
+        if(!sid) {
+            sid = generateUUID();
+            localStorage.setItem('comic_sid', sid);
+        }
+        console.log("Using Session ID:", sid);
         let interval;
         async function upload() {
             const file = document.getElementById('file-upload').files[0];
             if(!file) return alert("Select file");
             document.querySelector('.upload-box').style.display='none';
             document.getElementById('loading').style.display='block';
+            // Pass SID in URL
+            const res = await fetch(`/uploader?sid=${sid}`, {method:'POST', body:fd});
             if(res.ok) interval = setInterval(checkStatus, 2000);
             else { alert("Upload failed"); location.reload(); }
         }
         async function checkStatus() {
+            const res = await fetch(`/status?sid=${sid}`);
             const data = await res.json();
             document.getElementById('status').innerText = data.message;
         }
         function loadComic() {
+            // Fetch pages.json using SID
+            fetch(`/output/pages.json?sid=${sid}`).then(r=>r.json()).then(data => {
                 const c = document.getElementById('comic-pages');
+                c.innerHTML = ''; // clear
                 data.forEach((p, i) => {
                     const div = document.createElement('div');
                     div.className = 'comic-page';
                         const pDiv = document.createElement('div');
                         pDiv.className = 'panel';
                         const img = document.createElement('img');
+                        // Fetch image using SID
+                        img.src = `/frames/${pan.image}?sid=${sid}`;
                         pDiv.appendChild(img);
                         if(p.bubbles && p.bubbles[j] && p.bubbles[j].dialog) {
                             const b = document.createElement('div');
+                            b.className = 'speech-bubble speech';
                             b.innerText = p.bubbles[j].dialog;
                             b.style.left = (p.bubbles[j].bubble_offset_x || 50) + 'px';
                             b.style.top = (p.bubbles[j].bubble_offset_y || 20) + 'px';
                             pDiv.appendChild(b);
                     div.appendChild(grid);
                     c.appendChild(div);
                 });
+            }).catch(e => console.error(e));
         }
         function makeInteractive(el) {
             el.onmousedown = function(e) {
                 e.stopPropagation();
                 const offX = e.clientX - el.offsetLeft;
                 const offY = e.clientY - el.offsetTop;
                 document.onmousemove = function(ev) {
         async function exportToPng() {
             const pages = document.querySelectorAll('.comic-page');
+            for(let i=0; i<pages.length; i++) {
+                try {
+                    const url = await htmlToImage.toPng(pages[i], {pixelRatio: 3});
+                    const a = document.createElement('a');
+                    a.download = `comic-page-${i+1}.png`;
+                    a.href = url;
+                    a.click();
+                } catch(e) { console.error(e); alert("Export failed"); }
             }
         }
     </script>
                 json.dump({'message': message, 'progress': progress}, f)
         except: pass
+    def cleanup(self):
+        # Simple cleanup
+        if os.path.exists(self.frames_dir): shutil.rmtree(self.frames_dir)
+        os.makedirs(self.frames_dir, exist_ok=True)
     def generate_comic(self):
         try:
+            if cv2 is None: raise Exception("OpenCV missing on server.")
             self.update_status("Processing Video...", 10)
             cap = cv2.VideoCapture(self.video_path)
+            if not cap.isOpened(): raise Exception("Invalid Video")
+            self.video_fps = cap.get(cv2.CAP_PROP_FPS) or 25
             cap.release()
+            # 1. Subtitles
             self.update_status("Extracting Dialogue...", 30)
             try:
+                get_real_subtitles(self.video_path) # Creates test1.srt
+                # Move to user dir if created in root
                 if os.path.exists('test1.srt'):
                     shutil.move('test1.srt', os.path.join(self.user_dir, 'subs.srt'))
+            except:
+                # Fallback dummy sub
                 with open(os.path.join(self.user_dir, 'subs.srt'), 'w') as f:
+                    f.write("1\n00:00:01,000 --> 00:00:04,000\nSample Text\n")
+            # 2. Extract Frames
+            self.update_status("Generating Panels...", 50)
             cap = cv2.VideoCapture(self.video_path)
             subs_path = os.path.join(self.user_dir, 'subs.srt')
             with open(subs_path, 'r', encoding='utf-8') as f:
                 subs = list(srt.parse(f.read()))
             frame_files = []
             bubbles = []
+            # Limit frames to avoid timeout
+            limit_subs = subs[:12]
+            for i, sub in enumerate(limit_subs):
+                mid = (sub.start.total_seconds() + sub.end.total_seconds()) / 2
+                cap.set(cv2.CAP_PROP_POS_MSEC, mid * 1000)
                 ret, frame = cap.read()
                 if ret:
                     fname = f"frame_{i}.png"
                     cv2.imwrite(os.path.join(self.frames_dir, fname), frame)
                     frame_files.append(fname)
                     bubbles.append(bubble(dialog=sub.content, bubble_offset_x=50, bubble_offset_y=20))
             cap.release()
+            # 3. Assemble
+            self.update_status("Finalizing...", 80)
             pages_data = []
             for i in range(0, len(frame_files), 4):
+                batch_f = frame_files[i:i+4]
+                batch_b = bubbles[i:i+4]
+                panels = [{'image': f} for f in batch_f]
+                # safe object conversion
+                b_data = [b.__dict__ if hasattr(b, '__dict__') else b for b in batch_b]
                 pages_data.append({'panels': panels, 'bubbles': b_data})
             with open(os.path.join(self.output_dir, 'pages.json'), 'w') as f:
                 json.dump(pages_data, f)
             self.update_status("Done!", 100)
         except Exception as e:
             traceback.print_exc()
+            self.update_status(f"Error: {str(e)}", -1)
+# --- ROUTES (SID REQUIRED IN QUERY PARAMS) ---
 @app.route('/')
 def index():
     return INDEX_HTML
 @app.route('/uploader', methods=['POST'])
 def upload():
+    sid = request.args.get('sid')
+    if not sid: return "Missing SID", 400
     if 'file' not in request.files: return "No file", 400
     gen = EnhancedComicGenerator(sid)
+    gen.cleanup() # Clean previous run
     request.files['file'].save(gen.video_path)
     gen.update_status("Starting...", 5)
     threading.Thread(target=gen.generate_comic).start()
     return jsonify({'success': True})
 @app.route('/status')
 def get_status():
+    sid = request.args.get('sid')
+    if not sid: return jsonify({'message': 'No SID', 'progress': -1})
     status_path = os.path.join(BASE_USER_DIR, sid, 'output', 'status.json')
     if os.path.exists(status_path):
         return send_file(status_path)
+    return jsonify({'message': 'Waiting...', 'progress': 0})
 @app.route('/output/<path:filename>')
 def get_output(filename):
+    sid = request.args.get('sid')
+    if not sid: return "No SID", 400
+    return send_from_directory(os.path.join(BASE_USER_DIR, sid, 'output'), filename)
 @app.route('/frames/<path:filename>')
 def get_frame(filename):
+    sid = request.args.get('sid')
+    if not sid: return "No SID", 400
+    return send_from_directory(os.path.join(BASE_USER_DIR, sid, 'frames'), filename)
 if __name__ == '__main__':
     os.makedirs(BASE_USER_DIR, exist_ok=True)
+    port = int(os.getenv("PORT", 7860))
+    app.run(host='0.0.0.0', port=port)