Update main.py

main.py

@@ -6,25 +6,27 @@ import uuid
 import gradio as gr
 
 from utils.logger import logger
-from utils.job_queue import start_worker, create_job, get_job
-
+from utils.job_queue import (
+    start_worker,
+    create_job,
+    get_job
+)
 
 # =====================================================
-# INIT APP
+# INIT
 # =====================================================
 
-app = FastAPI(title="Fast Whisper Production V3")
+app = FastAPI(title="Fast Whisper Production V4")
 
 UPLOAD_DIR = "jobs"
 os.makedirs(UPLOAD_DIR, exist_ok=True)
 
-
-# Start background worker ONCE
+# Start V4 worker
 start_worker()
 
 
 # =====================================================
-# API: START RENDER JOB
+# API: START BATCH RENDER
 # =====================================================
 
 @app.post("/render")
@@ -34,24 +36,26 @@ async def render(
 ):
 
     job_id = str(uuid.uuid4())
-    input_path = f"{UPLOAD_DIR}/{job_id}_{file.filename}"
+    path = f"{UPLOAD_DIR}/{job_id}_{file.filename}"
 
-    with open(input_path, "wb") as buffer:
-        shutil.copyfileobj(file.file, buffer)
+    with open(path, "wb") as f:
+        shutil.copyfileobj(file.file, f)
 
-    logger.info(f"Incoming render request: {job_id}")
+    logger.info(f"[V4] New batch job: {job_id}")
 
-    job_id = create_job(input_path, webhook)
+    job_id = create_job(path, webhook)
 
     return {
         "job_id": job_id,
+        "mode": "batch",
+        "status": "queued",
         "status_url": f"/status/{job_id}",
         "download_url": f"/download/{job_id}"
     }
 
 
 # =====================================================
-# API: CHECK STATUS
+# API: STATUS (NOW SUPPORTS CLIPS)
 # =====================================================
 
 @app.get("/status/{job_id}")
@@ -67,12 +71,13 @@ def status(job_id: str):
         "status": job["status"],
         "stage": job.get("stage"),
         "progress": job.get("progress"),
+        "clips": job.get("clips", []),
         "error": job.get("error"),
     }
 
 
 # =====================================================
-# API: DOWNLOAD RESULT
+# API: DOWNLOAD (FIRST CLIP OR FINAL OUTPUT)
 # =====================================================
 
 @app.get("/download/{job_id}")
@@ -84,20 +89,23 @@ def download(job_id: str):
         return {"error": "Job not found"}
 
     if job["status"] != "completed":
-        return {
-            "error": "Job not completed",
-            "status": job["status"]
-        }
+        return {"error": "Job not completed"}
+
+    clips = job.get("clips", [])
+
+    if not clips:
+        return {"error": "No clips generated"}
 
+    # return first clip for simplicity
     return FileResponse(
-        job["output"],
+        clips[0],
         media_type="video/mp4",
-        filename="rendered.mp4"
+        filename="clip_1.mp4"
     )
 
 
 # =====================================================
-# GRADIO BACKEND (UI)
+# GRADIO UI
 # =====================================================
 
 def ui_upload(video, webhook):
@@ -105,14 +113,9 @@ def ui_upload(video, webhook):
     if video is None:
         return "No video uploaded", None
 
-    job_id = str(uuid.uuid4())
-    path = f"{UPLOAD_DIR}/{job_id}.mp4"
-
-    shutil.copy(video, path)
-
-    jid = create_job(path, webhook)
+    job_id = create_job(video, webhook)
 
-    return f"Job Started: {jid}", jid
+    return f"Batch Job Started: {job_id}", job_id
 
 
 def ui_status(job_id):
@@ -122,49 +125,34 @@ def ui_status(job_id):
     if not job:
         return "Job not found", None
 
-    status = job["status"]
+    if job["status"] == "completed":
+        clips = job.get("clips", [])
+        return f"Completed ({len(clips)} clips)", clips[0] if clips else None
 
-    if status == "completed":
-        return "✅ Completed", job["output"]
+    if job["status"] == "failed":
+        return f"Failed: {job.get('error')}", None
 
-    if status == "failed":
-        return f"❌ Failed: {job.get('error')}", None
+    return f"{job['stage']} ({job.get('progress', 0)}%)", None
 
-    return f"Processing... {job.get('progress', 0)}%", None
 
+with gr.Blocks(title="Fast Whisper V4 Batch Engine") as demo:
 
-# =====================================================
-# GRADIO UI LAYOUT
-# =====================================================
-
-with gr.Blocks(title="Fast Whisper V3 Renderer") as demo:
+    gr.Markdown("# 🎬 Fast Whisper Production V4 (Batch Engine)")
 
-    gr.Markdown("# 🎬 Fast Whisper Production V3")
+    video = gr.Video(label="Upload Video")
+    webhook = gr.Textbox(label="Webhook URL (optional)")
 
-    with gr.Row():
-        video = gr.Video(label="Upload Video")
-        webhook = gr.Textbox(label="Webhook URL (optional)")
-
-    start_btn = gr.Button("Start Render")
+    start = gr.Button("Start Batch Render")
 
     job_id_box = gr.Textbox(label="Job ID")
-    status_btn = gr.Button("Check Status")
 
-    status_out = gr.Textbox(label="Status")
-    video_out = gr.Video(label="Output")
+    check = gr.Button("Check Status")
 
-    start_btn.click(
-        ui_upload,
-        inputs=[video, webhook],
-        outputs=[status_out, job_id_box],
-    )
+    status = gr.Textbox(label="Status")
+    output = gr.Video(label="First Clip Preview")
 
-    status_btn.click(
-        ui_status,
-        inputs=job_id_box,
-        outputs=[status_out, video_out],
-    )
+    start.click(ui_upload, inputs=[video, webhook], outputs=[status, job_id_box])
+    check.click(ui_status, inputs=job_id_box, outputs=[status, output])
 
 
-# Mount Gradio inside FastAPI root
 app = gr.mount_gradio_app(app, demo, path="/")