merleau 0.1.1__py3-none-any.whl → 0.3.0__py3-none-any.whl

merleau/__init__.py

@@ -0,0 +1,3 @@
+"""Merleau - Video analysis using Google's Gemini API."""
+
+__version__ = "0.3.0"

merleau/cli.py

@@ -0,0 +1,215 @@
+"""Command-line interface for Merleau video analysis."""
+
+import argparse
+import os
+import sys
+import time
+from dataclasses import dataclass
+from typing import Callable, Optional
+
+from dotenv import load_dotenv
+from google import genai
+
+
+@dataclass
+class AnalysisResult:
+    """Result from video analysis."""
+    text: str
+    prompt_tokens: int
+    response_tokens: int
+    total_tokens: int
+    input_cost: float
+    output_cost: float
+    total_cost: float
+
+
+def wait_for_processing(client, file, on_progress: Optional[Callable] = None):
+    """Wait for file to finish processing."""
+    while file.state.name == "PROCESSING":
+        if on_progress:
+            on_progress()
+        else:
+            print(".", end="", flush=True)
+        time.sleep(2)
+        file = client.files.get(name=file.name)
+    if not on_progress:
+        print()
+    return file
+
+
+def calculate_cost(usage):
+    """Calculate cost from usage metadata."""
+    # Gemini 2.5 Flash pricing (as of 2025):
+    # Input: $0.15 per 1M tokens (text/image), $0.075 per 1M tokens for video
+    # Output: $0.60 per 1M tokens, $3.50 for thinking tokens
+    input_cost = (usage.prompt_token_count / 1_000_000) * 0.15
+    output_cost = (usage.candidates_token_count / 1_000_000) * 0.60
+    return input_cost, output_cost, input_cost + output_cost
+
+
+def print_usage(usage):
+    """Print token usage and cost estimation."""
+    print("\n--- Usage Information ---")
+    print(f"Prompt tokens: {usage.prompt_token_count}")
+    print(f"Response tokens: {usage.candidates_token_count}")
+    print(f"Total tokens: {usage.total_token_count}")
+
+    input_cost, output_cost, total_cost = calculate_cost(usage)
+    print(f"\nEstimated cost:")
+    print(f"  Input:  ${input_cost:.6f}")
+    print(f"  Output: ${output_cost:.6f}")
+    print(f"  Total:  ${total_cost:.6f}")
+
+
+def analyze_video(
+    video_path: str,
+    prompt: str = "Explain what happens in this video",
+    model: str = "gemini-2.5-flash",
+    api_key: Optional[str] = None,
+    on_upload: Optional[Callable[[str], None]] = None,
+    on_processing: Optional[Callable] = None,
+    on_analyzing: Optional[Callable] = None,
+) -> AnalysisResult:
+    """
+    Analyze a video file using Gemini.
+
+    Args:
+        video_path: Path to the video file
+        prompt: Analysis prompt
+        model: Gemini model to use
+        api_key: Optional API key (falls back to env var)
+        on_upload: Callback when upload completes (receives file URI)
+        on_processing: Callback during processing (called repeatedly)
+        on_analyzing: Callback when analysis starts
+
+    Returns:
+        AnalysisResult with text, tokens, and cost
+
+    Raises:
+        ValueError: If API key not found or file doesn't exist
+        RuntimeError: If file processing fails
+    """
+    load_dotenv()
+
+    api_key = api_key or os.getenv("GEMINI_API_KEY")
+    if not api_key:
+        raise ValueError("GEMINI_API_KEY not found in environment or .env file")
+
+    if not os.path.exists(video_path):
+        raise ValueError(f"Video file not found: {video_path}")
+
+    client = genai.Client(api_key=api_key)
+
+    # Upload video
+    myfile = client.files.upload(file=video_path)
+    if on_upload:
+        on_upload(myfile.uri)
+
+    # Wait for processing
+    myfile = wait_for_processing(client, myfile, on_progress=on_processing)
+
+    if myfile.state.name == "FAILED":
+        raise RuntimeError("File processing failed")
+
+    # Generate analysis
+    if on_analyzing:
+        on_analyzing()
+
+    response = client.models.generate_content(
+        model=model,
+        contents=[myfile, prompt]
+    )
+
+    # Extract usage info
+    usage = response.usage_metadata
+    input_cost, output_cost, total_cost = calculate_cost(usage)
+
+    return AnalysisResult(
+        text=response.text,
+        prompt_tokens=usage.prompt_token_count,
+        response_tokens=usage.candidates_token_count,
+        total_tokens=usage.total_token_count,
+        input_cost=input_cost,
+        output_cost=output_cost,
+        total_cost=total_cost,
+    )
+
+
+def analyze(video_path, prompt, model, show_cost):
+    """Analyze a video file using Gemini (CLI wrapper)."""
+    try:
+        print(f"Uploading video: {video_path}")
+
+        def on_upload(uri):
+            print(f"Upload complete. File URI: {uri}")
+            print("Waiting for file to be processed...", end="")
+
+        def on_processing():
+            print(".", end="", flush=True)
+
+        def on_analyzing():
+            print()
+            print(f"\nAnalyzing video with {model}...")
+
+        result = analyze_video(
+            video_path=video_path,
+            prompt=prompt,
+            model=model,
+            on_upload=on_upload,
+            on_processing=on_processing,
+            on_analyzing=on_analyzing,
+        )
+
+        print("\n--- Video Analysis ---")
+        print(result.text)
+
+        if show_cost:
+            print("\n--- Usage Information ---")
+            print(f"Prompt tokens: {result.prompt_tokens}")
+            print(f"Response tokens: {result.response_tokens}")
+            print(f"Total tokens: {result.total_tokens}")
+            print(f"\nEstimated cost:")
+            print(f"  Input:  ${result.input_cost:.6f}")
+            print(f"  Output: ${result.output_cost:.6f}")
+            print(f"  Total:  ${result.total_cost:.6f}")
+
+    except ValueError as e:
+        print(f"Error: {e}", file=sys.stderr)
+        sys.exit(1)
+    except RuntimeError as e:
+        print(f"Error: {e}", file=sys.stderr)
+        sys.exit(1)
+
+
+def main():
+    """Main entry point for the CLI."""
+    parser = argparse.ArgumentParser(
+        prog="ponty",
+        description="Analyze videos using Google's Gemini API"
+    )
+    parser.add_argument(
+        "video",
+        help="Path to the video file to analyze"
+    )
+    parser.add_argument(
+        "-p", "--prompt",
+        default="Explain what happens in this video",
+        help="Prompt for the analysis (default: 'Explain what happens in this video')"
+    )
+    parser.add_argument(
+        "-m", "--model",
+        default="gemini-2.5-flash",
+        help="Gemini model to use (default: gemini-2.5-flash)"
+    )
+    parser.add_argument(
+        "--no-cost",
+        action="store_true",
+        help="Hide usage and cost information"
+    )
+
+    args = parser.parse_args()
+    analyze(args.video, args.prompt, args.model, show_cost=not args.no_cost)
+
+
+if __name__ == "__main__":
+    main()

{merleau-0.1.1.dist-info → merleau-0.3.0.dist-info}/METADATA

@@ -1,7 +1,9 @@
 Metadata-Version: 2.4
 Name: merleau
-Version: 0.1.1
+Version: 0.3.0
 Summary: Video analysis using Google's Gemini 2.5 Flash API
 Requires-Python: >=3.10
 Requires-Dist: google-genai
 Requires-Dist: python-dotenv
+Provides-Extra: web
+Requires-Dist: streamlit>=1.30.0; extra == 'web'

merleau-0.3.0.dist-info/RECORD

@@ -0,0 +1,6 @@
+merleau/__init__.py,sha256=zV-fMC9hX4Uq5Kb4Jegk84CouKUslY58WkdBNCMMVck,81
+merleau/cli.py,sha256=1NofguSaSdB1RQpSb9Po6BudhQp6Q-3vYiGNky6vbmo,6396
+merleau-0.3.0.dist-info/METADATA,sha256=XKLO1o4-g-sK2xFm38CB33NRzpQk5c7mc5sSfi6Ry8g,261
+merleau-0.3.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+merleau-0.3.0.dist-info/entry_points.txt,sha256=pMCcADIqyZbK-lQ5d2iLKIWqNeeY3pAQDHI1IBmtdEQ,43
+merleau-0.3.0.dist-info/RECORD,,

{merleau-0.1.1.dist-info → merleau-0.3.0.dist-info}/WHEEL

@@ -1,5 +1,4 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.10.2)
+Generator: hatchling 1.28.0
 Root-Is-Purelib: true
 Tag: py3-none-any
-

merleau-0.3.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+ponty = merleau.cli:main

merleau-0.1.1.dist-info/RECORD

@@ -1,4 +0,0 @@
-merleau-0.1.1.dist-info/METADATA,sha256=fXz21-WXC04ec7uVE7V6f4uUkX_udosyBOC5qmXbUBQ,192
-merleau-0.1.1.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-merleau-0.1.1.dist-info/top_level.txt,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-merleau-0.1.1.dist-info/RECORD,,

merleau-0.1.1.dist-info/top_level.txt

@@ -1 +0,0 @@
-