youtube_mp3_cli.py

#!/usr/bin/env python3
"""
YouTube MP3 Transcription CLI

A simple command-line interface for transcribing YouTube videos to text.
"""

import argparse
import sys
from pathlib import Path

# Add src to path
sys.path.insert(0, str(Path(__file__).parent / "src"))

from src.youtube_mp3_service import YouTubeMP3TranscriptionService

def main():
    parser = argparse.ArgumentParser(
        description="Download MP3 from YouTube and transcribe with Google Gemini",
        epilog="""
Examples:
  %(prog)s "https://www.youtube.com/watch?v=dQw4w9WgXcQ"
  %(prog)s "https://youtu.be/9bZkp7q19f0" --keep-mp3
  %(prog)s "https://www.youtube.com/watch?v=example" --output transcript.txt
        """,
        formatter_class=argparse.RawDescriptionHelpFormatter
    )

    parser.add_argument(
        "url",
        help="YouTube URL to transcribe"
    )

    parser.add_argument(
        "--keep-mp3",
        action="store_true",
        help="Keep the downloaded MP3 file after transcription"
    )

    parser.add_argument(
        "--output", "-o",
        type=str,
        help="Output file to save the transcript (default: print to stdout)"
    )

    parser.add_argument(
        "--info-only",
        action="store_true",
        help="Only get video information without transcription"
    )

    parser.add_argument(
        "--chunk-duration",
        type=int,
        default=300,
        help="Maximum chunk duration in seconds (default: 300)"
    )

    args = parser.parse_args()

    try:
        # Initialize the service
        print("🔄 Initializing YouTube MP3 transcription service...")
        service = YouTubeMP3TranscriptionService()

        # Set custom chunk duration if specified
        if args.chunk_duration != 300:
            service.max_chunk_duration = args.chunk_duration
            print(f"⚙️  Set chunk duration to {args.chunk_duration} seconds")

        # Validate URL
        if not service.is_youtube_url(args.url):
            print(f"❌ Error: Invalid YouTube URL: {args.url}")
            sys.exit(1)

        # Get video info
        print("📋 Getting video information...")
        video_info = service.get_video_info(args.url)

        print(f"📝 Title: {video_info.get('title', 'Unknown')}")
        print(f"🆔 Video ID: {video_info.get('id', 'Unknown')}")
        print(f"⏱️  Duration: {video_info.get('duration', 0)} seconds ({video_info.get('duration', 0) / 60:.1f} minutes)")
        print(f"👤 Uploader: {video_info.get('uploader', 'Unknown')}")

        if args.info_only:
            print("ℹ️  Info-only mode. Exiting without transcription.")
            return

        # Transcribe
        print("🎵 Starting MP3 download and transcription...")
        print("⏳ This may take a few minutes depending on video length...")

        result = service.transcribe_youtube_video(args.url, keep_mp3=args.keep_mp3)

        if result.success:
            print(f"✅ Transcription completed successfully!")
            print(f"📊 Processing time: {result.processing_time:.2f} seconds")
            print(f"💾 Audio file size: {result.audio_file_size_mb:.2f} MB")
            print(f"🧩 Chunks processed: {result.chunks_processed}")
            print(f"📝 Transcript length: {len(result.transcript or '')} characters")

            if args.keep_mp3:
                print("💾 MP3 file was preserved")

            # Output transcript
            if args.output:
                # Save to file
                with open(args.output, 'w', encoding='utf-8') as f:
                    f.write(result.transcript or '')
                print(f"💾 Transcript saved to: {args.output}")
            else:
                # Print to stdout
                print("\n" + "="*60)
                print("📖 TRANSCRIPT:")
                print("="*60)
                print(result.transcript or "No transcript generated")
                print("="*60)

        else:
            print(f"❌ Transcription failed: {result.error}")
            sys.exit(1)

    except KeyboardInterrupt:
        print("\n⛔ Operation cancelled by user")
        sys.exit(1)
    except Exception as e:
        print(f"❌ Error: {e}")
        sys.exit(1)

if __name__ == "__main__":
    main()