emdash-cli 0.1.25__py3-none-any.whl → 0.1.30__py3-none-any.whl

emdash_cli/client.py

@@ -52,6 +52,7 @@ class EmdashClient:
         session_id: Optional[str] = None,
         max_iterations: int = _get_max_iterations(),
         options: Optional[dict] = None,
+        images: Optional[list[dict]] = None,
     ) -> Iterator[str]:
         """Stream agent chat response via SSE.
 
@@ -61,6 +62,7 @@ class EmdashClient:
             session_id: Session ID for continuity (optional)
             max_iterations: Max agent iterations
             options: Additional options (mode, save, no_graph_tools, etc.)
+            images: List of images [{"data": base64_str, "format": "png"}]
 
         Yields:
             SSE lines from the response
@@ -87,6 +89,8 @@ class EmdashClient:
             payload["model"] = model
         if session_id:
             payload["session_id"] = session_id
+        if images:
+            payload["images"] = images
 
         try:
             with self._client.stream(
@@ -105,17 +109,21 @@ class EmdashClient:
         self,
         session_id: str,
         message: str,
+        images: Optional[list[dict]] = None,
     ) -> Iterator[str]:
         """Continue an existing agent session.
 
         Args:
             session_id: Existing session ID
             message: Continuation message
+            images: List of images [{"data": base64_str, "format": "png"}]
 
         Yields:
             SSE lines from the response
         """
         payload = {"message": message}
+        if images:
+            payload["images"] = images
 
         try:
             with self._client.stream(

emdash_cli/clipboard.py

@@ -0,0 +1,123 @@
+"""Clipboard utilities for image handling."""
+
+import base64
+import io
+from typing import Optional, Tuple
+
+
+def get_clipboard_image() -> Optional[Tuple[str, str]]:
+    """Get image from clipboard if available.
+
+    Returns:
+        Tuple of (base64_data, format) if image found, None otherwise.
+    """
+    try:
+        from PIL import ImageGrab, Image
+
+        # Try to grab image from clipboard
+        image = ImageGrab.grabclipboard()
+
+        if image is None:
+            return None
+
+        # Handle list of file paths (Windows)
+        if isinstance(image, list):
+            # It's a list of file paths
+            if image and isinstance(image[0], str):
+                try:
+                    image = Image.open(image[0])
+                except Exception:
+                    return None
+            else:
+                return None
+
+        # Convert to PNG bytes
+        if isinstance(image, Image.Image):
+            buffer = io.BytesIO()
+            # Convert to RGB if necessary (for RGBA images)
+            if image.mode in ('RGBA', 'LA') or (image.mode == 'P' and 'transparency' in image.info):
+                # Keep as PNG to preserve transparency
+                image.save(buffer, format='PNG')
+                img_format = 'png'
+            else:
+                # Convert to JPEG for smaller size
+                if image.mode != 'RGB':
+                    image = image.convert('RGB')
+                image.save(buffer, format='JPEG', quality=85)
+                img_format = 'jpeg'
+
+            buffer.seek(0)
+            base64_data = base64.b64encode(buffer.read()).decode('utf-8')
+            return base64_data, img_format
+
+    except ImportError:
+        # PIL not available
+        return None
+    except Exception:
+        # Any other error (no clipboard access, etc.)
+        return None
+
+    return None
+
+
+def get_image_from_path(path: str) -> Optional[Tuple[str, str]]:
+    """Load image from file path.
+
+    Args:
+        path: Path to image file
+
+    Returns:
+        Tuple of (base64_data, format) if successful, None otherwise.
+    """
+    try:
+        from PIL import Image
+
+        image = Image.open(path)
+        buffer = io.BytesIO()
+
+        # Determine format from file extension
+        ext = path.lower().split('.')[-1]
+        if ext in ('jpg', 'jpeg'):
+            if image.mode != 'RGB':
+                image = image.convert('RGB')
+            image.save(buffer, format='JPEG', quality=85)
+            img_format = 'jpeg'
+        elif ext == 'png':
+            image.save(buffer, format='PNG')
+            img_format = 'png'
+        elif ext == 'gif':
+            image.save(buffer, format='GIF')
+            img_format = 'gif'
+        elif ext == 'webp':
+            image.save(buffer, format='WEBP')
+            img_format = 'webp'
+        else:
+            # Default to PNG
+            image.save(buffer, format='PNG')
+            img_format = 'png'
+
+        buffer.seek(0)
+        base64_data = base64.b64encode(buffer.read()).decode('utf-8')
+        return base64_data, img_format
+
+    except Exception:
+        return None
+
+
+def get_image_dimensions(base64_data: str) -> Optional[Tuple[int, int]]:
+    """Get dimensions of base64-encoded image.
+
+    Args:
+        base64_data: Base64-encoded image data
+
+    Returns:
+        Tuple of (width, height) if successful, None otherwise.
+    """
+    try:
+        from PIL import Image
+
+        image_bytes = base64.b64decode(base64_data)
+        image = Image.open(io.BytesIO(image_bytes))
+        return image.size
+    except Exception:
+        return None

emdash_cli/commands/agent.py

@@ -465,12 +465,15 @@ def _run_interactive(
     current_mode = AgentMode(options.get("mode", "code"))
     session_id = None
     current_spec = None
+    # Attached images for next message
+    attached_images: list[dict] = []
 
     # Style for prompt
     PROMPT_STYLE = Style.from_dict({
         "prompt.mode.plan": "#ffcc00 bold",
         "prompt.mode.code": "#00cc66 bold",
         "prompt.prefix": "#888888",
+        "prompt.image": "#00ccff",
         "completion-menu": "bg:#1a1a2e #ffffff",
         "completion-menu.completion": "bg:#1a1a2e #ffffff",
         "completion-menu.completion.current": "bg:#4a4a6e #ffffff bold",
@@ -517,6 +520,22 @@ def _run_interactive(
         """Insert a newline character with Alt+Enter or Ctrl+J."""
         event.current_buffer.insert_text("\n")
 
+    @kb.add("c-v")  # Ctrl+V to paste (check for images)
+    def paste_with_image_check(event):
+        """Paste text or attach image from clipboard."""
+        nonlocal attached_images
+        from ..clipboard import get_clipboard_image
+
+        # Try to get image from clipboard
+        image_data = get_clipboard_image()
+        if image_data:
+            base64_data, img_format = image_data
+            attached_images.append({"data": base64_data, "format": img_format})
+            console.print(f"[green]📎 Image attached[/green] [dim]({img_format})[/dim]")
+        else:
+            # No image, do normal paste
+            event.current_buffer.paste_clipboard_data(event.app.clipboard.get_data())
+
     session = PromptSession(
         history=history,
         completer=SlashCommandCompleter(),
@@ -528,14 +547,14 @@ def _run_interactive(
     )
 
     def get_prompt():
-        """Get formatted prompt based on current mode."""
-        mode_colors = {
-            AgentMode.PLAN: "class:prompt.mode.plan",
-            AgentMode.CODE: "class:prompt.mode.code",
-        }
-        mode_name = current_mode.value
-        color_class = mode_colors.get(current_mode, "class:prompt.mode.code")
-        return [(color_class, f"[{mode_name}]"), ("", " "), ("class:prompt.prefix", "> ")]
+        """Get formatted prompt."""
+        nonlocal attached_images
+        parts = []
+        # Add image indicator if images attached
+        if attached_images:
+            parts.append(("class:prompt.image", f"📎{len(attached_images)} "))
+        parts.append(("class:prompt.prefix", "> "))
+        return parts
 
     def show_help():
         """Show available commands."""
@@ -695,14 +714,13 @@ def _run_interactive(
     except Exception:
         pass
 
+    # Welcome banner
     console.print()
-    console.print(f"[bold cyan]EmDash Agent[/bold cyan] [dim]v{__version__}[/dim]")
-    console.print(f"Mode: [bold]{current_mode.value}[/bold] | Model: [dim]{model or 'default'}[/dim]")
+    console.print(f"[bold cyan]Emdash Code[/bold cyan] [dim]v{__version__}[/dim]")
     if git_repo:
-        console.print(f"Repo: [bold green]{git_repo}[/bold green] | Path: [dim]{cwd}[/dim]")
+        console.print(f"[dim]Repo:[/dim] [bold green]{git_repo}[/bold green] [dim]| Mode:[/dim] [bold]{current_mode.value}[/bold] [dim]| Model:[/dim] {model or 'default'}")
     else:
-        console.print(f"Path: [dim]{cwd}[/dim]")
-    console.print("Type your task or [cyan]/help[/cyan] for commands. Use Ctrl+C to exit.")
+        console.print(f"[dim]Mode:[/dim] [bold]{current_mode.value}[/bold] [dim]| Model:[/dim] {model or 'default'}")
     console.print()
 
     while True:
@@ -731,16 +749,25 @@ def _run_interactive(
 
             # Run agent with current mode
             try:
+                # Prepare images for API call
+                images_to_send = attached_images if attached_images else None
+
                 if session_id:
-                    stream = client.agent_continue_stream(session_id, user_input)
+                    stream = client.agent_continue_stream(
+                        session_id, user_input, images=images_to_send
+                    )
                 else:
                     stream = client.agent_chat_stream(
                         message=user_input,
                         model=model,
                         max_iterations=max_iterations,
                         options=request_options,
+                        images=images_to_send,
                     )
 
+                # Clear attached images after sending
+                attached_images = []
+
                 # Render the stream and capture any spec output
                 result = _render_with_interrupt(renderer, stream)
 

emdash_cli/sse_renderer.py

@@ -611,7 +611,6 @@ class SSERenderer:
         entities_found = adding.get("entities_found", 0)
         context_tokens = adding.get("context_tokens", 0)
         context_breakdown = adding.get("context_breakdown", {})
-        largest_messages = adding.get("largest_messages", [])
 
         # Get reading stats
         item_count = reading.get("item_count", 0)
@@ -636,15 +635,6 @@ class SSERenderer:
             if breakdown_parts:
                 self.console.print(f"  [dim]Breakdown: {' | '.join(breakdown_parts)}[/dim]")
 
-        # Show largest messages (context hogs)
-        if largest_messages:
-            self.console.print(f"  [yellow]Largest messages:[/yellow]")
-            for msg in largest_messages[:5]:
-                label = msg.get("label", "unknown")
-                tokens = msg.get("tokens", 0)
-                preview = msg.get("preview", "")[:50].replace("\n", " ")
-                self.console.print(f"    [dim]{tokens:,} tokens[/dim] - {label}: {preview}...")
-
         # Show other stats
         stats = []
         if step_count > 0:

{emdash_cli-0.1.25.dist-info → emdash_cli-0.1.30.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: emdash-cli
-Version: 0.1.25
+Version: 0.1.30
 Summary: EmDash CLI - Command-line interface for code intelligence
 Author: Em Dash Team
 Requires-Python: >=3.10,<4.0
@@ -10,8 +10,10 @@ Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Classifier: Programming Language :: Python :: 3.14
+Provides-Extra: images
 Requires-Dist: click (>=8.1.7,<9.0.0)
-Requires-Dist: emdash-core (>=0.1.25)
+Requires-Dist: emdash-core (>=0.1.30)
 Requires-Dist: httpx (>=0.25.0)
+Requires-Dist: pillow (>=10.0.0) ; extra == "images"
 Requires-Dist: prompt_toolkit (>=3.0.43,<4.0.0)
 Requires-Dist: rich (>=13.7.0)

{emdash_cli-0.1.25.dist-info → emdash_cli-0.1.30.dist-info}/RECORD

@@ -1,7 +1,8 @@
 emdash_cli/__init__.py,sha256=Rnn2O7B8OCEKlVtNRbWOU2-GN75_KLmhEJgOZzY-KwE,232
-emdash_cli/client.py,sha256=sPgX2CEfiHc4TXN9TYqFhc8DVKgPyk20Efm66KQAOhE,16761
+emdash_cli/client.py,sha256=aQO_wF4XQaqig6RWhTA3FIJslweo7LsZHCk_GBVGdvw,17117
+emdash_cli/clipboard.py,sha256=hcg5sbIhbixqzpJdonoFLGBlSo2AKjplNrWy5PGnqaY,3564
 emdash_cli/commands/__init__.py,sha256=D9edXBHm69tueUtE4DggTA1_Yjsl9YZaKjBVDY2D_gQ,712
-emdash_cli/commands/agent.py,sha256=RCgSYW6OBx3UFGVAM_byhyrtBKlsXN2xfJaT3w51uDY,27735
+emdash_cli/commands/agent.py,sha256=dLNNyJNL9uzZcBapzFghdjb-Xa27PglpbFNAMPGa3qc,28772
 emdash_cli/commands/analyze.py,sha256=c9ztbv0Ra7g2AlDmMOy-9L51fDVuoqbuzxnRfomoFIQ,4403
 emdash_cli/commands/auth.py,sha256=SpWdqO1bJCgt4x1B4Pr7hNOucwTuBFJ1oGPOzXtvwZM,3816
 emdash_cli/commands/db.py,sha256=nZK7gLDVE2lAQVYrMx6Swscml5OAtkbg-EcSNSvRIlA,2922
@@ -21,8 +22,8 @@ emdash_cli/commands/team.py,sha256=K1-IJg6iG-9HMF_3JmpNDlNs1PYbb-ThFHU9KU_jKRo,1
 emdash_cli/keyboard.py,sha256=haYYAuhYGtdjomzhIFy_3Z3eN3BXfMdb4uRQjwB0tbk,4593
 emdash_cli/main.py,sha256=c-faWp-jzf9a0BbXhVoPvPQfGWSryXpYfswehqZCYPM,2593
 emdash_cli/server_manager.py,sha256=RrLteSHUmcFV4cyHJAEmgM9qHru2mJS08QNLWno6Y3Y,7051
-emdash_cli/sse_renderer.py,sha256=PEbD53ZohMp9yvii_1ELGwVKb8nnA_n17jICeaURkuY,23738
-emdash_cli-0.1.25.dist-info/METADATA,sha256=RFuFUhHJlRcpfzwztnQXTWheB853OVlyWd1tAIzRKsE,662
-emdash_cli-0.1.25.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
-emdash_cli-0.1.25.dist-info/entry_points.txt,sha256=31CuYD0k-tM8csFWDunc-JoZTxXaifj3oIXz4V0p6F0,122
-emdash_cli-0.1.25.dist-info/RECORD,,
+emdash_cli/sse_renderer.py,sha256=aDOoHKglOkaYEXuKg937mH6yFPDxjU7Rqa_-APyM9Dc,23215
+emdash_cli-0.1.30.dist-info/METADATA,sha256=czNXf-GzyfHKHe8g8Dfd7Bm7kI189H-VLRpFDVgtejc,738
+emdash_cli-0.1.30.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
+emdash_cli-0.1.30.dist-info/entry_points.txt,sha256=31CuYD0k-tM8csFWDunc-JoZTxXaifj3oIXz4V0p6F0,122
+emdash_cli-0.1.30.dist-info/RECORD,,