Update main.py

main.py

@@ -44,6 +44,15 @@ from publisher.bulk import execute as bulk_execute
 from publisher.metadata_engine import generate_metadata
 from publisher.thumbnail_engine import generate_thumbnail
 
+# ==============================
+# SAFETY PATCH: bcrypt/passlib crash shield
+# ==============================
+import passlib
+try:
+    import bcrypt  # noqa
+except Exception as e:
+    logger.warning(f"bcrypt import issue detected (ignored at runtime startup): {e}")
+
 # ==============================
 # INIT
 # ==============================
@@ -52,22 +61,25 @@ os.makedirs(UPLOAD_DIR, exist_ok=True)
 
 
 # ==============================
-# LIFECYCLE (FIXED)
+# LIFECYCLE (HARDENED)
 # ==============================
 @asynccontextmanager
 async def lifespan(app: FastAPI):
-
     logger.info("Starting Basyx Whisper V10.1")
 
-    # DB init (AUTH)
-    Base.metadata.create_all(bind=engine)
+    try:
+        Base.metadata.create_all(bind=engine)
+    except Exception as e:
+        logger.error(f"DB init failed: {e}")
+        raise
 
-    # workers
     start_worker()
     init_scheduler()
 
-    # autonomous engine
-    asyncio.create_task(autonomous_loop())
+    try:
+        asyncio.create_task(autonomous_loop())
+    except Exception as e:
+        logger.error(f"autonomous_loop failed to start: {e}")
 
     yield
 
@@ -79,10 +91,8 @@ app = FastAPI(
     lifespan=lifespan,
 )
 
-# AUTH ROUTES
 app.include_router(auth_router)
 
-
 # ==============================
 # TASK REGISTRY
 # ==============================
@@ -113,19 +123,19 @@ def normalize_task(task: str):
 
 
 # ==============================
-# SAFE INPUT RESOLVER (FIXED CRASH)
+# SAFE INPUT RESOLVER (FIXED)
 # ==============================
 async def safe_resolve(file, source):
-
     try:
         if not file and not source:
             return None
 
-        # 🔴 FIX: resolve_input expects UploadFile OR path, not raw string confusion
+        upload_file = file if isinstance(file, UploadFile) else None
+
         return await asyncio.to_thread(
             resolve_input,
             source,
-            file if isinstance(file, UploadFile) else None,
+            upload_file,
         )
 
     except Exception as e:
@@ -141,57 +151,34 @@ async def execute_task(video_path, task, payload=None, webhook=None):
     payload = payload or {}
     logger.info(f"[TASK] {task}")
 
-    # -------- BULK ----------
     if task == "bulk-publish":
         return await bulk_execute(payload), None
 
-    # -------- INPUT GUARD ----------
     if task not in ["bulk-publish", "schedule-post"] and not video_path:
         return {"error": "No valid input resolved"}, None
 
-    # -------- AUTONOMOUS ----------
     if task == "autonomous":
-        result = await asyncio.to_thread(run_autonomous_engine, video_path)
-        return result, None
+        return await asyncio.to_thread(run_autonomous_engine, video_path), None
 
     if task == "auto-publish":
         auto = await asyncio.to_thread(run_autonomous_engine, video_path)
-        return await dispatch_publish(
-            variants=auto.get("all_variants", [])
-        ), None
+        return await dispatch_publish(variants=auto.get("all_variants", [])), None
 
-    # -------- PUBLISH ----------
     if task == "publish":
-        return await dispatch_publish(
-            video_path=video_path,
-            payload=payload,
-        ), None
+        return await dispatch_publish(video_path=video_path, payload=payload), None
 
-    # -------- METADATA ----------
     if task == "generate-metadata":
         return generate_metadata(video_path), None
 
-    # -------- THUMBNAIL ----------
     if task == "generate-thumbnail":
-
-        output_path = os.path.join(
-            UPLOAD_DIR,
-            f"{uuid.uuid4()}.jpg"
-        )
-
-        thumb = generate_thumbnail(
-            video_path,
-            output=output_path,
-        )
-
+        output_path = os.path.join(UPLOAD_DIR, f"{uuid.uuid4()}.jpg")
+        thumb = generate_thumbnail(video_path, output=output_path)
         return {"thumbnail": thumb}, output_path
 
-    # -------- QUEUE ----------
     if task == "batch":
         job_id = create_job(video_path, webhook=webhook)
         return {"status": "queued", "job_id": job_id}, None
 
-    # -------- TRANSCRIBE ----------
     if task == "transcribe":
         words = await asyncio.to_thread(transcribe_video, video_path)
         return {"words": words}, None
@@ -200,56 +187,34 @@ async def execute_task(video_path, task, payload=None, webhook=None):
         words = await asyncio.to_thread(transcribe_video, video_path)
         return {"srt": generate_srt(words)}, None
 
-    # -------- RENDER ----------
     if task == "render":
-
         words = await asyncio.to_thread(transcribe_video, video_path)
         srt = generate_srt(words)
 
-        output = os.path.join(
-            UPLOAD_DIR,
-            f"{uuid.uuid4()}_render.mp4",
-        )
+        output = os.path.join(UPLOAD_DIR, f"{uuid.uuid4()}_render.mp4")
 
-        await asyncio.to_thread(
-            render_subtitles,
-            video_path,
-            srt,
-            output,
-        )
+        await asyncio.to_thread(render_subtitles, video_path, srt, output)
 
         return {"status": "render_complete"}, output
 
-    # -------- HIGHLIGHTS ----------
     if task == "highlights":
-
         words = await asyncio.to_thread(transcribe_video, video_path)
         highlights = detect_highlights(words) or []
-
         clips = create_clips(video_path, highlights)
 
         return {"clips_created": len(clips)}, (clips[0] if clips else None)
 
     if task == "clips":
-
         words = await asyncio.to_thread(transcribe_video, video_path)
         highlights = detect_highlights(words) or []
-
         return {"clips": create_clips(video_path, highlights)}, None
 
-    # -------- VIRAL SCORE ----------
     if task == "viral-score":
-
         words = await asyncio.to_thread(transcribe_video, video_path)
         segments = detect_highlights(words) or []
+        return {"scores": [score_clip(s) for s in segments]}, None
 
-        return {
-            "scores": [score_clip(s) for s in segments]
-        }, None
-
-    # -------- STRATEGY ----------
     if task == "strategy":
-
         words = await asyncio.to_thread(transcribe_video, video_path)
 
         script = rewrite_script(words)
@@ -283,9 +248,7 @@ async def execute_router(
 ):
 
     try:
-
         task = normalize_task(task_name)
-
         payload = {}
 
         if request.headers.get("content-type", "").startswith("application/json"):
@@ -293,12 +256,7 @@ async def execute_router(
 
         video_path = await safe_resolve(file, url_input or source)
 
-        result, output = await execute_task(
-            video_path,
-            task,
-            payload,
-            webhook,
-        )
+        result, output = await execute_task(video_path, task, payload, webhook)
 
         if output and isinstance(output, str) and os.path.exists(output):
             return FileResponse(output)
@@ -329,7 +287,6 @@ def status(job_id: str):
 async def ui_handler(video, task, webhook, url_input):
 
     source = url_input or video
-
     video_path = await safe_resolve(video, source)
 
     result, output = await execute_task(
@@ -343,7 +300,6 @@ async def ui_handler(video, task, webhook, url_input):
 
 
 with gr.Blocks() as demo:
-
     gr.Markdown("# 🚀 Basyx Whisper V10.1 Stable Operator")
 
     video_input = gr.Video()