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

lattifai/bin/align.py

@@ -1,6 +1,5 @@
 import asyncio
 import os
-from pathlib import Path
 
 import click
 import colorful
@@ -13,74 +12,74 @@ from lattifai.io import INPUT_SUBTITLE_FORMATS, OUTPUT_SUBTITLE_FORMATS
 
 @cli.command()
 @click.option(
-    '-F',
-    '--input_format',
-    '--input-format',
+    "-F",
+    "--input_format",
+    "--input-format",
     type=click.Choice(INPUT_SUBTITLE_FORMATS, case_sensitive=False),
-    default='auto',
-    help='Input subtitle format.',
+    default="auto",
+    help="Input subtitle format.",
 )
 @click.option(
-    '-S',
-    '--split-sentence',
-    '--split_sentence',
+    "-S",
+    "--split-sentence",
+    "--split_sentence",
     is_flag=True,
     default=False,
-    help='Re-segment subtitles by semantics.',
+    help="Re-segment subtitles by semantics.",
 )
 @click.option(
-    '-W',
-    '--word-level',
-    '--word_level',
+    "-W",
+    "--word-level",
+    "--word_level",
     is_flag=True,
     default=False,
-    help='Include word-level alignment timestamps in output (for JSON, TextGrid, and subtitle formats).',
+    help="Include word-level alignment timestamps in output (for JSON, TextGrid, and subtitle formats).",
 )
 @click.option(
-    '-D',
-    '--device',
-    type=click.Choice(['cpu', 'cuda', 'mps'], case_sensitive=False),
-    default='cpu',
-    help='Device to use for inference.',
+    "-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',
+    "-M",
+    "--model-name-or-path",
+    "--model_name_or_path",
     type=str,
-    default='Lattifai/Lattice-1-Alpha',
-    help='Model name or path for alignment.',
+    default="Lattifai/Lattice-1-Alpha",
+    help="Model name or path for alignment.",
 )
 @click.option(
-    '-K',
-    '-L',
-    '--api-key',
-    '--api_key',
+    "-K",
+    "-L",
+    "--api-key",
+    "--api_key",
     type=str,
     default=None,
-    help='API key for LattifAI.',
+    help="API key for LattifAI.",
 )
 @click.argument(
-    'input_media_path',
+    "input_media_path",
     type=click.Path(exists=True, dir_okay=False),
 )
 @click.argument(
-    'input_subtitle_path',
+    "input_subtitle_path",
     type=click.Path(exists=True, dir_okay=False),
 )
 @click.argument(
-    'output_subtitle_path',
+    "output_subtitle_path",
     type=click.Path(allow_dash=True),
 )
 def align(
     input_media_path: Pathlike,
     input_subtitle_path: Pathlike,
     output_subtitle_path: Pathlike,
-    input_format: str = 'auto',
+    input_format: str = "auto",
     split_sentence: bool = False,
     word_level: bool = False,
-    device: str = 'cpu',
-    model_name_or_path: str = 'Lattifai/Lattice-1-Alpha',
+    device: str = "cpu",
+    model_name_or_path: str = "Lattifai/Lattice-1-Alpha",
     api_key: str = None,
 ):
     """
@@ -96,129 +95,129 @@ def align(
             return_details=word_level,
             output_subtitle_path=output_subtitle_path,
         )
-        click.echo(colorful.green(f'✅ Alignment completed successfully: {output_subtitle_path}'))
+        click.echo(colorful.green(f"✅ Alignment completed successfully: {output_subtitle_path}"))
     except Exception as e:
         from lattifai.errors import LattifAIError
 
         # Display error message
         if isinstance(e, LattifAIError):
-            click.echo(colorful.red('❌ Alignment failed:'))
+            click.echo(colorful.red("❌ Alignment failed:"))
             click.echo(e.get_message())
             # Show support info
             click.echo(e.get_support_info())
         else:
-            click.echo(colorful.red(f'❌ Alignment failed: {str(e)}'))
+            click.echo(colorful.red(f"❌ Alignment failed: {str(e)}"))
 
-        raise click.ClickException('Alignment failed')
+        raise click.ClickException("Alignment failed")
 
 
 @cli.command()
 @click.option(
-    '-M',
-    '--media-format',
-    '--media_format',
+    "-M",
+    "--media-format",
+    "--media_format",
     type=click.Choice(
         [
             # Audio formats
-            'mp3',
-            'wav',
-            'm4a',
-            'aac',
-            'flac',
-            'ogg',
-            'opus',
-            'aiff',
+            "mp3",
+            "wav",
+            "m4a",
+            "aac",
+            "flac",
+            "ogg",
+            "opus",
+            "aiff",
             # Video formats
-            'mp4',
-            'webm',
-            'mkv',
-            'avi',
-            'mov',
+            "mp4",
+            "webm",
+            "mkv",
+            "avi",
+            "mov",
         ],
         case_sensitive=False,
     ),
-    default='mp3',
-    help='Media format for YouTube download (audio or video).',
+    default="mp3",
+    help="Media format for YouTube download (audio or video).",
 )
 @click.option(
-    '-S',
-    '--split-sentence',
-    '--split_sentence',
+    "-S",
+    "--split-sentence",
+    "--split_sentence",
     is_flag=True,
     default=False,
-    help='Re-segment subtitles by semantics.',
+    help="Re-segment subtitles by semantics.",
 )
 @click.option(
-    '-W',
-    '--word-level',
-    '--word_level',
+    "-W",
+    "--word-level",
+    "--word_level",
     is_flag=True,
     default=False,
-    help='Include word-level alignment timestamps in output (for JSON, TextGrid, and subtitle formats).',
+    help="Include word-level alignment timestamps in output (for JSON, TextGrid, and subtitle formats).",
 )
 @click.option(
-    '-O',
-    '--output-dir',
-    '--output_dir',
+    "-O",
+    "--output-dir",
+    "--output_dir",
     type=click.Path(file_okay=False, dir_okay=True, writable=True),
-    default='.',
-    help='Output directory (default: current directory).',
+    default=".",
+    help="Output directory (default: current directory).",
 )
 @click.option(
-    '-D',
-    '--device',
-    type=click.Choice(['cpu', 'cuda', 'mps'], case_sensitive=False),
-    default='cpu',
-    help='Device to use for inference.',
+    "-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',
+    "-M",
+    "--model-name-or-path",
+    "--model_name_or_path",
     type=str,
-    default='Lattifai/Lattice-1-Alpha',
-    help='Model name or path for alignment.',
+    default="Lattifai/Lattice-1-Alpha",
+    help="Model name or path for alignment.",
 )
 @click.option(
-    '-K',
-    '-L',
-    '--api-key',
-    '--api_key',
+    "-K",
+    "-L",
+    "--api-key",
+    "--api_key",
     type=str,
     default=None,
-    help='API key for LattifAI.',
+    help="API key for LattifAI.",
 )
 @click.option(
-    '-G',
-    '--gemini-api-key',
-    '--gemini_api_key',
+    "-G",
+    "--gemini-api-key",
+    "--gemini_api_key",
     type=str,
     default=None,
-    help='Gemini API key for transcription fallback when subtitles are unavailable.',
+    help="Gemini API key for transcription fallback when subtitles are unavailable.",
 )
 @click.option(
-    '-F',
-    '--output-format',
-    '--output_format',
+    "-F",
+    "--output-format",
+    "--output_format",
     type=click.Choice(OUTPUT_SUBTITLE_FORMATS, case_sensitive=False),
-    default='vtt',
-    help='Subtitle output format.',
+    default="vtt",
+    help="Subtitle output format.",
 )
 @click.argument(
-    'yt_url',
+    "yt_url",
     type=str,
 )
 def youtube(
     yt_url: str,
-    media_format: str = 'mp3',
+    media_format: str = "mp3",
     split_sentence: bool = False,
     word_level: bool = False,
-    output_dir: str = '.',
-    device: str = 'cpu',
-    model_name_or_path: str = 'Lattifai/Lattice-1-Alpha',
+    output_dir: str = ".",
+    device: str = "cpu",
+    model_name_or_path: str = "Lattifai/Lattice-1-Alpha",
     api_key: str = None,
     gemini_api_key: str = None,
-    output_format: str = 'vtt',
+    output_format: str = "vtt",
 ):
     """
     Download media and subtitles from YouTube for further alignment.
@@ -227,7 +226,7 @@ def youtube(
     from lattifai.workflows.youtube import YouTubeDownloader, YouTubeSubtitleAgent
 
     # Get Gemini API key
-    gemini_key = gemini_api_key or os.getenv('GEMINI_API_KEY')
+    gemini_key = gemini_api_key or os.getenv("GEMINI_API_KEY")
 
     async def _process():
         # Initialize components with their configuration (only config, not runtime params)
@@ -258,27 +257,27 @@ def youtube(
         result = asyncio.run(_process())
 
         # Display results
-        click.echo(colorful.green('✅ Processing completed!'))
+        click.echo(colorful.green("✅ Processing completed!"))
         click.echo()
 
         # Show metadata
-        metadata = result.get('metadata', {})
+        metadata = result.get("metadata", {})
         if metadata:
             click.echo(f'🎬    Title: {metadata.get("title", "Unknown")}')
             click.echo(f'⏱️  Duration: {metadata.get("duration", 0)} seconds')
             click.echo()
 
         # Show exported files
-        exported_files = result.get('exported_files', {})
+        exported_files = result.get("exported_files", {})
         if exported_files:
-            click.echo(colorful.green('📄 Generated subtitle files:'))
+            click.echo(colorful.green("📄 Generated subtitle files:"))
             for format_name, file_path in exported_files.items():
-                click.echo(f'  {format_name}: {file_path}')
+                click.echo(f"  {format_name}: {file_path}")
             click.echo()
 
         # Show subtitle count
-        subtitle_count = result.get('subtitle_count', 0)
-        click.echo(f'📝 Generated {subtitle_count} subtitle segments')
+        subtitle_count = result.get("subtitle_count", 0)
+        click.echo(f"📝 Generated {subtitle_count} subtitle segments")
 
     except Exception as e:
         from lattifai.errors import LattifAIError
@@ -286,11 +285,11 @@ def youtube(
         # 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('❌ Failed to process YouTube URL:'))
+            click.echo(colorful.red("❌ Failed to process YouTube URL:"))
             click.echo(e.get_message())
             # Show support info once at the end
             click.echo(e.get_support_info())
         else:
-            click.echo(colorful.red(f'❌ Failed to process YouTube URL: {str(e)}'))
+            click.echo(colorful.red(f"❌ Failed to process YouTube URL: {str(e)}"))
 
-        raise click.ClickException('Processing failed')
+        raise click.ClickException("Processing failed")

lattifai/bin/cli_base.py

@@ -15,11 +15,11 @@ def cli():
     load_dotenv(find_dotenv(usecwd=True))
 
     logging.basicConfig(
-        format='%(asctime)s %(levelname)s [%(filename)s:%(lineno)d] %(message)s',
+        format="%(asctime)s %(levelname)s [%(filename)s:%(lineno)d] %(message)s",
         level=logging.INFO,
     )
 
     import os
 
-    os.environ['KMP_DUPLICATE_LIB_OK'] = 'TRUE'
-    os.environ['TOKENIZERS_PARALLELISM'] = 'FALSE'
+    os.environ["KMP_DUPLICATE_LIB_OK"] = "TRUE"
+    os.environ["TOKENIZERS_PARALLELISM"] = "FALSE"

lattifai/bin/subtitle.py

@@ -16,11 +16,11 @@ def subtitle():
 
 @subtitle.command()
 @click.argument(
-    'input_subtitle_path',
+    "input_subtitle_path",
     type=click.Path(exists=True, dir_okay=False),
 )
 @click.argument(
-    'output_subtitle_path',
+    "output_subtitle_path",
     type=click.Path(allow_dash=True),
 )
 def convert(
@@ -30,7 +30,7 @@ def convert(
     """
     Convert subtitle file to another format.
     """
-    if str(output_subtitle_path).lower().endswith('.TextGrid'.lower()):
+    if str(output_subtitle_path).lower().endswith(".TextGrid".lower()):
         from lattifai.io import SubtitleIO
 
         alignments = SubtitleIO.read(input_subtitle_path)
@@ -44,30 +44,30 @@ def convert(
 
 
 @subtitle.command()
-@click.argument('url', type=str, required=True)
+@click.argument("url", type=str, required=True)
 @click.option(
-    '--output-dir',
-    '--output_dir',
-    '-o',
+    "--output-dir",
+    "--output_dir",
+    "-o",
     type=click.Path(file_okay=False, dir_okay=True),
-    default='.',
-    help='Output directory for downloaded subtitle files (default: current directory).',
+    default=".",
+    help="Output directory for downloaded subtitle files (default: current directory).",
 )
 @click.option(
-    '--output-format',
-    '--output_format',
-    '-f',
-    type=click.Choice(SUBTITLE_FORMATS + ['best'], case_sensitive=False),
-    default='best',
-    help='Preferred subtitle format to download (default: best available).',
+    "--output-format",
+    "--output_format",
+    "-f",
+    type=click.Choice(SUBTITLE_FORMATS + ["best"], case_sensitive=False),
+    default="best",
+    help="Preferred subtitle format to download (default: best available).",
 )
-@click.option('--force-overwrite', '-F', is_flag=True, help='Overwrite existing files without prompting.')
+@click.option("--force-overwrite", "-F", is_flag=True, help="Overwrite existing files without prompting.")
 @click.option(
-    '--lang',
-    '-l',
-    '-L',
-    '--subtitle-lang',
-    '--subtitle_lang',
+    "--lang",
+    "-l",
+    "-L",
+    "--subtitle-lang",
+    "--subtitle_lang",
     type=str,
     help='Specific subtitle language/track to download (e.g., "en").',
 )
@@ -88,8 +88,8 @@ def download(
 
     # Validate URL format
     if not _is_valid_youtube_url(url):
-        click.echo(f'Error: Invalid YouTube URL format: {url}', err=True)
-        click.echo('Please provide a valid YouTube URL (e.g., https://www.youtube.com/watch?v=VIDEO_ID)', err=True)
+        click.echo(f"Error: Invalid YouTube URL format: {url}", err=True)
+        click.echo("Please provide a valid YouTube URL (e.g., https://www.youtube.com/watch?v=VIDEO_ID)", err=True)
         raise click.Abort()
 
     # Convert relative path to absolute
@@ -98,13 +98,13 @@ def download(
     # Create output directory if it doesn't exist
     output_path.mkdir(parents=True, exist_ok=True)
 
-    click.echo(f'Downloading subtitles from: {url}')
-    click.echo(f'          Output directory: {output_path}')
-    click.echo(f'         Preferred format: {output_format}')
+    click.echo(f"Downloading subtitles from: {url}")
+    click.echo(f"          Output directory: {output_path}")
+    click.echo(f"         Preferred format: {output_format}")
     if lang:
-        click.echo(f'       Subtitle language: {lang}')
+        click.echo(f"       Subtitle language: {lang}")
     else:
-        click.echo('       Subtitle language: All available')
+        click.echo("       Subtitle language: All available")
 
     # Initialize downloader and download
     downloader = YouTubeDownloader()
@@ -119,28 +119,28 @@ def download(
             )
 
             if result:
-                click.echo('✅ Subtitles downloaded successfully!')
+                click.echo("✅ Subtitles downloaded successfully!")
                 return result
             else:
-                click.echo('⚠️  No subtitles available for this video')
+                click.echo("⚠️  No subtitles available for this video")
                 return None
 
         except Exception as e:
-            click.echo(f'❌ Error downloading subtitles: {str(e)}', err=True)
+            click.echo(f"❌ Error downloading subtitles: {str(e)}", err=True)
             raise click.Abort()
 
     # Run the async function
     result = asyncio.run(download_subtitles())
 
     if result:
-        if result == 'gemini':
-            click.echo('✨ Gemini transcription selected (use the agent command to transcribe)')
+        if result == "gemini":
+            click.echo("✨ Gemini transcription selected (use the agent command to transcribe)")
         else:
-            click.echo(f'📄 Subtitle file saved to: {result}')
+            click.echo(f"📄 Subtitle file saved to: {result}")
 
 
 @subtitle.command()
-@click.argument('url', type=str, required=True)
+@click.argument("url", type=str, required=True)
 def list_subs(url: str):
     """
     List available subtitle tracks for a YouTube video.
@@ -152,11 +152,11 @@ def list_subs(url: str):
 
     # Validate URL format
     if not _is_valid_youtube_url(url):
-        click.echo(f'Error: Invalid YouTube URL format: {url}', err=True)
-        click.echo('Please provide a valid YouTube URL (e.g., https://www.youtube.com/watch?v=VIDEO_ID)', err=True)
+        click.echo(f"Error: Invalid YouTube URL format: {url}", err=True)
+        click.echo("Please provide a valid YouTube URL (e.g., https://www.youtube.com/watch?v=VIDEO_ID)", err=True)
         raise click.Abort()
 
-    click.echo(f'Listing available subtitles for: {url}')
+    click.echo(f"Listing available subtitles for: {url}")
 
     # Initialize downloader
     downloader = YouTubeDownloader()
@@ -166,20 +166,20 @@ def list_subs(url: str):
             result = await downloader.list_available_subtitles(url)
 
             if result:
-                click.echo('📋 Available subtitle tracks:')
+                click.echo("📋 Available subtitle tracks:")
                 for subtitle_info in result:
                     click.echo(f'  🎬 Language: {subtitle_info["language"]} - {subtitle_info["name"]}')
                     click.echo(f'     📄 Formats: {", ".join(subtitle_info["formats"])}')
                     click.echo()
 
-                click.echo('💡 To download a specific track, use:')
+                click.echo("💡 To download a specific track, use:")
                 click.echo(f'   lattifai subtitle download "{url}" --lang <language_code>')
                 click.echo('   Example: lattifai subtitle download "{}" --lang en-JkeT_87f4cc'.format(url))
             else:
-                click.echo('⚠️  No subtitles available for this video')
+                click.echo("⚠️  No subtitles available for this video")
 
         except Exception as e:
-            click.echo(f'❌ Error listing subtitles: {str(e)}', err=True)
+            click.echo(f"❌ Error listing subtitles: {str(e)}", err=True)
             raise click.Abort()
 
     # Run the async function
@@ -199,9 +199,9 @@ def _is_valid_youtube_url(url: str) -> bool:
     import re
 
     patterns = [
-        r'(?:youtube\.com/watch\?v=|youtu\.be/|youtube\.com/shorts/)([a-zA-Z0-9_-]{11})',
-        r'youtube\.com/embed/([a-zA-Z0-9_-]{11})',
-        r'youtube\.com/v/([a-zA-Z0-9_-]{11})',
+        r"(?:youtube\.com/watch\?v=|youtu\.be/|youtube\.com/shorts/)([a-zA-Z0-9_-]{11})",
+        r"youtube\.com/embed/([a-zA-Z0-9_-]{11})",
+        r"youtube\.com/v/([a-zA-Z0-9_-]{11})",
     ]
 
     for pattern in patterns: