main.py

#!/usr/bin/env python3
"""
WhatsApp Custom Virtual Background
Main entry point for the virtual background system

Usage:
    python main.py --mode preview --bg your_room.png
    python main.py --mode virtualcam --bg your_room.glb
    python main.py --test
"""

import argparse
import sys
import os
from pathlib import Path

# Ensure imports work
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))

from config import VirtualBGConfig, CONFIG_DEFAULT, CONFIG_HIGH_QUALITY, CONFIG_LOW_LATENCY, CONFIG_MOBILE


def run_preview(args):
    """Run in preview mode with GUI"""
    from virtual_camera import WhatsAppVirtualCamera

    print("=" * 50)
    print("WhatsApp Virtual Background - Preview Mode")
    print("=" * 50)

    config = CONFIG_DEFAULT
    if args.quality == "high":
        config = CONFIG_HIGH_QUALITY
    elif args.quality == "low":
        config = CONFIG_LOW_LATENCY
    elif args.quality == "mobile":
        config = CONFIG_MOBILE

    # Override with command line args
    if args.width:
        config.width = args.width
    if args.height:
        config.height = args.height
    if args.fps:
        config.fps = args.fps

    # Get background path
    bg_path = args.bg if args.bg else None
    bg_3d_path = args.bg_3d if args.bg_3d else None

    if not bg_path and not bg_3d_path:
        print("\nWarning: No background specified. Using default gradient.")
        print("To use your Polycam scan, provide the path with --bg or --bg-3d")

    print(f"\nSettings:")
    print(f"  Resolution: {config.width}x{config.height}")
    print(f"  FPS: {config.fps}")
    print(f"  Background: {bg_path or bg_3d_path or 'Default'}")
    print(f"  Quality: {args.quality}")
    print()

    try:
        vcam = WhatsAppVirtualCamera(
            bg_path=bg_path,
            bg_3d_path=bg_3d_path,
            width=config.width,
            height=config.height,
            fps=config.fps
        )
        vcam.start(webcam_index=args.camera)
    except KeyboardInterrupt:
        print("\nInterrupted by user")
    except Exception as e:
        print(f"\nError: {e}")
        import traceback
        traceback.print_exc()


def run_virtualcam(args):
    """Run as virtual camera source"""
    from virtual_camera import WhatsAppVirtualCamera

    print("=" * 50)
    print("WhatsApp Virtual Background - Virtual Camera Mode")
    print("=" * 50)
    print("\nThis will create a virtual camera device.")
    print("Select it in WhatsApp > Settings > Camera")
    print("Press Ctrl+C to stop\n")

    config = CONFIG_DEFAULT
    bg_path = args.bg if args.bg else None
    bg_3d_path = args.bg_3d if args.bg_3d else None

    try:
        vcam = WhatsAppVirtualCamera(
            bg_path=bg_path,
            bg_3d_path=bg_3d_path,
            width=config.width,
            height=config.height,
            fps=config.fps
        )
        vcam.start(webcam_index=args.camera)
    except Exception as e:
        print(f"Error: {e}")


def run_test(args):
    """Run test suite"""
    from virtual_camera import test_virtual_camera
    test_virtual_camera()


def main():
    parser = argparse.ArgumentParser(
        description='WhatsApp Custom Virtual Background',
        formatter_class=argparse.RawDescriptionHelpFormatter,
        epilog="""
Examples:
  %(prog)s --mode preview --bg my_room.jpg
  %(prog)s --mode virtualcam --bg-3d room_scan.glb
  %(prog)s --test
  %(prog)s --mode preview --quality high --bg background.png
        """
    )

    # Main arguments
    parser.add_argument('--mode', type=str, default='preview',
                       choices=['preview', 'virtualcam', 'test'],
                       help='Operation mode (default: preview)')

    # Background options
    parser.add_argument('--bg', type=str, help='Path to 2D background image')
    parser.add_argument('--bg-3d', type=str, help='Path to 3D scene file (GLB/GLTF/OBJ)')

    # Camera options
    parser.add_argument('--camera', type=int, default=0, help='Webcam index (default: 0)')
    parser.add_argument('--width', type=int, help='Output width')
    parser.add_argument('--height', type=int, help='Output height')
    parser.add_argument('--fps', type=int, help='Frame rate')

    # Quality settings
    parser.add_argument('--quality', type=str, default='default',
                       choices=['default', 'high', 'low', 'mobile'],
                       help='Quality preset (default: default)')

    # Test mode
    parser.add_argument('--test', action='store_true', help='Run test mode')

    args = parser.parse_args()

    if args.test or args.mode == 'test':
        run_test(args)
    elif args.mode == 'preview':
        run_preview(args)
    elif args.mode == 'virtualcam':
        run_virtualcam(args)
    else:
        parser.print_help()


if __name__ == "__main__":
    main()