lattifai 0.4.5__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.5.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-0.4.5.dist-info → lattifai-1.0.0.dist-info}/licenses/LICENSE

@@ -18,4 +18,4 @@ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
-SOFTWARE.
+SOFTWARE.

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,325 +0,0 @@
-"""
-Agent command for YouTube workflow
-"""
-
-import asyncio
-import os
-import sys
-from typing import List, Optional
-
-import click
-import colorful
-from lhotse.utils import Pathlike
-
-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
-    except ImportError:
-        missing_deps.append('google-genai')
-
-    try:
-        import yt_dlp
-    except ImportError:
-        missing_deps.append('yt-dlp')
-
-    try:
-        from dotenv import load_dotenv
-    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,
-        )
-    )