Update main.py

main.py

@@ -8,6 +8,12 @@ import gradio as gr
 from utils.logger import logger
 from utils.job_queue import start_worker, create_job, get_job
 
+from ingestion.resolver import resolve_input
+
+# ==============================
+# CORE PIPELINE
+# ==============================
+
 from utils.transcription import transcribe_video
 from utils.srt import generate_srt
 from utils.render import render_subtitles
@@ -19,51 +25,54 @@ from utils.platform import adapt_platform
 from utils.persona import predict_audience
 from utils.clipper import create_clips
 
-from ingestion.resolver import resolve_input
 from utils.autonomous_engine import run_autonomous_engine
 
-# ---------- Publisher ----------
-from publisher.publisher_scheduler_ai import autonomous_publish
-from publisher.metadata import generate_metadata
-from publisher.hashtags import generate_hashtags
-from publisher.thumbnail import generate_thumbnail
-from publisher.scheduler import schedule_post
-from publisher.bulk import execute as bulk_execute
 
-from publisher.platforms.tiktok import publish_tiktok
-from publisher.platforms.reels import publish_reels
-from publisher.platforms.shorts import publish_shorts
-from publisher.platforms.facebook import publish_facebook
+# ==============================
+# ✅ PUBLISHER V10
+# ==============================
+
+from publisher.publisher_ai import autonomous_loop
+from publisher.scheduler_engine import init_scheduler
+from publisher.platform_dispatcher import dispatch_publish
+from publisher.bulk import execute as bulk_execute
+from publisher.metadata_engine import generate_metadata
+from publisher.thumbnail_engine import generate_thumbnail
 
 
-# =====================================================
-# INIT
-# =====================================================
+# ==============================
+# INIT APP
+# ==============================
 
-app = FastAPI(title="Basyx Whisper V9 Autonomous Content Operator")
+app = FastAPI(title="Basyx Whisper V10 Autonomous Content Operator")
 
 UPLOAD_DIR = "jobs"
 os.makedirs(UPLOAD_DIR, exist_ok=True)
 
 start_worker()
 
+# ---- Start Autonomous Systems ----
+init_scheduler()
+
 
-# =====================================================
+@app.on_event("startup")
+async def startup_event():
+    logger.info("Starting Autonomous Publisher Brain V10")
+    asyncio.create_task(autonomous_loop())
+
+
+# ==============================
 # TASK NORMALIZER
-# =====================================================
+# ==============================
 
 VALID_TASKS = {
     "autonomous",
     "auto-publish",
-    "publish-tiktok",
-    "publish-reels",
-    "publish-shorts",
-    "publish-facebook",
+    "publish",
+    "bulk-publish",
     "generate-metadata",
-    "generate-hashtags",
     "generate-thumbnail",
     "schedule-post",
-    "bulk-publish",
     "transcribe",
     "subtitles",
     "render",
@@ -73,6 +82,7 @@ VALID_TASKS = {
     "batch",
 }
 
+
 def normalize_task(task: str):
     task = task.lower().replace("_", "-")
     if task not in VALID_TASKS:
@@ -80,78 +90,63 @@ def normalize_task(task: str):
     return task
 
 
-# =====================================================
+# ==============================
 # EXECUTION ENGINE
-# =====================================================
+# ==============================
 
 async def execute_task(
     video_path: str | None,
     task: str,
     payload: dict | None = None,
-    webhook=None
+    webhook=None,
 ):
 
+    payload = payload or {}
+
     logger.info(f"[TASK] {task}")
 
-    # ---------- BULK ----------
+    # ---------------- BULK ----------------
     if task == "bulk-publish":
-        return await bulk_execute(payload or {}), None
+        return await bulk_execute(payload), None
 
-    # ---------- AUTONOMOUS ----------
+    # ---------------- AUTONOMOUS ----------------
     if task == "autonomous":
         result = await asyncio.to_thread(
             run_autonomous_engine,
-            video_path
+            video_path,
         )
-        best = result.get("best_clip")
-        return result, best.get("clip") if best else None
+        return result, None
 
     if task == "auto-publish":
+
         autonomous = await asyncio.to_thread(
             run_autonomous_engine,
-            video_path
+            video_path,
         )
 
-        publish_result = await autonomous_publish(
+        publish_result = await dispatch_publish(
             variants=autonomous["all_variants"]
         )
 
         return publish_result, None
 
-    # ---------- METADATA ----------
+    # ---------------- PUBLISH ----------------
+    if task == "publish":
+        result = await dispatch_publish(
+            video_path=video_path,
+            payload=payload,
+        )
+        return result, None
+
+    # ---------------- METADATA ----------------
     if task == "generate-metadata":
         return generate_metadata(video_path), None
 
-    if task == "generate-hashtags":
-        return generate_hashtags(video_path), None
-
     if task == "generate-thumbnail":
         thumb = generate_thumbnail(video_path)
         return {"thumbnail": thumb}, thumb
 
-    # ---------- PLATFORM PUBLISH ----------
-    publish_payload = {
-        "video_path": video_path,
-        **(payload or {})
-    }
-
-    if task == "publish-tiktok":
-        return await publish_tiktok(publish_payload), None
-
-    if task == "publish-reels":
-        return await publish_reels(publish_payload), None
-
-    if task == "publish-shorts":
-        return await publish_shorts(publish_payload), None
-
-    if task == "publish-facebook":
-        return await publish_facebook(publish_payload), None
-
-    # ---------- SCHEDULER ----------
-    if task == "schedule-post":
-        return schedule_post(payload or {}), None
-
-    # ---------- PIPELINE ----------
+    # ---------------- PIPELINE ----------------
     if task == "batch":
         job_id = create_job(video_path, webhook=webhook)
         return {"status": "queued", "job_id": job_id}, None
@@ -165,6 +160,7 @@ async def execute_task(
         return {"srt": generate_srt(words)}, None
 
     if task == "render":
+
         words = await asyncio.to_thread(transcribe_video, video_path)
         srt = generate_srt(words)
 
@@ -183,18 +179,23 @@ async def execute_task(
         return {"status": "render_complete"}, output
 
     if task == "highlights":
+
         words = await asyncio.to_thread(transcribe_video, video_path)
         highlights = detect_highlights(words)
         clips = create_clips(video_path, highlights)
+
         return {"clips_created": len(clips)}, clips[0]
 
     if task == "viral-score":
+
         words = await asyncio.to_thread(transcribe_video, video_path)
         segments = detect_highlights(words) or []
         scores = [score_clip(s) for s in segments]
+
         return {"scores": scores}, None
 
     if task == "strategy":
+
         words = await asyncio.to_thread(transcribe_video, video_path)
 
         script = rewrite_script(words)
@@ -214,9 +215,9 @@ async def execute_task(
     raise Exception("Task execution failed")
 
 
-# =====================================================
+# ==============================
 # UNIVERSAL EXECUTE ROUTER
-# =====================================================
+# ==============================
 
 @app.post("/execute/{task_name}")
 async def execute_router(
@@ -237,10 +238,10 @@ async def execute_router(
         if request.headers.get("content-type", "").startswith("application/json"):
             payload = await request.json()
 
-        # ---- Resolve video only if needed ----
+        # ---- UNIVERSAL INPUT LAYER ----
         video_path = None
 
-        if task != "bulk-publish" and (
+        if task not in ["bulk-publish"] and (
             file or url_input or source
         ):
             input_source = url_input or source
@@ -268,13 +269,13 @@ async def execute_router(
         return JSONResponse({"error": str(e)}, status_code=500)
 
 
-# =====================================================
+# ==============================
 # HEALTH
-# =====================================================
+# ==============================
 
 @app.get("/api/health")
 def api_health():
-    return {"status": "online", "version": "V9.3"}
+    return {"status": "online", "version": "V10"}
 
 
 @app.get("/api/status/{job_id}")
@@ -282,9 +283,9 @@ def api_status(job_id: str):
     return get_job(job_id) or {"error": "Job not found"}
 
 
-# =====================================================
+# ==============================
 # GRADIO UI
-# =====================================================
+# ==============================
 
 async def ui_handler(video, task, webhook, url_input):
 
@@ -310,7 +311,7 @@ async def ui_handler(video, task, webhook, url_input):
 
 with gr.Blocks() as demo:
 
-    gr.Markdown("# Basyx Whisper V9 Autonomous Operator")
+    gr.Markdown("# Basyx Whisper V10 Autonomous Operator")
 
     video_input = gr.Video()
     url_input = gr.Textbox(label="Video URL")