fix: ensure output directory exists before saving featured images

app.py

@@ -695,6 +695,16 @@ def _start_agent():
 print("[STARTUP] Initializing database...")
 init_db()
 
+# Ensure working directories exist (downloads, output, uploads)
+from pathlib import Path
+_DATA_DIR = Path("/data") if Path("/data").is_dir() or os.getenv("HUGGING_FACE_SPACES") else Path(__file__).parent
+if _DATA_DIR == Path("/data"):
+    _DATA_DIR.mkdir(parents=True, exist_ok=True)
+(_DATA_DIR / "downloads").mkdir(exist_ok=True)
+(_DATA_DIR / "output").mkdir(exist_ok=True)
+(_DATA_DIR / "uploads").mkdir(exist_ok=True)
+print(f"[STARTUP] Working directories ensured: {_DATA_DIR}")
+
 # Recover any jobs stuck in 'running' from a previous crash/deploy
 _conn = get_db()
 _recovered = _conn.execute("UPDATE jobs SET status = 'pending', started_at = NULL WHERE status = 'running'").rowcount

article_rewriter.py

@@ -1479,6 +1479,9 @@ def create_article_featured_image(
 
     canvas = _collage_3_to_1(padded, WIDTH, HEIGHT)
 
+    # Ensure output directory exists (defensive)
+    OUTPUT_DIR.mkdir(parents=True, exist_ok=True)
+    
     output_path = OUTPUT_DIR / f"featured_article_{output_name}.jpg"
     canvas.save(str(output_path), "JPEG", quality=92)
     print(f"   ✅ Featured image saved: {output_path}")

main.py

@@ -1161,6 +1161,9 @@ def create_featured_image(title: str, frames: list[Image.Image], output_name: st
     draw_top = ImageDraw.Draw(canvas)
     draw_top.rectangle([(0, 0), (WIDTH, 5)], fill=(255, 69, 58))
 
+    # Ensure output directory exists (defensive)
+    OUTPUT_DIR.mkdir(parents=True, exist_ok=True)
+    
     output_path = OUTPUT_DIR / f"featured_{output_name}.jpg"
     canvas.save(str(output_path), "JPEG", quality=92)
     print(f"   [OK] Featured image saved: {output_path}")
@@ -1264,8 +1267,9 @@ def create_flyer(title: str, frames: list[Image.Image], output_name: str, catego
             frame = frame.resize((frame_w, frame_h), Image.LANCZOS)
             canvas.paste(frame, (i * frame_w, 0))
 
-
-
+    # Ensure output directory exists (defensive)
+    OUTPUT_DIR.mkdir(parents=True, exist_ok=True)
+    
     output_path = OUTPUT_DIR / f"flyer_{output_name}.jpg"
     canvas.save(str(output_path), "JPEG", quality=92)
     print(f"   [OK] Square flyer saved: {output_path}")