Update pipeline/pipeline.py

pipeline/pipeline.py

@@ -1,9 +1,5 @@
-# pipeline.py
 import asyncio
 import logging
-from api.server import pending_confirmations  # Access the confirmation events
-
-# Import your modules
 import core.script_gen as script_gen
 import core.story_script as story_script
 import core.image_generator as image_gen
@@ -16,35 +12,48 @@ logging.basicConfig(
     format="%(asctime)s [%(levelname)s] %(message)s"
 )
 
-async def run_pipeline(task: dict):
-    task_id = task["task_id"]
+async def run_pipeline(task: dict, confirmation_event: asyncio.Event):
+    """
+    Executes the full workflow:
+    1. Script generation
+    2. Wait for user confirmation
+    3. Story generation
+    4. Image generation
+    Video/music/assemble placeholders for now
+    """
+
+    task_id = task["id"]
     idea = task["idea"]
 
     logging.info(f"[Pipeline] Starting script generation for task {task_id}")
-    script = await script_gen.generate_script(idea)  # Async script generation
-
-    logging.info(f"[Pipeline] Waiting for user confirmation for task {task_id}")
-    # Wait for confirmation (manual or auto)
-    if task_id in pending_confirmations:
-        await pending_confirmations[task_id].wait()
-    else:
-        logging.info(f"[Pipeline] No pending confirmation found, auto-confirming task {task_id}")
-
-    logging.info(f"[Pipeline] Generating story script for task {task_id}")
+    script = await script_gen.generate_script(idea)
+    task["result"]["script"] = script
+    task["status"] = "waiting_for_confirmation"
+    task["confirmation_required"] = True
+    logging.info(f"[Pipeline] Script ready for task {task_id}, waiting confirmation...")
+
+    # Wait for user confirmation
+    await confirmation_event.wait()
+    task["status"] = "confirmed"
+    task["confirmation_required"] = False
+    logging.info(f"[Pipeline] Task {task_id} confirmed. Continuing pipeline...")
+
+    # Story generation
+    logging.info(f"[Pipeline] Generating story for task {task_id}")
     story = await story_script.generate_story(script)
-    # final_output = story  # Placeholder for final output
+    task["result"]["story_script"] = story
+
+    # Image generation
     logging.info(f"[Pipeline] Generating images for task {task_id}")
     images = await image_gen.generate_images(story)
-    final_output=images
-
-    # logging.info(f"[Pipeline] Generating video for task {task_id}")
-    # video = await video_gen.generate_video(images)
-
-    # logging.info(f"[Pipeline] Generating music/audio for task {task_id}")
-    # audio = await music_gen.generate_music(story)
+    task["result"]["images"] = images
 
-    # logging.info(f"[Pipeline] Assembling final output for task {task_id}")
-    # final_output = await assemble.create_final(video, audio)
+    # Placeholder for future stages
+    # logging.info(f"[Pipeline] Generating video/music/assembling for task {task_id}")
+    # task["result"]["video"] = None
+    # task["result"]["music"] = None
+    # task["result"]["final_output"] = None
 
-    logging.info(f"[Pipeline] Task {task_id} completed. Output: {final_output}")
-    return final_output
+    task["status"] = "completed"
+    logging.info(f"[Pipeline] Task {task_id} completed. Output: {images}")
+    return task["result"]