Get Translated Story Result

import requests
import json

# Your API authentication
headers = {
    "x-api-key": "your-api-key",  # Replace with your actual API key
    "Content-Type": "application/json"
}

def get_translation_result(run_id, target_language, include_transcript=False):
    """
    Retrieves the complete results of a translated story including audio and text assets.

    Parameters:
    - run_id: The ID of the original story that was translated
    - target_language: The language code of the target language used for translation
    - include_transcript: Boolean indicating whether to include the detailed transcript (default: False)

    Returns:
    - A dictionary containing all available translation assets
    """
    try:
        # Build the request URL with optional query parameter
        base_url = f"https://client.camb.ai/apis/translated-story-result/{run_id}/{target_language}"
        if include_transcript:
            url = f"{base_url}?include_transcript=true"
        else:
            url = base_url

        # Request the translation results
        response = requests.get(
            url,
            headers=headers
        )

        # Verify the request was successful
        response.raise_for_status()

        # Parse the result information
        result_data = response.json()

        # Display information about the retrieved assets
        print("Translation assets successfully retrieved!")
        print(f"Complete Story Audio: {result_data.get('audio_url')}")
        print(f"Dialogue Audio: {result_data.get('dialogue_url')}")

        # Check if transcript was requested and is available
        if include_transcript and 'transcript' in result_data:
            print(f"Transcript entries: {len(result_data.get('transcript'))} dialogue items")

        return result_data

    except requests.exceptions.RequestException as e:
        print(f"Error retrieving translation results: {e}")
        if hasattr(e, 'response') and e.response is not None:
            print(f"Response content: {e.response.text}")
        return None

# Example usage
run_id = 12345  # Replace with your actual `run_id`
target_language = 76  # Example: French language code
translation_result = get_translation_result(run_id, target_language, include_transcript=True)

def process_translation_assets(result_data, run_id):
    """
    Demonstrates common ways to process and utilize the translation assets.

    Parameters:
    - result_data: The dictionary returned from get_translation_result()
    - run_id: The ID of the translation run for file naming
    """
    if not result_data:
        print("No translation data available to process.")
        return

    # Download the complete story audio file
    if 'audio_url' in result_data:
        audio_response = requests.get(result_data['audio_url'])
        if audio_response.status_code == 200:
            # Save the audio file locally
            with open(f"translated_story_{run_id}.flac", 'wb') as f:
                f.write(audio_response.content)
            print("Complete story audio file downloaded successfully.")

    # Download the dialogue audio file
    if 'dialogue_url' in result_data:
        dialogue_response = requests.get(result_data['dialogue_url'])
        if dialogue_response.status_code == 200:
            # Save the dialogue audio file locally
            with open(f"dialogue_audio_{run_id}.flac", 'wb') as f:
                f.write(dialogue_response.content)
            print("Dialogue audio file downloaded successfully.")

    # Process the transcript if available
    if 'transcript' in result_data and result_data['transcript']:
        # Example: Export the transcript as a formatted text file
        with open(f"transcript_{run_id}.txt", 'w', encoding='utf-8') as f:
            for dialogue_item in result_data['transcript']:
                speaker = dialogue_item.get('speaker', 'Narrator')
                text = dialogue_item.get('text', '')
                f.write(f"{speaker}: {text}\n\n")
        print("Transcript exported successfully.")

        # Example: Count dialogue entries by speaker
        speaker_counts = {}
        for dialogue_item in result_data['transcript']:
            speaker = dialogue_item.get('speaker', 'Narrator')
            speaker_counts[speaker] = speaker_counts.get(speaker, 0) + 1

        print("Dialogue distribution by speaker:")
        for speaker, count in speaker_counts.items():
            print(f"  - {speaker}: {count} lines")

# Process the retrieved translation assets
run_id = 12345  # Use the same run_id from earlier
if translation_result:
    process_translation_assets(translation_result, run_id)