cowork-dash 0.1.2__py3-none-any.whl

cowork_dash/file_utils.py

@@ -0,0 +1,226 @@
+"""File tree and file operations utilities."""
+
+import base64
+from pathlib import Path
+from typing import List, Dict, Tuple
+from dash import html
+
+
+TEXT_EXTENSIONS = {
+    ".py", ".js", ".ts", ".jsx", ".tsx", ".html", ".css", ".json", ".md", ".txt",
+    ".yaml", ".yml", ".toml", ".xml", ".csv", ".sh", ".bash", ".sql", ".env",
+    ".gitignore", ".dockerignore", ".cfg", ".ini", ".conf", ".log"
+}
+
+
+def is_text_file(filename: str) -> bool:
+    """Check if a file can be viewed as text."""
+    ext = Path(filename).suffix.lower()
+    return ext in TEXT_EXTENSIONS or ext == ""
+
+
+def build_file_tree(root: Path, workspace_root: Path, lazy_load: bool = True) -> List[Dict]:
+    """
+    Build file tree structure.
+
+    Args:
+        root: Directory to scan
+        workspace_root: Root workspace directory for relative paths
+        lazy_load: If True, only load immediate children (subdirs not expanded)
+
+    Returns:
+        List of file/folder items
+    """
+    items = []
+    try:
+        entries = sorted(root.iterdir(), key=lambda x: (not x.is_dir(), x.name.lower()))
+        for entry in entries:
+            if entry.name.startswith('.'):
+                continue
+            rel_path = str(entry.relative_to(workspace_root))
+            if entry.is_dir():
+                # Count immediate children to show if folder is empty
+                try:
+                    has_children = any(not item.name.startswith('.') for item in entry.iterdir())
+                except (PermissionError, OSError):
+                    has_children = False
+
+                items.append({
+                    "type": "folder",
+                    "name": entry.name,
+                    "path": rel_path,
+                    "has_children": has_children,
+                    # Only recursively load children if not lazy loading
+                    "children": [] if lazy_load else build_file_tree(entry, workspace_root, lazy_load=False)
+                })
+            else:
+                items.append({
+                    "type": "file",
+                    "name": entry.name,
+                    "path": rel_path,
+                    "viewable": is_text_file(entry.name)
+                })
+    except PermissionError:
+        pass
+    return items
+
+
+def load_folder_contents(folder_path: str, workspace_root: Path) -> List[Dict]:
+    """
+    Load contents of a specific folder (for lazy loading).
+
+    Args:
+        folder_path: Relative path to the folder from workspace root
+        workspace_root: Root workspace directory
+
+    Returns:
+        List of file/folder items in the specified folder
+    """
+    full_path = workspace_root / folder_path
+    return build_file_tree(full_path, workspace_root, lazy_load=True)
+
+
+def render_file_tree(items: List[Dict], colors: Dict, styles: Dict, level: int = 0, parent_path: str = "") -> List:
+    """Render file tree with collapsible folders using CSS classes for theming."""
+    components = []
+    indent = level * 15  # Scaled up indent
+
+    for item in items:
+        if item["type"] == "folder":
+            folder_id = item["path"].replace("/", "_").replace("\\", "_")
+            children = item.get("children", [])
+
+            # Folder header (clickable to expand/collapse)
+            components.append(
+                html.Div([
+                    html.Span(
+                        "▶",
+                        id={"type": "folder-icon", "path": folder_id},
+                        className="folder-icon",
+                        style={
+                            "marginRight": "5px",
+                            "fontSize": "10px",
+                            "transition": "transform 0.15s",
+                            "display": "inline-block",
+                        }
+                    ),
+                    html.Span(item["name"], className="folder-name", style={
+                        "fontWeight": "500",
+                        "fontSize": "14px",
+                    })
+                ],
+                id={"type": "folder-header", "path": folder_id},
+                **{"data-realpath": item["path"]},  # Store actual path for lazy loading
+                className="folder-header file-tree-item",
+                style={
+                    "display": "flex",
+                    "alignItems": "center",
+                    "padding": "5px 10px",
+                    "paddingLeft": f"{10 + indent}px",
+                    "cursor": "pointer",
+                    "userSelect": "none",
+                },
+                )
+            )
+
+            # Folder children (hidden by default) - always create even if empty
+            # Show different content based on whether children are loaded
+            has_children = item.get("has_children", True)
+
+            if children:
+                # Children are loaded, render them
+                child_content = render_file_tree(children, colors, styles, level + 1, item["path"])
+            elif not has_children:
+                # Folder is known to be empty
+                child_content = [
+                    html.Div("(empty)", className="file-tree-empty", style={
+                        "padding": "4px 10px",
+                        "paddingLeft": f"{25 + (level + 1) * 15}px",
+                        "fontSize": "12px",
+                        "fontStyle": "italic",
+                    })
+                ]
+            else:
+                # Children not yet loaded (lazy loading)
+                child_content = [
+                    html.Div("Loading...",
+                        id={"type": "folder-loading", "path": folder_id},
+                        className="file-tree-loading",
+                        style={
+                            "padding": "4px 10px",
+                            "paddingLeft": f"{25 + (level + 1) * 15}px",
+                            "fontSize": "12px",
+                            "fontStyle": "italic",
+                        }
+                    )
+                ]
+
+            components.append(
+                html.Div(
+                    child_content,
+                    id={"type": "folder-children", "path": folder_id},
+                    style={"display": "none"}  # Collapsed by default
+                )
+            )
+        else:
+            # File item
+            components.append(
+                html.Div(
+                    item["name"],
+                    id={"type": "file-item", "path": item["path"]},
+                    className="file-item file-tree-item",
+                    style={
+                        "fontSize": "14px",
+                        "padding": "5px 10px",
+                        "paddingLeft": f"{25 + indent}px",
+                        "cursor": "pointer",
+                    },
+                    **{"data-viewable": "true" if item["viewable"] else "false"}
+                )
+            )
+
+    return components
+
+
+def read_file_content(workspace_root: Path, path: str) -> Tuple[str, bool, str]:
+    """Read file content. Returns (content, is_text, error)."""
+    full_path = workspace_root / path
+    if not full_path.exists() or not full_path.is_file():
+        return None, False, "File not found"
+
+    if is_text_file(path):
+        try:
+            content = full_path.read_text(encoding="utf-8")
+            return content, True, None
+        except UnicodeDecodeError:
+            return None, False, "Binary file - cannot display"
+        except Exception as e:
+            return None, False, str(e)
+    else:
+        return None, False, "Binary file - download to view"
+
+
+def get_file_download_data(workspace_root: Path, path: str) -> Tuple[str, str, str]:
+    """Get file data for download. Returns (base64_content, filename, mime_type)."""
+    full_path = workspace_root / path
+    if not full_path.exists():
+        return None, None, None
+
+    try:
+        content = full_path.read_bytes()
+        b64 = base64.b64encode(content).decode('utf-8')
+
+        # Determine MIME type
+        ext = full_path.suffix.lower()
+        mime_types = {
+            ".txt": "text/plain", ".py": "text/x-python", ".js": "text/javascript",
+            ".json": "application/json", ".html": "text/html", ".css": "text/css",
+            ".md": "text/markdown", ".csv": "text/csv", ".xml": "text/xml",
+            ".pdf": "application/pdf", ".png": "image/png", ".jpg": "image/jpeg",
+            ".gif": "image/gif", ".zip": "application/zip",
+        }
+        mime = mime_types.get(ext, "application/octet-stream")
+
+        return b64, full_path.name, mime
+    except Exception:
+        return None, None, None

cowork_dash/layout.py

@@ -0,0 +1,250 @@
+"""Layout components for DeepAgent Dash."""
+
+from dash import html, dcc
+import dash_mantine_components as dmc
+from dash_iconify import DashIconify
+
+from .file_utils import build_file_tree, render_file_tree
+
+
+def create_layout(workspace_root, app_title, app_subtitle, colors, styles, agent):
+    """
+    Create the app layout with current configuration.
+
+    Args:
+        workspace_root: Path to workspace directory
+        app_title: Application title
+        app_subtitle: Application subtitle
+        colors: Color scheme dictionary
+        styles: Styles dictionary
+        agent: Agent instance (or None)
+
+    Returns:
+        Dash layout component
+    """
+    return dmc.MantineProvider(
+        id="mantine-provider",
+        forceColorScheme="light",
+        children=[
+            # State stores
+            dcc.Store(id="chat-history", data=[{
+                "role": "assistant",
+                "content": f"""This is your AI-powered workspace. I can help you write code, analyze files, create visualizations, and more.
+
+**Getting Started:**
+- Type a message below to chat with me
+- Browse files on the right (click to view, ↓ to download)
+- Switch to **Canvas** tab to see charts and diagrams I create
+
+Let's get started!"""
+            }]),
+            dcc.Store(id="pending-message", data=None),
+            dcc.Store(id="expanded-folders", data=[]),
+            dcc.Store(id="file-to-view", data=None),
+            dcc.Store(id="file-click-tracker", data={}),
+            dcc.Store(id="theme-store", data="light", storage_type="local"),
+            dcc.Download(id="file-download"),
+
+            # Interval for polling agent updates (disabled by default)
+            dcc.Interval(id="poll-interval", interval=250, disabled=True),
+
+            # File viewer modal
+            dmc.Modal(
+                id="file-modal",
+                title="",
+                size="xl",
+                children=[
+                    html.Div(id="modal-content"),
+                    html.Div([
+                        dmc.Button(
+                            "Download",
+                            id="modal-download-btn",
+                            variant="outline",
+                            color="blue",
+                            style={"marginTop": "16px"}
+                        )
+                    ], style={"textAlign": "right"})
+                ],
+                opened=False,
+            ),
+
+            html.Div([
+                # Compact Header
+                html.Header([
+                    html.Div([
+                        html.Div([
+                            html.H1(app_title or "DeepAgent Dash", id="app-title", style={
+                                "fontSize": "17px", "fontWeight": "600", "margin": "0",
+                            }),
+                            html.Span(app_subtitle or "AI-Powered Workspace", id="app-subtitle", style={
+                                "fontSize": "14px", "color": "var(--mantine-color-dimmed)", "marginLeft": "10px",
+                            })
+                        ], style={"display": "flex", "alignItems": "baseline"}),
+                        html.Div([
+                            dmc.ActionIcon(
+                                DashIconify(icon="radix-icons:moon", width=18),
+                                id="theme-toggle-btn",
+                                variant="subtle",
+                                color="gray",
+                                size="md",
+                                radius="sm",
+                                style={"marginRight": "8px"},
+                            ),
+                            html.Div(style={
+                                "width": "8px", "height": "8px",
+                                "borderRadius": "50%",
+                                "background": "var(--mantine-color-green-6)" if agent else "var(--mantine-color-red-6)",
+                                "marginRight": "5px",
+                            }, id="agent-status-indicator"),
+                            dmc.Text("Ready" if agent else "No Agent", size="sm", c="dimmed", id="agent-status-text")
+                        ], style={"display": "flex", "alignItems": "center"})
+                    ], style={
+                        "display": "flex", "justifyContent": "space-between",
+                        "alignItems": "center", "maxWidth": "1600px",
+                        "margin": "0 auto", "padding": "0 12px",
+                    })
+                ], id="header", style={
+                    "background": "var(--mantine-color-body)",
+                    "borderBottom": "1px solid var(--mantine-color-default-border)",
+                    "padding": "8px 0",
+                }),
+
+            # Main content
+            html.Main([
+                # Chat panel (no header)
+                html.Div([
+                    # Messages
+                    html.Div(id="chat-messages", style={
+                        "flex": "1", "overflowY": "auto", "padding": "15px",
+                        "display": "flex", "flexDirection": "column", "gap": "10px",
+                    }),
+
+                    # Compact Input
+                    html.Div([
+                        dcc.Upload(
+                            id="file-upload",
+                            children=dmc.ActionIcon(
+                                DashIconify(icon="radix-icons:plus", width=18),
+                                id="upload-plus",
+                                variant="default",
+                                size="md",
+                            ),
+                            style={"cursor": "pointer"},
+                            multiple=True
+                        ),
+                        dmc.TextInput(
+                            id="chat-input",
+                            placeholder="Type a message...",
+                            className="chat-input",
+                            style={"flex": "1"},
+                            size="md",
+                        ),
+                        dmc.Button("Send", id="send-btn", className="send-btn", size="md"),
+                    ], id="chat-input-area", style={
+                        "display": "flex", "gap": "8px", "padding": "10px 15px",
+                        "borderTop": "1px solid var(--mantine-color-default-border)",
+                        "background": "var(--mantine-color-body)",
+                    }),
+                    dmc.Text(id="upload-status", size="sm", c="dimmed", style={
+                        "padding": "0 15px 8px",
+                    }),
+                ], id="chat-panel", style={
+                    "flex": "1", "display": "flex", "flexDirection": "column",
+                    "background": "var(--mantine-color-body)", "minWidth": "0",
+                }),
+
+                # Resize handle
+                html.Div(id="resize-handle", className="resize-handle", style={
+                    "width": "3px",
+                    "cursor": "col-resize",
+                    "background": "transparent",
+                    "flexShrink": "0",
+                }),
+
+                # Sidebar (Files/Canvas toggle)
+                html.Div([
+                    # Compact header with toggle
+                    html.Div([
+                        dmc.SegmentedControl(
+                            id="sidebar-view-toggle",
+                            data=[
+                                {"value": "files", "label": "Files"},
+                                {"value": "canvas", "label": "Canvas"},
+                            ],
+                            value="files",
+                            size="sm",
+                        ),
+                        dmc.Group([
+                            dmc.ActionIcon(
+                                DashIconify(icon="mdi:console", width=18),
+                                id="open-terminal-btn",
+                                variant="default",
+                                size="md",
+                            ),
+                            dmc.ActionIcon(
+                                DashIconify(icon="mdi:refresh", width=18),
+                                id="refresh-btn",
+                                variant="default",
+                                size="md",
+                            ),
+                        ], id="files-actions", gap=5)
+                    ], id="sidebar-header", style={
+                        "display": "flex", "justifyContent": "space-between",
+                        "alignItems": "center", "padding": "8px 12px",
+                        "borderBottom": "1px solid var(--mantine-color-default-border)",
+                    }),
+
+                    # Files view
+                    html.Div([
+                        html.Div(
+                            id="file-tree",
+                            children=render_file_tree(build_file_tree(workspace_root, workspace_root), colors, styles),
+                            style={
+                                "flex": "1",
+                                "overflowY": "auto",
+                                "minHeight": "0",
+                            }
+                        ),
+                    ], id="files-view", style={
+                        "flex": "1",
+                        "minHeight": "0",
+                        "display": "flex",
+                        "flexDirection": "column",
+                    }),
+
+                    # Canvas view (hidden by default)
+                    html.Div([
+                        html.Div(id="canvas-content", style={
+                            "flex": "1",
+                            "minHeight": "0",
+                            "overflowY": "auto",
+                            "padding": "15px",
+                            "background": "var(--mantine-color-body)",
+                        }),
+                        # Canvas action button
+                        dmc.Group([
+                            dmc.Button("Clear", id="clear-canvas-btn", size="sm", color="red", variant="light"),
+                        ], id="canvas-actions", justify="center", style={
+                            "padding": "8px 15px",
+                            "borderTop": "1px solid var(--mantine-color-default-border)",
+                            "background": "var(--mantine-color-body)",
+                        })
+                    ], id="canvas-view", style={
+                        "flex": "1",
+                        "minHeight": "0",
+                        "display": "none",
+                        "flexDirection": "column",
+                        "overflow": "hidden"
+                    }),
+                ], id="sidebar-panel", style={
+                    "flex": "1",
+                    "minWidth": "0",
+                    "minHeight": "0",
+                    "display": "flex",
+                    "flexDirection": "column",
+                    "background": "var(--mantine-color-body)",
+                    "borderLeft": "1px solid var(--mantine-color-default-border)",
+                }),
+            ], id="main-container", style={"display": "flex", "flex": "1", "overflow": "hidden"}),
+        ], id="app-container", style={"display": "flex", "flexDirection": "column", "height": "100vh"})
+    ])