Update utils/job_queue.py

utils/job_queue.py

@@ -9,6 +9,8 @@ from .validators import validate_video
 from .transcription import transcribe_video
 from .highlights import detect_highlights
 from .viral_scorer import score_clip
+from .retention import predict_retention
+from .caption_seo import generate_caption
 from .clipper import create_clip
 
 
@@ -23,7 +25,7 @@ WORKER_STARTED = False
 
 
 # =====================================================
-# JOB CREATION (VIRAL MODE)
+# CREATE STRATEGY JOB
 # =====================================================
 
 def create_job(video_path: str, webhook: str | None = None):
@@ -35,9 +37,16 @@ def create_job(video_path: str, webhook: str | None = None):
         "status": "queued",
         "stage": "waiting",
         "progress": 0,
+
         "video": video_path,
+
+        # V6 INTELLIGENCE OUTPUTS
         "clips": [],
         "scores": [],
+        "retention_scores": [],
+        "caption": None,
+        "strategy_summary": None,
+
         "webhook": webhook,
         "error": None,
         "created_at": time.time(),
@@ -45,13 +54,13 @@ def create_job(video_path: str, webhook: str | None = None):
 
     queue.put(job_id)
 
-    logger.info(f"[V5] Viral job queued: {job_id}")
+    logger.info(f"[V6] Strategy job queued: {job_id}")
 
     return job_id
 
 
 # =====================================================
-# JOB ACCESS
+# GET JOB
 # =====================================================
 
 def get_job(job_id: str):
@@ -59,7 +68,7 @@ def get_job(job_id: str):
 
 
 # =====================================================
-# UPDATE HELPER
+# UPDATE HELPERS
 # =====================================================
 
 def update(job_id, **kwargs):
@@ -68,12 +77,12 @@ def update(job_id, **kwargs):
 
 
 # =====================================================
-# WORKER (V5 VIRAL ENGINE)
+# WORKER (V6 STRATEGIC ENGINE)
 # =====================================================
 
 def worker():
 
-    logger.info("[V5] Viral engine worker started")
+    logger.info("[V6] Viral AI Strategist worker started")
 
     while True:
 
@@ -95,60 +104,82 @@ def worker():
             words = transcribe_video(job["video"])
 
             # -----------------------------
-            # HIGHLIGHTS DETECTION
+            # HIGHLIGHT DETECTION
             # -----------------------------
-            update(job_id, stage="detecting highlights", progress=40)
-            highlights = detect_highlights(words)
+            update(job_id, stage="detecting moments", progress=35)
+            segments = detect_highlights(words)
+
+            clips = []
+            scores = []
+            retention_scores = []
 
             # -----------------------------
-            # VIRAL SCORING ENGINE (CORE V5)
+            # STRATEGIC SCORING ENGINE
             # -----------------------------
-            update(job_id, stage="scoring virality", progress=55)
+            update(job_id, stage="strategic analysis", progress=50)
 
-            scored_clips = []
+            for i, seg in enumerate(segments):
 
-            for i, segment in enumerate(highlights):
+                virality = score_clip(seg)
+                retention = predict_retention(seg)
 
-                score = score_clip(segment)
+                final_score = (virality * 0.6) + (retention * 0.4)
 
-                # FILTER LOW VIRALITY
-                if score < 45:
+                if final_score < 50:
                     continue
 
                 clip_path = create_clip(
                     job["video"],
-                    segment[0]["start"],
-                    segment[-1]["end"],
+                    seg[0]["start"],
+                    seg[-1]["end"],
                     i
                 )
 
-                scored_clips.append({
-                    "path": clip_path,
-                    "score": score
-                })
+                clips.append(clip_path)
+                scores.append(final_score)
+                retention_scores.append(retention)
 
                 update(job_id,
-                       progress=55 + int((i / max(len(highlights),1)) * 35))
+                       progress=50 + int((i / max(len(segments),1)) * 40))
+
+            # -----------------------------
+            # RANKING (STRATEGY CORE)
+            # -----------------------------
+            ranked = sorted(
+                zip(clips, scores, retention_scores),
+                key=lambda x: x[1],
+                reverse=True
+            )
+
+            top_clips = [c[0] for c in ranked[:5]]
+            top_scores = [c[1] for c in ranked[:5]]
+            top_retention = [c[2] for c in ranked[:5]]
 
             # -----------------------------
-            # SORT BY VIRALITY
+            # CAPTION STRATEGY
             # -----------------------------
-            scored_clips.sort(key=lambda x: x["score"], reverse=True)
+            caption = generate_caption(words)
 
-            top_clips = [c["path"] for c in scored_clips[:5]]
-            top_scores = [c["score"] for c in scored_clips[:5]]
+            strategy_summary = {
+                "total_segments": len(segments),
+                "kept_clips": len(top_clips),
+                "avg_score": sum(top_scores) / max(len(top_scores), 1)
+            }
 
             # -----------------------------
-            # COMPLETE
+            # FINALIZE
             # -----------------------------
             update(job_id,
                    status="completed",
-                   stage="viral selection done",
+                   stage="strategy finalized",
                    progress=100,
                    clips=top_clips,
-                   scores=top_scores)
+                   scores=top_scores,
+                   retention_scores=top_retention,
+                   caption=caption,
+                   strategy_summary=strategy_summary)
 
-            logger.info(f"[V5] Viral job complete: {job_id}")
+            logger.info(f"[V6] Strategy complete: {job_id}")
 
         except Exception as e:
 
@@ -164,7 +195,7 @@ def worker():
 
 
 # =====================================================
-# START WORKER (SAFE SINGLETON)
+# START WORKER
 # =====================================================
 
 def start_worker():
@@ -179,4 +210,4 @@ def start_worker():
     t = threading.Thread(target=worker, daemon=True)
     t.start()
 
-    logger.info("[V5] Worker initialized")
+    logger.info("[V6] Worker initialized")