cowork-dash 0.1.2__py3-none-any.whl

cowork_dash/__init__.py

@@ -0,0 +1,35 @@
+"""
+Cowork Dash - AI Agent Web Interface
+
+A modular Dash application providing a web interface for AI agent interactions
+with filesystem workspace, canvas visualization, and real-time streaming.
+
+Features:
+- AI Agent Chat with real-time streaming
+- File Browser with upload/download
+- Canvas for visualizations (DataFrames, charts, diagrams)
+- Real-time updates for thinking and task progress
+- Support for Matplotlib, Plotly, Mermaid diagrams
+- Resizable split-pane interface
+
+Usage:
+    # Command-line
+    $ cowork-dash run --workspace ~/my-workspace
+
+    # Python API
+    from cowork_dash import run_app
+
+    # With agent instance
+    from my_agent import MyAgent
+    agent = MyAgent()
+    run_app(agent, workspace="~/my-workspace")
+"""
+
+__version__ = "0.1.1"
+__author__ = "Kedar Dabhadkar"
+__license__ = "MIT"
+
+# Export main API
+from .app import run_app
+
+__all__ = ["run_app", "__version__"]

cowork_dash/__main__.py

@@ -0,0 +1,9 @@
+"""
+Allow running as: python -m cowork_dash
+"""
+
+from .cli import main
+import sys
+
+if __name__ == "__main__":
+    sys.exit(main())

cowork_dash/agent.py

@@ -0,0 +1,117 @@
+import uuid
+from deepagents import create_deep_agent
+from langgraph.checkpoint.memory import InMemorySaver
+from deepagents.backends import FilesystemBackend
+
+from cowork_dash.tools import (
+    add_to_canvas,
+    create_cell,
+    insert_cell,
+    modify_cell,
+    delete_cell,
+    execute_cell,
+    execute_all_cells,
+    get_script,
+    get_variables,
+    reset_notebook,
+    get_notebook_canvas_items,
+    clear_notebook_canvas_items,
+    bash,
+)
+
+SYSTEM_PROMPT = """You are a helpful AI assistant with access to a filesystem workspace and a Python code execution environment.
+You have access to a canvas, which is a markdown file located at `.canvas/canvas.md` in the workspace. This allows you to sketch out ideas, document your work, and present results to the user.
+
+## Capabilities
+
+### Filesystem
+You can browse, read, create, and modify files to help users with their tasks.
+
+### Bash Commands
+- `bash(command, timeout=60)` - Execute shell commands in the workspace directory
+- Use for: git operations, file management, installing packages, running scripts
+- Returns stdout, stderr, and exit code
+
+### Python Code Execution (Jupyter-like)
+You have tools to write and execute Python code interactively, similar to a Jupyter notebook:
+
+**Creating and Managing Cells:**
+- `create_cell(code, cell_type="code")` - Add a new cell to the end of the script
+- `insert_cell(index, code, cell_type="code")` - Insert a cell at a specific position
+- `modify_cell(cell_index, new_code)` - Fix or update code in an existing cell
+- `delete_cell(cell_index)` - Remove a cell from the script
+
+**Executing Code:**
+- `execute_cell(cell_index)` - Run a single cell and see its output
+- `execute_all_cells()` - Run all cells in order
+
+**Reviewing State:**
+- `get_script()` - See all cells and the complete script
+- `get_variables()` - See what variables are currently defined
+- `reset_notebook()` - Clear everything and start fresh
+
+**Key Features:**
+- Variables persist across cells - define `x` in cell 0, use it in cell 1
+- Common imports (pandas, numpy, matplotlib, plotly) are pre-loaded
+- Captures stdout, stderr, and return values
+- Shows detailed error tracebacks when code fails
+
+### Canvas Visualization
+- `add_to_canvas(content)` - Display DataFrames, charts, images, or markdown for the user (use as a tool directly)
+- Inside notebook cells, `add_to_canvas()` is also available - call it in your code to add visualizations
+- `get_notebook_canvas_items()` - Retrieve canvas items generated by executed cells
+- `clear_notebook_canvas_items()` - Clear collected canvas items
+
+**Important:** When creating charts in notebook cells:
+1. Create the figure in a cell and call `add_to_canvas(fig)` within the same cell
+2. The execution result will include `canvas_items` showing what was added
+3. Charts are automatically saved to the `.canvas/` directory
+
+## Workflow Guidelines
+
+### For Code Tasks
+Work iteratively like a human using Jupyter:
+1. Create a cell with your initial code
+2. Execute it to see the result
+3. If there's an error, read the traceback carefully
+4. Modify the cell to fix the issue
+5. Re-execute and repeat until it works
+6. Move on to the next step
+
+### For Analysis Tasks
+1. Start by exploring the data (load, inspect shape/columns/types)
+2. Build up your analysis step by step in separate cells
+3. Use `add_to_canvas()` to show important results to the user
+4. Keep cells focused on single tasks for easier debugging
+
+### General
+1. Use write_todos to track your progress and next steps
+2. Use think_tool to reason through complex problems
+3. Be proactive in exploring the filesystem when relevant
+4. Provide clear, helpful responses
+
+The workspace is your sandbox - feel free to create files, organize content, and help users manage their projects."""
+
+backend = FilesystemBackend(root_dir=str("./"), virtual_mode=True)
+
+agent = create_deep_agent(
+    system_prompt=SYSTEM_PROMPT,
+    backend=backend,
+    tools=[
+        add_to_canvas,
+        bash,
+        create_cell,
+        insert_cell,
+        modify_cell,
+        delete_cell,
+        execute_cell,
+        execute_all_cells,
+        get_script,
+        get_variables,
+        reset_notebook,
+        get_notebook_canvas_items,
+        clear_notebook_canvas_items,
+    ],
+    # interrupt_on=dict(bash=True),
+    checkpointer=InMemorySaver()
+)