webm_ogg_url_examples.py

#!/usr/bin/env python3
"""
WebM and OGG Audio URL Processing Examples

Examples specifically for processing WebM and OGG audio files via URL
using the Text Extraction API.
"""

import requests
import json
from typing import Dict, Any

# API Configuration
API_BASE_URL = "http://localhost:8000"

def test_webm_ogg_url_processing():
    """Test WebM and OGG audio file processing via URL."""

    print("🎵 WebM and OGG Audio URL Processing Test")
    print("=" * 50)

    # Replace these with actual public WebM/OGG URLs
    test_urls = {
        "WebM Audio": "https://example.com/sample.webm",
        "OGG Audio": "https://example.com/sample.ogg",
        # You can also test with any publicly available WebM/OGG files
        # For example: "https://upload.wikimedia.org/wikipedia/commons/..."
    }

    for audio_type, url in test_urls.items():
        print(f"\n📡 Testing {audio_type}: {url}")

        # Prepare the request
        payload = {
            "url": url,
            "filename": f"test_{audio_type.lower().replace(' ', '_')}"
        }

        try:
            # Make the API request
            response = requests.post(
                f"{API_BASE_URL}/extract-url",
                json=payload,
                headers={"Content-Type": "application/json"},
                timeout=120  # Give enough time for processing
            )

            if response.status_code == 200:
                result = response.json()
                print(f"✅ Success!")
                print(f"   File ID: {result['file_id']}")
                print(f"   Processor: {result['processor_used']}")
                print(f"   Processing time: {result['processing_time']:.2f}s")
                print(f"   Text length: {result['text_length']} characters")
                print(f"   File size: {result['file_info']['size_mb']:.2f} MB")

                # Show transcript preview
                text = result['extracted_text']
                if text:
                    preview = text[:200] + "..." if len(text) > 200 else text
                    print(f"\n   Transcript preview:")
                    print(f"   \"{preview}\"")
                else:
                    print(f"   No text extracted (possibly silent audio)")

            else:
                error_detail = response.json() if response.text else {}
                print(f"❌ Error {response.status_code}: {error_detail.get('detail', 'Unknown error')}")

        except requests.exceptions.RequestException as e:
            print(f"❌ Request failed: {e}")
        except Exception as e:
            print(f"❌ Unexpected error: {e}")

def curl_examples():
    """Show curl command examples for WebM and OGG processing."""

    print("\n🔧 Curl Command Examples")
    print("=" * 30)

    print("\n1. Process WebM audio from URL:")
    print("curl -X POST http://localhost:8000/extract-url \\")
    print("     -H 'Content-Type: application/json' \\")
    print("     -d '{")
    print("       \"url\": \"https://example.com/audio.webm\",")
    print("       \"filename\": \"my_audio.webm\"")
    print("     }'")

    print("\n2. Process OGG audio from URL:")
    print("curl -X POST http://localhost:8000/extract-url \\")
    print("     -H 'Content-Type: application/json' \\")
    print("     -d '{")
    print("       \"url\": \"https://example.com/audio.ogg\",")
    print("       \"filename\": \"my_audio.ogg\"")
    print("     }'")

    print("\n3. Process multiple audio files:")
    print("curl -X POST http://localhost:8000/extract-batch-url \\")
    print("     -H 'Content-Type: application/json' \\")
    print("     -d '{")
    print("       \"urls\": [")
    print("         \"https://example.com/audio1.webm\",")
    print("         \"https://example.com/audio2.ogg\"")
    print("       ],")
    print("       \"filenames\": [\"audio1.webm\", \"audio2.ogg\"]")
    print("     }'")

def python_code_examples():
    """Show Python code examples for WebM and OGG processing."""

    print("\n🐍 Python Code Examples")
    print("=" * 25)

    example_code = '''
import requests

# Single WebM file processing
def process_webm_url(url):
    payload = {
        "url": url,
        "filename": "audio.webm"  # Optional custom filename
    }

    response = requests.post(
        "http://localhost:8000/extract-url",
        json=payload,
        headers={"Content-Type": "application/json"}
    )

    if response.status_code == 200:
        result = response.json()
        print(f"Transcript: {result['extracted_text']}")
        return result['extracted_text']
    else:
        print(f"Error: {response.json()}")
        return None

# Single OGG file processing
def process_ogg_url(url):
    payload = {"url": url}

    response = requests.post(
        "http://localhost:8000/extract-url",
        json=payload
    )

    return response.json()

# Batch processing multiple audio URLs
def process_multiple_audio_urls(urls):
    payload = {"urls": urls}

    response = requests.post(
        "http://localhost:8000/extract-batch-url",
        json=payload
    )

    return response.json()

# Example usage:
webm_transcript = process_webm_url("https://example.com/audio.webm")
ogg_result = process_ogg_url("https://example.com/audio.ogg")

multiple_results = process_multiple_audio_urls([
    "https://example.com/audio1.webm",
    "https://example.com/audio2.ogg"
])
'''

    print(example_code)

def api_response_format():
    """Show the expected API response format."""

    print("\n📋 Expected API Response Format")
    print("=" * 35)

    response_example = '''
{
    "file_id": "uuid-string",
    "success": true,
    "extracted_text": "This is the transcribed audio content...",
    "text_length": 1234,
    "processor_used": "GeminiProcessor",
    "processing_time": 15.67,
    "timestamp": "2024-01-15T10:30:00",
    "file_info": {
        "name": "audio.webm",
        "size_mb": 2.5,
        "url": "https://example.com/audio.webm"
    },
    "error": null
}
'''

    print("Single file response:")
    print(response_example)

    batch_example = '''
{
    "batch_id": "uuid-string",
    "total_files": 2,
    "successful": 2,
    "failed": 0,
    "total_processing_time": 25.34,
    "total_characters": 2468,
    "timestamp": "2024-01-15T10:30:00",
    "results": [
        {
            "file_id": "uuid-1",
            "success": true,
            "extracted_text": "First audio transcript...",
            "text_length": 1234,
            "processor_used": "GeminiProcessor",
            "processing_time": 12.45,
            "file_info": {
                "name": "audio1.webm",
                "size_mb": 1.8,
                "url": "https://example.com/audio1.webm"
            }
        },
        {
            "file_id": "uuid-2",
            "success": true,
            "extracted_text": "Second audio transcript...",
            "text_length": 1234,
            "processor_used": "GeminiProcessor",
            "processing_time": 12.89,
            "file_info": {
                "name": "audio2.ogg",
                "size_mb": 2.1,
                "url": "https://example.com/audio2.ogg"
            }
        }
    ]
}
'''

    print("\nBatch processing response:")
    print(batch_example)

def main():
    """Run WebM and OGG URL processing examples."""

    print("WebM and OGG Audio URL Processing Examples")
    print("=" * 60)

    # Check if API is running
    try:
        response = requests.get(f"{API_BASE_URL}/health", timeout=5)
        if response.status_code == 200:
            health = response.json()
            print(f"✅ API Status: {health['status']}")
            print(f"   Gemini configured: {health['api_keys_configured']['gemini']}")
        else:
            print(f"❌ API not healthy: {response.status_code}")
            return
    except Exception as e:
        print(f"❌ Cannot connect to API: {e}")
        print("   Make sure the API server is running:")
        print("   python api_server.py")
        return

    # Show examples and documentation
    curl_examples()
    python_code_examples()
    api_response_format()

    print("\n" + "=" * 60)
    print("📝 Usage Notes:")
    print("• Replace example URLs with actual WebM/OGG file URLs")
    print("• Files are processed using Google Gemini for transcription")
    print("• Supported audio formats: MP3, WAV, M4A, WebM, OGG")
    print("• Video formats (MP4, AVI, MOV, MKV) extract audio for transcription")
    print("• Maximum file size: 100MB (configurable)")
    print("• API automatically handles format conversion via PyDub/FFmpeg")

    print("\n🚀 To test with real files:")
    print("1. Find public WebM/OGG audio file URLs")
    print("2. Replace the example URLs in the test functions")
    print("3. Run: python webm_ogg_url_examples.py")

    # Uncomment to test with real URLs
    # test_webm_ogg_url_processing()

if __name__ == "__main__":
    main()