from sarvamai import SarvamAI
import requests

client = SarvamAI(api_subscription_key="sk_0td9brgk_yh8yJFGUzrJRVA7OuBdLexjC")
job_id = "20260308_f4673b77-482e-4490-a50a-b9a312fb6ab1"

dl = client.speech_to_text_job.get_download_links(job_id=job_id, files=["lalal_voice.wav"])
print(f"type: {type(dl.download_urls)}")

urls = dl.download_urls
if isinstance(urls, dict):
    for fname, obj in urls.items():
        url = obj.file_url if hasattr(obj, "file_url") else str(obj)
        print(f"Downloading {fname}...")
        r = requests.get(url)
        with open("/home/ubuntu/test_mIp/sarvam_diarization.json", "w") as f:
            f.write(r.text)
        print(f"Saved ({len(r.text)} chars)")
        print(r.text[:5000])
elif isinstance(urls, list):
    for obj in urls:
        url = obj.file_url if hasattr(obj, "file_url") else str(obj)
        r = requests.get(url)
        with open("/home/ubuntu/test_mIp/sarvam_diarization.json", "w") as f:
            f.write(r.text)
        print(f"Saved ({len(r.text)} chars)")
        print(r.text[:5000])