from tools.youtube_transcript import get_transcript, list_transcript_languages
import json
print("=== Testing list_transcript_languages ===")
result = list_transcript_languages("dQw4w9WgXcQ")
print(json.dumps(result, indent=2))
print("\n=== Testing get_transcript (plain) ===")
result = get_transcript("dQw4w9WgXcQ", ["en"], "plain")
if result["success"]:
print("Language:", result["language"])
print("Generated:", result["is_generated"])
print("Length:", len(result["transcript"]), "chars")
print("First 300 chars:")
print(result["transcript"][:300])
else:
print("Error:", result)
print("\n=== Testing get_transcript (structured) ===")
result = get_transcript("dQw4w9WgXcQ", ["en"], "structured")
if result["success"]:
entries = result["transcript"]["entries"]
print("Entries count:", len(entries))
print("First 3 entries:")
for entry in entries[:3]:
print(" [%.2fs] %s" % (entry["start"], entry["text"]))
else:
print("Error:", result)
