Update main.py

main.py

@@ -14,7 +14,7 @@ from utils.job_queue import start_worker, create_job, get_job
 from ingestion.resolver import resolve_input
 
 # ==============================
-# AUTH SYSTEM ✅ NEW
+# AUTH SYSTEM
 # ==============================
 from auth.routes import router as auth_router
 from auth.database import Base, engine
@@ -45,22 +45,28 @@ from publisher.metadata_engine import generate_metadata
 from publisher.thumbnail_engine import generate_thumbnail
 
 # ==============================
-# INIT (LIFESPAN SAFE)
+# INIT
 # ==============================
 UPLOAD_DIR = "jobs"
 os.makedirs(UPLOAD_DIR, exist_ok=True)
 
 
+# ==============================
+# LIFECYCLE (FIXED)
+# ==============================
 @asynccontextmanager
 async def lifespan(app: FastAPI):
 
     logger.info("Starting Basyx Whisper V10.1")
 
+    # DB init (AUTH)
     Base.metadata.create_all(bind=engine)
 
+    # workers
     start_worker()
     init_scheduler()
 
+    # autonomous engine
     asyncio.create_task(autonomous_loop())
 
     yield
@@ -73,9 +79,10 @@ app = FastAPI(
     lifespan=lifespan,
 )
 
-# Register Auth Routes
+# AUTH ROUTES
 app.include_router(auth_router)
 
+
 # ==============================
 # TASK REGISTRY
 # ==============================
@@ -106,7 +113,7 @@ def normalize_task(task: str):
 
 
 # ==============================
-# SAFE INPUT RESOLVER
+# SAFE INPUT RESOLVER (FIXED CRASH)
 # ==============================
 async def safe_resolve(file, source):
 
@@ -114,10 +121,11 @@ async def safe_resolve(file, source):
         if not file and not source:
             return None
 
+        # 🔴 FIX: resolve_input expects UploadFile OR path, not raw string confusion
         return await asyncio.to_thread(
             resolve_input,
             source,
-            file,
+            file if isinstance(file, UploadFile) else None,
         )
 
     except Exception as e:
@@ -137,20 +145,17 @@ async def execute_task(video_path, task, payload=None, webhook=None):
     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
-        )
+        result = await asyncio.to_thread(run_autonomous_engine, video_path)
         return result, None
 
     if task == "auto-publish":
-        auto = await asyncio.to_thread(
-            run_autonomous_engine, video_path
-        )
+        auto = await asyncio.to_thread(run_autonomous_engine, video_path)
         return await dispatch_publish(
             variants=auto.get("all_variants", [])
         ), None
@@ -166,7 +171,7 @@ async def execute_task(video_path, task, payload=None, webhook=None):
     if task == "generate-metadata":
         return generate_metadata(video_path), None
 
-    # ✅ THUMBNAIL FIX
+    # -------- THUMBNAIL ----------
     if task == "generate-thumbnail":
 
         output_path = os.path.join(
@@ -188,27 +193,17 @@ async def execute_task(video_path, task, payload=None, webhook=None):
 
     # -------- TRANSCRIBE ----------
     if task == "transcribe":
-        words = await asyncio.to_thread(
-            transcribe_video,
-            video_path,
-        )
+        words = await asyncio.to_thread(transcribe_video, video_path)
         return {"words": words}, None
 
     if task == "subtitles":
-        words = await asyncio.to_thread(
-            transcribe_video,
-            video_path,
-        )
+        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,
-        )
-
+        words = await asyncio.to_thread(transcribe_video, video_path)
         srt = generate_srt(words)
 
         output = os.path.join(
@@ -228,40 +223,24 @@ async def execute_task(video_path, task, payload=None, webhook=None):
     # -------- HIGHLIGHTS ----------
     if task == "highlights":
 
-        words = await asyncio.to_thread(
-            transcribe_video,
-            video_path,
-        )
-
+        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)
+        return {"clips_created": len(clips)}, (clips[0] if clips else None)
 
     if task == "clips":
 
-        words = await asyncio.to_thread(
-            transcribe_video,
-            video_path,
-        )
-
+        words = await asyncio.to_thread(transcribe_video, video_path)
         highlights = detect_highlights(words) or []
 
-        return {
-            "clips": create_clips(video_path, highlights)
-        }, None
+        return {"clips": create_clips(video_path, highlights)}, None
 
-    # -------- SCORING ----------
+    # -------- VIRAL SCORE ----------
     if task == "viral-score":
 
-        words = await asyncio.to_thread(
-            transcribe_video,
-            video_path,
-        )
-
+        words = await asyncio.to_thread(transcribe_video, video_path)
         segments = detect_highlights(words) or []
 
         return {
@@ -271,10 +250,7 @@ async def execute_task(video_path, task, payload=None, webhook=None):
     # -------- STRATEGY ----------
     if task == "strategy":
 
-        words = await asyncio.to_thread(
-            transcribe_video,
-            video_path,
-        )
+        words = await asyncio.to_thread(transcribe_video, video_path)
 
         script = rewrite_script(words)
         persona = predict_audience(words)
@@ -294,7 +270,7 @@ async def execute_task(video_path, task, payload=None, webhook=None):
 
 
 # ==============================
-# EXECUTE ROUTER
+# ROUTER
 # ==============================
 @app.post("/execute/{task_name}")
 async def execute_router(
@@ -312,15 +288,10 @@ async def execute_router(
 
         payload = {}
 
-        if request.headers.get(
-            "content-type", ""
-        ).startswith("application/json"):
+        if request.headers.get("content-type", "").startswith("application/json"):
             payload = await request.json()
 
-        video_path = await safe_resolve(
-            file,
-            url_input or source,
-        )
+        video_path = await safe_resolve(file, url_input or source)
 
         result, output = await execute_task(
             video_path,
@@ -329,17 +300,14 @@ async def execute_router(
             webhook,
         )
 
-        if output and os.path.exists(output):
+        if output and isinstance(output, str) and os.path.exists(output):
             return FileResponse(output)
 
         return {"task": task, "result": result}
 
     except Exception as e:
         logger.exception(e)
-        return JSONResponse(
-            {"error": str(e)},
-            status_code=500,
-        )
+        return JSONResponse({"error": str(e)}, status_code=500)
 
 
 # ==============================