Stories

import os
import time
from camb.client import CambAI
from camb.types.language_enums import Languages

client = CambAI(api_key=os.getenv("CAMB_API_KEY"))

def narrate_story():
    # Replace with your story file (.txt or .docx)
    story_path = "story.txt"

    # Step 1: Submit the story job (multipart upload)
    response = client.story.create_story(
        file=open(story_path, "rb"),
        source_language=Languages.EN_US,
        title="My Story",
        description="Sample narrated story",
        narrator_voice_id=147320,
    )

    # Usually returns task_id for async processing (see Note below)
    task_id = getattr(response, "task_id", None)
    if not task_id:
        print("No task_id on response; if you received run_id directly, skip polling and call get_story_run_info.")
        return

    print(f"Story task created: {task_id}")

    # Step 2: Poll until complete
    max_attempts = 60
    attempt = 0
    while attempt < max_attempts:
        status = client.story.get_story_status(task_id=task_id)
        print(f"Status: {status.status}")

        if status.status == "SUCCESS":
            # Step 3: Full narration, dialogue track, optional transcript
            result = client.story.get_story_run_info(
                status.run_id,
                include_transcript=True,
            )
            print(f"Audio URL: {result.get('audio_url')}")
            print(f"Dialogue URL: {result.get('dialogue_url')}")
            transcript = result.get("transcript")
            if transcript:
                for row in transcript[:2]:
                    print(f"[{row.get('start')}-{row.get('end')}ms] {row.get('speaker')}: {row.get('text')}")
            break
        elif status.status == "ERROR":
            print(f"Story failed: {status.exception_reason}")
            break

        time.sleep(5)
        attempt += 1

narrate_story()