lattifai 0.4.6__py3-none-any.whl → 1.0.0__py3-none-any.whl

lattifai-1.0.0.dist-info/RECORD

@@ -0,0 +1,52 @@
+lattifai/__init__.py,sha256=K46XVINrXgjGehO8uXByTIbUnBCdB7QwsvVNWzKbdeU,2364
+lattifai/audio2.py,sha256=WPAhcaEoIMRQBf2QZe-0yyAbgyyiqUVAthJ-z54R9Wc,7761
+lattifai/client.py,sha256=Wkz7Q1XvCQ9KxD0uZ_M1ix457ZbgIG1gAxxt8nMBUj4,22147
+lattifai/errors.py,sha256=dFQ_7c8rwuHrq2pDPjpzA755tAV3t8daXXFbHmWblbs,11015
+lattifai/logging.py,sha256=MbUEeOUFlF92pA9v532DiPPWKl03S7UHCJ6Z652cf0w,2860
+lattifai/mixin.py,sha256=7D6zeI9EIhva5FRomnj2RaJkY6ghR86otbqudcijGK4,23380
+lattifai/types.py,sha256=SjYBfwrCBOXlICvH04niFQJ7OzTx7oTaa_npfRkB67U,659
+lattifai/utils.py,sha256=TqOPrd_Et7KxrbfI_JbBNIGZ5-oGJY8ZUyJMPDTih1I,3848
+lattifai/alignment/__init__.py,sha256=ehpkKfjNIYUx7_M-RWD_8Efcrzd9bE-NSm0QgMMVLW0,178
+lattifai/alignment/lattice1_aligner.py,sha256=soBRZ98jRIju-wN5eqYUmQfF56KiEUxVGw0UvtRcx4A,4464
+lattifai/alignment/lattice1_worker.py,sha256=U3izvaBCnrRmr2jBnaEcHLJLGHbZlHNb1qroFxTH8oQ,7440
+lattifai/alignment/phonemizer.py,sha256=fbhN2DOl39lW4nQWKzyUUTMUabg7v61lB1kj8SKK-Sw,1761
+lattifai/alignment/segmenter.py,sha256=-FKtIwv9Z4fU9Fs08jhL9VyREVSYcfcwuTqb8jxCiuo,6228
+lattifai/alignment/tokenizer.py,sha256=WilqU9Ecdkl_cW86IkB1mh_PFlHN-35Jsreiyse2r-8,22355
+lattifai/caption/__init__.py,sha256=6MM_2j6CaqwZ81LfSy4di2EP0ykvheRjMZKAYDx2rQs,477
+lattifai/caption/caption.py,sha256=NNkBJbSdfXe4CwlCvMplrd4UOxlZyxq5Cs5g-dReB1E,46974
+lattifai/caption/gemini_reader.py,sha256=GqY2w78xGYCMDP5kD5WGS8jK0gntel2SK-EPpPKTrwU,15138
+lattifai/caption/gemini_writer.py,sha256=sYPxYEmVQcEan5WVGgSrcraxs3QJRQRh8CJkl2yUQ1s,6515
+lattifai/caption/supervision.py,sha256=DRrM8lfKU_x9aVBcLG6xnT0xIJrnc8jzHpzcSwQOg8c,905
+lattifai/caption/text_parser.py,sha256=XDb8KTt031uJ1hg6dpbINglGOTX-6pBcghbg3DULM1I,4633
+lattifai/cli/__init__.py,sha256=dIUmrpN-OwR4h6BqMhXp87_5ZwgO41ShPru_iZGnpQs,463
+lattifai/cli/alignment.py,sha256=uKMTE95_JMikfbyCcwLbQxms-EQmZXEj7oYugiupk9I,5890
+lattifai/cli/caption.py,sha256=ucgYxJ43ab71nGpZBAiVn8QA0DAVht2QMZFE5IdgxP0,6853
+lattifai/cli/server.py,sha256=k3kuQsZ2JNHZD_eMDxGSn5OFHiUrC2ybHN_xQQlLEgU,493
+lattifai/cli/transcribe.py,sha256=6uJfvtB1o_u1uQwxt4fje_koyfN93mGaFLlskmjqx2c,7406
+lattifai/cli/youtube.py,sha256=9_erdIkhX8pCiy7BRzNstEiO9saM-VKZ1WVqvbXbmrc,5267
+lattifai/config/__init__.py,sha256=Z8OudvS6fgfLNLu_2fvoXartQiYCECOnNfzDt-PfCN4,543
+lattifai/config/alignment.py,sha256=z0b9tg67ftDI90j9Td3qmXFg3WZjSbgszjzeTdwrjZA,3291
+lattifai/config/caption.py,sha256=nmfdsJ-18l4UmapdVgxF1ARJbA4aOr7jek1bmse2F_E,6787
+lattifai/config/client.py,sha256=I1JqLQlsQNU5ouovTumr-PP_8GWC9DI_e9B5UwsDZws,1492
+lattifai/config/diarization.py,sha256=cIkwCfsYqfMns3i6tKWcwBBBkdnhhmB_Eo0TuOPCw9o,2484
+lattifai/config/media.py,sha256=5JOPjifXDM2WWQERySDZen4-7YfgQNcYM2NkkKp0LjQ,13610
+lattifai/config/transcription.py,sha256=bzghOGgcNWzTnDYd_cqCOB7GT8OnzHDiyam7LSixqxM,2901
+lattifai/diarization/__init__.py,sha256=MgBDQ1ehL2qDnZprEp8KqON7CmbG-qaP37gzBsV0jzk,119
+lattifai/diarization/lattifai.py,sha256=SE2BpIZ3_deKyhXdBqe77bsDLXIUV9AQV34gfINv7_s,2657
+lattifai/server/app.py,sha256=ndMBC_FuAFZGs9bvrgHUMtgDDDeYGhCkW0Lx49SEh7k,15193
+lattifai/transcription/__init__.py,sha256=mEoMTbs5jAgtXQn1jTjlFY_GUr-S0WmPn8uZ6WZCkU0,2643
+lattifai/transcription/base.py,sha256=59b4nQHFMyTRyyzBJTM8ZpEuUy1KjwA2o6rNfrNluKY,3911
+lattifai/transcription/gemini.py,sha256=pdc4Q1rk1QLpheC7zIauUOXWViUOM1qGnwC5cRoxH-4,7992
+lattifai/transcription/lattifai.py,sha256=h0nhXST0qljhyndf80IEddM7Y_N1jiS28YoaE536eME,3483
+lattifai/transcription/prompts/__init__.py,sha256=G9b42COaCYv3sPPNkHsGDLOMBuVGKt4mXGYal_BYtYQ,1351
+lattifai/workflow/__init__.py,sha256=GOT9jptXwpIMiNRqJ_LToEt_5Dt0k7XXbLkFzhrl31o,548
+lattifai/workflow/agents.py,sha256=yEOnxnhcTvr1iOhCorNvp8B76P6nQsLRXJCu_rCYFfM,38
+lattifai/workflow/base.py,sha256=8QoVIBZwJfr5mppJbtUFafHv5QR9lL-XrULjTWD0oBg,6257
+lattifai/workflow/file_manager.py,sha256=d106KHLY8A9amLy5h1vR32e4od8mmJGqMD-iDyiRPLI,32917
+lattifai/workflow/youtube.py,sha256=n8L1c6tl8FuYzAzKZ-B76zf5yZsvVggZEJ9mPdbEWGQ,22989
+lattifai-1.0.0.dist-info/licenses/LICENSE,sha256=_IkHdwOWLAWcE1M_tIpDoRWdNSJwFdtIqI-XSkK3yPU,1066
+lattifai-1.0.0.dist-info/METADATA,sha256=orTvmokTIN-W7Lcm5LBs3nof6T8DWKcY9mrU1MwOApE,23109
+lattifai-1.0.0.dist-info/WHEEL,sha256=SmOxYU7pzNKBqASvQJ7DjX3XGUF92lrGhMb3R6_iiqI,91
+lattifai-1.0.0.dist-info/entry_points.txt,sha256=Ay-Rxb214lYSuJIXO_wuNlUu1LbvOXtMJ-G-PcvGZFw,424
+lattifai-1.0.0.dist-info/top_level.txt,sha256=tHSoXF26r-IGfbIP_JoYATqbmf14h5NrnNJGH4j5reI,9
+lattifai-1.0.0.dist-info/RECORD,,

{lattifai-0.4.6.dist-info → lattifai-1.0.0.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.9.0)
+Generator: setuptools (79.0.1)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

lattifai-1.0.0.dist-info/entry_points.txt

@@ -0,0 +1,13 @@
+[console_scripts]
+lai-align = lattifai.cli.alignment:main
+lai-server = lattifai.cli.server:main
+lai-transcribe = lattifai.cli.transcribe:main
+lai-youtube = lattifai.cli.youtube:main
+laicap-convert = lattifai.cli.caption:main_convert
+laicap-normalize = lattifai.cli.caption:main_normalize
+laicap-shift = lattifai.cli.caption:main_shift
+
+[lai_run.cli]
+alignment = lattifai.cli
+caption = lattifai.cli
+transcribe = lattifai.cli

lattifai/base_client.py

@@ -1,126 +0,0 @@
-"""Base client classes for LattifAI SDK."""
-
-import os
-from abc import ABC
-from typing import Any, Awaitable, Callable, Dict, Optional, Union  # noqa: F401
-
-import httpx
-
-# Import from errors module for consistency
-from .errors import APIError, ConfigurationError, LattifAIError
-
-
-class BaseAPIClient(ABC):
-    """Abstract base class for API clients."""
-
-    def __init__(
-        self,
-        *,
-        api_key: Optional[str] = None,
-        base_url: Optional[str] = None,
-        timeout: Union[float, httpx.Timeout] = 60.0,
-        max_retries: int = 2,
-        default_headers: Optional[Dict[str, str]] = None,
-    ) -> None:
-        if api_key is None:
-            api_key = os.environ.get("LATTIFAI_API_KEY")
-        if api_key is None:
-            raise ConfigurationError(
-                "The api_key client option must be set either by passing api_key to the client "
-                "or by setting the LATTIFAI_API_KEY environment variable"
-            )
-
-        self._api_key = api_key
-        self._base_url = base_url
-        self._timeout = timeout
-        self._max_retries = max_retries
-
-        headers = {
-            "User-Agent": "LattifAI/Python",
-            "Authorization": f"Bearer {self._api_key}",
-        }
-        if default_headers:
-            headers.update(default_headers)
-        self._default_headers = headers
-
-
-class SyncAPIClient(BaseAPIClient):
-    """Synchronous API client."""
-
-    def __init__(self, **kwargs) -> None:
-        super().__init__(**kwargs)
-        self._client = httpx.Client(
-            base_url=self._base_url,
-            timeout=self._timeout,
-            headers=self._default_headers,
-        )
-
-    def __enter__(self):
-        return self
-
-    def __exit__(self, exc_type, exc_val, exc_tb):
-        self.close()
-
-    def close(self) -> None:
-        """Close the HTTP client."""
-        self._client.close()
-
-    def _request(
-        self,
-        method: str,
-        url: str,
-        *,
-        json: Optional[Dict[str, Any]] = None,
-        **kwargs,
-    ) -> httpx.Response:
-        """Make an HTTP request."""
-        return self._client.request(method=method, url=url, json=json, **kwargs)
-
-    def post(self, api_endpoint: str, *, json: Optional[Dict[str, Any]] = None, **kwargs) -> httpx.Response:
-        """Make a POST request to the specified API endpoint."""
-        return self._request("POST", api_endpoint, json=json, **kwargs)
-
-
-class AsyncAPIClient(BaseAPIClient):
-    """Asynchronous API client."""
-
-    def __init__(self, **kwargs) -> None:
-        super().__init__(**kwargs)
-        self._client = httpx.AsyncClient(
-            base_url=self._base_url,
-            timeout=self._timeout,
-            headers=self._default_headers,
-        )
-
-    async def __aenter__(self):
-        return self
-
-    async def __aexit__(self, exc_type, exc_val, exc_tb):
-        await self.close()
-
-    async def close(self) -> None:
-        """Close the HTTP client."""
-        await self._client.aclose()
-
-    async def _request(
-        self,
-        method: str,
-        url: str,
-        *,
-        json: Optional[Dict[str, Any]] = None,
-        files: Optional[Dict[str, Any]] = None,
-        **kwargs,
-    ) -> httpx.Response:
-        """Make an HTTP request."""
-        return await self._client.request(method=method, url=url, json=json, files=files, **kwargs)
-
-    async def post(
-        self,
-        api_endpoint: str,
-        *,
-        json: Optional[Dict[str, Any]] = None,
-        files: Optional[Dict[str, Any]] = None,
-        **kwargs,
-    ) -> httpx.Response:
-        """Make a POST request to the specified API endpoint."""
-        return await self._request("POST", api_endpoint, json=json, files=files, **kwargs)

lattifai/bin/__init__.py

@@ -1,3 +0,0 @@
-from .agent import *  # noqa
-from .align import *  # noqa
-from .subtitle import *  # noqa

lattifai/bin/agent.py

@@ -1,324 +0,0 @@
-"""
-Agent command for YouTube workflow
-"""
-
-import asyncio
-import os
-import sys
-from typing import Optional
-
-import click
-import colorful
-
-from lattifai.bin.cli_base import cli
-from lattifai.io import OUTPUT_SUBTITLE_FORMATS
-
-
-@cli.command()
-@click.option("--youtube", "--yt", is_flag=True, help="Process YouTube URL through agentic workflow.")
-@click.option(
-    "-K",
-    "-L",
-    "--api-key",
-    "--api_key",
-    type=str,
-    help="LattifAI API key for alignment (overrides LATTIFAI_API_KEY env var).",
-)
-@click.option(
-    "-G",
-    "--gemini-api-key",
-    "--gemini_api_key",
-    type=str,
-    help="Gemini API key for transcription (overrides GEMINI_API_KEY env var).",
-)
-@click.option(
-    "-D",
-    "--device",
-    type=click.Choice(["cpu", "cuda", "mps"], case_sensitive=False),
-    default="cpu",
-    help="Device to use for inference.",
-)
-@click.option(
-    "-M",
-    "--model-name-or-path",
-    "--model_name_or_path",
-    type=str,
-    default="Lattifai/Lattice-1-Alpha",
-    help="Model name or path for alignment.",
-)
-@click.option(
-    "--media-format",
-    "--media_format",
-    type=click.Choice(
-        ["mp3", "wav", "m4a", "aac", "opus", "mp4", "webm", "mkv", "avi", "mov", "flv", "wmv", "mpeg", "mpg", "3gp"],
-        case_sensitive=False,
-    ),
-    default="mp4",
-    help="Media format for YouTube download (audio or video).",
-)
-@click.option(
-    "--output-format",
-    "--output_format",
-    type=click.Choice(OUTPUT_SUBTITLE_FORMATS, case_sensitive=False),
-    default="srt",
-    help="Subtitle output format.",
-)
-@click.option(
-    "--output-dir",
-    "--output_dir",
-    type=click.Path(exists=False, file_okay=False, dir_okay=True),
-    help="Output directory for generated files (default: current directory).",
-)
-@click.option(
-    "--max-retries",
-    "--max_retries",
-    type=int,
-    default=0,
-    help="Maximum number of retries for failed steps.",
-)
-@click.option(
-    "-S",
-    "--split-sentence",
-    "--split_sentence",
-    is_flag=True,
-    default=False,
-    help="Re-segment subtitles by semantics.",
-)
-@click.option(
-    "--word-level",
-    "--word_level",
-    is_flag=True,
-    default=False,
-    help="Include word-level alignment timestamps in output (for JSON, TextGrid, and subtitle formats).",
-)
-@click.option("--verbose", "-v", is_flag=True, help="Enable verbose logging.")
-@click.option("--force", "-f", is_flag=True, help="Force overwrite existing files without confirmation.")
-@click.argument("url", type=str, required=True)
-def agent(
-    youtube: bool,
-    url: str,
-    api_key: Optional[str] = None,
-    gemini_api_key: Optional[str] = None,
-    device: str = "cpu",
-    model_name_or_path: str = "Lattifai/Lattice-1-Alpha",
-    media_format: str = "mp4",
-    output_format: str = "srt",
-    output_dir: Optional[str] = None,
-    max_retries: int = 0,
-    split_sentence: bool = False,
-    word_level: bool = False,
-    verbose: bool = False,
-    force: bool = False,
-):
-    """
-    LattifAI Agentic Workflow Agent
-
-    Process multimedia content through intelligent agent-based pipelines.
-
-    Example:
-        lattifai agent --youtube https://www.youtube.com/watch?v=example
-    """
-
-    if not youtube:
-        click.echo(colorful.red("❌ Please specify a workflow type. Use --youtube for YouTube processing."))
-        return
-
-    # Setup logging
-    import logging
-
-    log_level = logging.DEBUG if verbose else logging.INFO
-    logging.basicConfig(level=log_level, format="%(asctime)s - %(name)s - %(levelname)s - %(message)s")
-
-    # Set default output directory
-    if not output_dir:
-        output_dir = os.getcwd()
-
-    # Get API keys
-    lattifai_api_key = api_key or os.getenv("LATTIFAI_API_KEY")
-    gemini_key = gemini_api_key or os.getenv("GEMINI_API_KEY")
-
-    try:
-        # Run the YouTube workflow
-        asyncio.run(
-            _run_youtube_workflow(
-                url=url,
-                lattifai_api_key=lattifai_api_key,
-                gemini_api_key=gemini_key,
-                device=device,
-                model_name_or_path=model_name_or_path,
-                media_format=media_format,
-                output_format=output_format,
-                output_dir=output_dir,
-                max_retries=max_retries,
-                split_sentence=split_sentence,
-                word_level=word_level,
-                force_overwrite=force,
-            )
-        )
-
-    except KeyboardInterrupt:
-        click.echo(colorful.yellow("\n⚠️ Process interrupted by user"))
-        sys.exit(1)
-    except Exception as e:
-        from lattifai.errors import LattifAIError
-
-        # Extract error message without support info (to avoid duplication)
-        if isinstance(e, LattifAIError):
-            # Use the get_message() method which includes proper formatting
-            click.echo(colorful.red("❌ Workflow failed:"))
-            click.echo(e.get_message())
-            # Show support info once at the end
-            click.echo(e.get_support_info())
-        else:
-            click.echo(colorful.red(f"❌ Workflow failed: {str(e)}"))
-
-        if verbose:
-            import traceback
-
-            traceback.print_exc()
-        sys.exit(1)
-
-
-async def _run_youtube_workflow(
-    url: str,
-    lattifai_api_key: Optional[str],
-    gemini_api_key: str,
-    device: str,
-    model_name_or_path: str,
-    media_format: str,
-    output_format: str,
-    output_dir: str,
-    max_retries: int,
-    split_sentence: bool = False,
-    word_level: bool = False,
-    force_overwrite: bool = False,
-):
-    """Run the YouTube processing workflow"""
-
-    click.echo(colorful.cyan("🚀 LattifAI Agentic Workflow - YouTube Processing"))
-    click.echo(f"📺      YouTube URL: {url}")
-    click.echo(f"🎬     Media format: {media_format}")
-    click.echo(f"📝    Output format: {output_format}")
-    click.echo(f"📁 Output directory: {output_dir}")
-    click.echo(f"🔄      Max retries: {max_retries}")
-    click.echo()
-
-    # Import workflow components
-    from lattifai.client import AsyncLattifAI
-    from lattifai.workflows import YouTubeSubtitleAgent
-    from lattifai.workflows.gemini import GeminiTranscriber
-    from lattifai.workflows.youtube import YouTubeDownloader
-
-    # Initialize components with their configuration (only persistent config, not runtime params)
-    downloader = YouTubeDownloader()
-    transcriber = GeminiTranscriber(api_key=gemini_api_key)
-    aligner = AsyncLattifAI(model_name_or_path=model_name_or_path, device=device, api_key=lattifai_api_key)
-
-    # Initialize agent with components
-    agent = YouTubeSubtitleAgent(
-        downloader=downloader,
-        transcriber=transcriber,
-        aligner=aligner,
-        max_retries=max_retries,
-    )
-
-    # Process the URL
-    result = await agent.process_youtube_url(
-        url=url,
-        output_dir=output_dir,
-        media_format=media_format,
-        force_overwrite=force_overwrite,
-        output_format=output_format,
-        split_sentence=split_sentence,
-        word_level=word_level,
-    )
-
-    # Display results
-    click.echo(colorful.bold_white_on_green("🎉 Workflow completed successfully!"))
-    click.echo()
-    click.echo(colorful.bold_white_on_green("📊 Results:"))
-
-    # Show metadata
-    metadata = result.get("metadata", {})
-    if metadata:
-        click.echo(f'🎬    Title: {metadata.get("title", "Unknown")}')
-        click.echo(f'👤 Uploader: {metadata.get("uploader", "Unknown").strip()}')
-        click.echo(f'⏱️  Duration: {metadata.get("duration", 0)} seconds')
-        click.echo()
-
-    # Show exported files
-    exported_files = result.get("exported_files", {})
-    if exported_files:
-        click.echo(colorful.bold_white_on_green("📄 Generated subtitle files:"))
-        for format_name, file_path in exported_files.items():
-            click.echo(f"  {format_name.upper()}: {file_path}")
-        click.echo()
-
-    # Show subtitle count
-    subtitle_count = result.get("subtitle_count", 0)
-    click.echo(f"📝 Generated {subtitle_count} subtitle segments")
-
-    click.echo(colorful.bold_white_on_green("✨ All done! Your aligned subtitles are ready."))
-
-
-# Add dependencies check
-def check_dependencies():
-    """Check if required dependencies are installed"""
-    missing_deps = []
-
-    try:
-        from google import genai  # noqa: F401
-    except ImportError:
-        missing_deps.append("google-genai")
-
-    try:
-        import yt_dlp  # noqa: F401
-    except ImportError:
-        missing_deps.append("yt-dlp")
-
-    try:
-        from dotenv import load_dotenv  # noqa: F401
-    except ImportError:
-        missing_deps.append("python-dotenv")
-
-    if missing_deps:
-        click.echo(colorful.red("❌ Missing required dependencies:"))
-        for dep in missing_deps:
-            click.echo(f"  - {dep}")
-        click.echo()
-        click.echo("Install them with:")
-        click.echo(f'  pip install {" ".join(missing_deps)}')
-        return False
-
-    return True
-
-
-# Check dependencies when module is imported
-if not check_dependencies():
-    pass  # Don't exit on import, let the command handle it
-
-
-if __name__ == "__main__":
-    import os
-
-    from dotenv import find_dotenv, load_dotenv
-
-    load_dotenv(find_dotenv(usecwd=True))
-
-    asyncio.run(
-        _run_youtube_workflow(
-            # url='https://www.youtube.com/watch?v=7nv1snJRCEI',
-            url="https://www.youtube.com/watch?v=DQacCB9tDaw",
-            lattifai_api_key=os.getenv("LATTIFAI_API_KEY"),
-            gemini_api_key=os.getenv("GEMINI_API_KEY", ""),
-            device="mps",
-            model_name_or_path="Lattifai/Lattice-1-Alpha",
-            media_format="mp3",
-            output_format="TextGrid",
-            output_dir="~/Downloads/lattifai_youtube",
-            max_retries=0,
-            split_sentence=True,
-            word_level=False,
-            force_overwrite=False,
-        )
-    )