test_api.py

from udio_api_wrapper import UdioApiClient
import os
import sys

def test_music_generation(token):
    """Test the music generation API with Rise & Run lyrics"""
    client = UdioApiClient(token)
    
    # Print API details for debugging
    print(f"\nAPI Base URL: {client.API_BASE_URL}")
    print(f"Using token: {token[:8]}...{token[-4:]}\n")
    
    try:
        # Generate the Rise & Run theme
        print("Generating Rise & Run theme...")
        
        lyrics = """[Verse]
In a world of care, where dreams align,
There's a force that stands the test of time.
Rise & Run, with hearts so true,
Building bridges, bringing life anew.

Every story, every smile,
They go the extra mile.
Lighting the way, like the morning sun,
That's the power of Rise & Run.

[Verse 2]
From the first hello to the last goodbye,
They lift the weary, help dreams fly high.
Compassion burns in all they do,
A guiding light that pulls us through.

Marketing more than just a brand,
They touch the soul, they understand.
It's not just work; it's a calling begun,
A legacy of care at Rise & Run.

[Chorus]
Rise & Run, together we thrive,
Changing the world, keeping hope alive.
With every heartbeat, every call,
They lift us higher, they catch us all.
Rise & Run, oh, they're second to none,
Healing the world, one step, one run.

[Verse 3]
Oh, they rise for the broken,
Run for the lost,
Building a future, no matter the cost.
They lift the heavy, embrace the small,
In their hands, there's room for all.

[Bridge]
So here's to the dreamers, the doers of love,
Guided by something from far above.
Rise & Run, your journey's begun,
Together, we're stronger—together, we've won.

[Chorus]
Rise & Run, together we thrive,
Changing the world, keeping hope alive.
With every heartbeat, every call,
They lift us higher, they catch us all.
Rise & Run, oh, they're second to none,
Healing the world, one step, one run."""
        
        result = client.generate_music(
            prompt=lyrics,
            title="Rise & Run Theme",
            model="chirp-v3.5",  # Using the latest model for better quality
            make_instrumental=False  # We want vocals for the lyrics
        )
        
        work_id = result.get('workId')
        if not work_id:
            print("Error: No work ID received")
            return
        
        print(f"Generation started with work ID: {work_id}")
        print("Waiting for completion (this may take a few minutes)...")
        
        # Wait for the generation to complete
        final_result = client.wait_for_completion(work_id)
        
        # Print the results
        print("\nGeneration completed!")
        print("Generated tracks:")
        for track in final_result.get('response_data', []):
            print(f"\nTitle: {track.get('title')}")
            print(f"Audio URL: {track.get('audio_url')}")
            print(f"Tags: {track.get('tags')}")
            if track.get('error_message'):
                print(f"Error: {track.get('error_message')}")
            
    except Exception as e:
        print(f"\nError Details:")
        print(f"Type: {type(e).__name__}")
        print(f"Message: {str(e)}")
        if hasattr(e, '__cause__') and e.__cause__:
            print(f"Cause: {str(e.__cause__)}")

if __name__ == "__main__":
    # Get token from command line argument or environment variable
    token = None
    
    if len(sys.argv) > 1:
        token = sys.argv[1]
    else:
        token = os.getenv("UDIO_API_TOKEN")
        
    if not token:
        print("Please provide your API token as a command line argument or set the UDIO_API_TOKEN environment variable")
        sys.exit(1)
        
    test_music_generation(token)