more-compute 0.1.0__py3-none-any.whl

kernel_run.py

@@ -0,0 +1,283 @@
+#!/usr/bin/env python3
+
+import argparse
+import subprocess
+import sys
+import os
+import time
+import signal
+import threading
+import webbrowser
+from pathlib import Path
+
+from morecompute.notebook import Notebook
+
+DEFAULT_NOTEBOOK_NAME = "notebook.ipynb"
+
+class NotebookLauncher:
+    def __init__(self, notebook_path: Path, debug=False):
+        self.backend_process = None
+        self.frontend_process = None
+        self.root_dir = Path(__file__).parent
+        self.debug = debug
+        self.notebook_path = notebook_path
+        root_dir = notebook_path.parent if notebook_path.parent != Path('') else Path.cwd()
+        os.environ["MORECOMPUTE_ROOT"] = str(root_dir.resolve())
+        os.environ["MORECOMPUTE_NOTEBOOK_PATH"] = str(self.notebook_path)
+
+    def start_backend(self):
+        """Start the FastAPI backend server"""
+        try:
+            # Force a stable port (default 8000); if busy, ask to free it
+            chosen_port = int(os.getenv("MORECOMPUTE_PORT", "8000"))
+            self._ensure_port_available(chosen_port)
+            cmd = [
+                sys.executable,
+                "-m",
+                "uvicorn",
+                "morecompute.server:app",
+                "--host",
+                "localhost",
+                "--port",
+                str(chosen_port),
+            ]
+
+            # Enable autoreload only when debugging or explicitly requested
+            enable_reload = (
+                self.debug
+                or os.getenv("MORECOMPUTE_RELOAD", "0") == "1"
+            )
+            if enable_reload:
+                # Limit reload scope to backend code and exclude large/changing artifacts
+                cmd.extend([
+                    "--reload",
+                    "--reload-dir", "morecompute",
+                    "--reload-exclude", "*.ipynb",
+                    "--reload-exclude", "frontend",
+                    "--reload-exclude", "assets",
+                ])
+
+            if not self.debug:
+                cmd.extend(["--log-level", "error", "--no-access-log"])
+
+            stdout_dest = None if self.debug else subprocess.DEVNULL
+            stderr_dest = None if self.debug else subprocess.DEVNULL
+
+            # Start the FastAPI server using uvicorn
+            self.backend_process = subprocess.Popen(
+                cmd,
+                cwd=self.root_dir,
+                stdout=stdout_dest,
+                stderr=stderr_dest,
+            )
+            # Save for later printing/opening
+            self.backend_port = chosen_port
+        except Exception as e:
+            print(f"Failed to start backend: {e}")
+            sys.exit(1)
+
+    def _ensure_port_available(self, port: int) -> None:
+        import socket
+        with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
+            s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
+            try:
+                s.bind(("127.0.0.1", port))
+                return  # free
+            except OSError:
+                pass  # in use
+        # Port is in use - show processes and ask to kill
+        print(f"\nPort {port} appears to be in use.")
+        pids = []
+        try:
+            out = subprocess.check_output(["lsof", "-nP", f"-iTCP:{port}", "-sTCP:LISTEN"]).decode("utf-8", errors="ignore")
+            print(out)
+            for line in out.splitlines()[1:]:
+                parts = line.split()
+                if len(parts) > 1 and parts[1].isdigit():
+                    pids.append(int(parts[1]))
+        except Exception:
+            print("Could not list processes with lsof. You may need to free the port manually.")
+        resp = input(f"Kill processes on port {port} and continue? [y/N]: ").strip().lower()
+        if resp != "y":
+            print("Aborting. Set MORECOMPUTE_PORT to a different port to override.")
+            sys.exit(1)
+        # Attempt to kill
+        for pid in pids:
+            try:
+                os.kill(pid, 9)
+            except Exception:
+                pass
+        # Fallback: kill known patterns
+        try:
+            subprocess.run(["pkill", "-f", "uvicorn .*morecompute.server:app"], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
+        except Exception:
+            pass
+        try:
+            subprocess.run(["pkill", "-f", "morecompute.execution.worker"], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
+        except Exception:
+            pass
+        # Brief pause to let the OS release the port
+        time.sleep(0.5)
+        # Poll until it binds
+        start = time.time()
+        while time.time() - start < 5.0:
+            with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s2:
+                s2.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
+                try:
+                    s2.bind(("127.0.0.1", port))
+                    return
+                except OSError:
+                    time.sleep(0.25)
+        print(f"Port {port} still busy. Please free it or set MORECOMPUTE_PORT to another port.")
+        sys.exit(1)
+
+    def start_frontend(self):
+        """Start the Next.js frontend server"""
+        try:
+            frontend_dir = self.root_dir / "frontend"
+
+            # Check if node_modules exists
+            if not (frontend_dir / "node_modules").exists():
+                print("Installing dependencies...")
+                subprocess.run(
+                    ["npm", "install"],
+                    cwd=frontend_dir,
+                    check=True,
+                    stdout=subprocess.DEVNULL,
+                    stderr=subprocess.DEVNULL
+                )
+
+            fe_stdout = None if self.debug else subprocess.DEVNULL
+            fe_stderr = None if self.debug else subprocess.DEVNULL
+
+            self.frontend_process = subprocess.Popen(
+                ["npm", "run", "dev"],
+                cwd=frontend_dir,
+                stdout=fe_stdout,
+                stderr=fe_stderr
+            )
+
+            # Wait a bit then open browser
+            time.sleep(3)
+            webbrowser.open("http://localhost:3000")
+
+        except Exception as e:
+            print(f"Failed to start frontend: {e}")
+            self.cleanup()
+            sys.exit(1)
+
+    def cleanup(self):
+        """Clean up processes on exit"""
+        if self.frontend_process:
+            self.frontend_process.terminate()
+            try:
+                self.frontend_process.wait(timeout=5)
+            except subprocess.TimeoutExpired:
+                self.frontend_process.kill()
+
+        if self.backend_process:
+            self.backend_process.terminate()
+            try:
+                self.backend_process.wait(timeout=5)
+            except subprocess.TimeoutExpired:
+                self.backend_process.kill()
+
+    def run(self):
+        """Main run method"""
+        print("\n        Edit notebook in your browser!\n")
+        print("        ➜  URL: http://localhost:3000\n")
+
+        # Set up signal handlers
+        def signal_handler(signum, frame):
+            print("\n\n        Thanks for using MoreCompute!\n")
+            self.cleanup()
+            sys.exit(0)
+
+        signal.signal(signal.SIGINT, signal_handler)
+        signal.signal(signal.SIGTERM, signal_handler)
+
+        # Start services
+        self.start_backend()
+        time.sleep(1)
+        self.start_frontend()
+
+        # Wait for processes
+        try:
+            while True:
+                # Check if processes are still running
+                if self.backend_process and self.backend_process.poll() is not None:
+                    self.cleanup()
+                    sys.exit(1)
+
+                if self.frontend_process and self.frontend_process.poll() is not None:
+                    self.cleanup()
+                    sys.exit(1)
+
+                time.sleep(1)
+
+        except KeyboardInterrupt:
+            print("\n\n        Thanks for using MoreCompute!\n")
+            self.cleanup()
+
+
+def build_parser() -> argparse.ArgumentParser:
+    parser = argparse.ArgumentParser(description="Launch the MoreCompute notebook")
+    parser.add_argument(
+        "notebook_path",
+        nargs="?",
+        default=None,
+        help="Path to the .ipynb notebook file",
+    )
+    parser.add_argument(
+        "-debug",
+        "--debug",
+        action="store_true",
+        help="Show backend/frontend logs (hidden by default)",
+    )
+    return parser
+
+
+def ensure_notebook_exists(notebook_path: Path):
+    if notebook_path.exists():
+        if notebook_path.suffix != '.ipynb':
+            raise ValueError("Notebook path must be a .ipynb file")
+        return
+
+    if notebook_path.suffix != '.ipynb':
+        raise ValueError("Notebook path must end with .ipynb")
+
+    notebook_path.parent.mkdir(parents=True, exist_ok=True)
+    notebook = Notebook()
+    notebook.save_to_file(str(notebook_path))
+
+
+def main(argv=None):
+    parser = build_parser()
+    args = parser.parse_args(argv)
+    raw_notebook_path = args.notebook_path
+
+    if raw_notebook_path == "new":
+        from datetime import datetime
+        timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+        raw_notebook_path = f"notebook_{timestamp}.ipynb"
+        print(f"Creating new notebook: {raw_notebook_path}")
+
+    notebook_path_env = os.getenv("MORECOMPUTE_NOTEBOOK_PATH")
+    if raw_notebook_path is None:
+        raw_notebook_path = notebook_path_env
+
+    if raw_notebook_path is None:
+        raw_notebook_path = DEFAULT_NOTEBOOK_NAME
+
+    notebook_path = Path(raw_notebook_path).expanduser().resolve()
+    ensure_notebook_exists(notebook_path)
+
+    launcher = NotebookLauncher(
+        notebook_path=notebook_path,
+        debug=args.debug
+    )
+    launcher.run()
+
+
+if __name__ == "__main__":
+    main()

more_compute-0.1.0.dist-info/METADATA

@@ -0,0 +1,163 @@
+Metadata-Version: 2.4
+Name: more-compute
+Version: 0.1.0
+Summary: An interactive notebook environment for local and GPU computing
+Home-page: https://github.com/DannyMang/MORECOMPUTE
+Author: MoreCompute Team
+Author-email: MoreCompute Team <hello@morecompute.dev>
+License: MIT
+Project-URL: Homepage, https://github.com/DannyMang/MORECOMPUTE
+Project-URL: Repository, https://github.com/DannyMang/MORECOMPUTE
+Project-URL: Issues, https://github.com/DannyMang/MORECOMPUTE/issues
+Keywords: jupyter,notebook,gpu,computing,interactive
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: fastapi>=0.104.0
+Requires-Dist: uvicorn[standard]>=0.24.0
+Requires-Dist: python-multipart>=0.0.5
+Requires-Dist: nbformat>=5.0.0
+Requires-Dist: click>=8.0.0
+Requires-Dist: pyzmq>=25.0.0
+Requires-Dist: psutil>=5.9.0
+Requires-Dist: httpx>=0.24.0
+Requires-Dist: cachetools>=5.3.0
+Requires-Dist: matplotlib>=3.5.0
+Dynamic: author
+Dynamic: home-page
+Dynamic: license-file
+Dynamic: requires-python
+
+# more-compute
+An interactive notebook environment similar to Marimo and Google Colab that runs locally.
+
+For references:
+
+https://marimo.io/
+
+https://colab.google/
+
+
+FOR LOCAL DEVELOPMENT:
+
+```bash
+pip install -e .
+```
+
+## Installation
+
+```bash
+(NOT IMPLEMENTED YET, TO-DO SET UP ON PyPi)
+pip install more-compute
+```
+
+## Usage
+
+### Create a new notebook
+```bash
+more-compute new
+```
+This creates a timestamped notebook like `notebook_20241007_153302.ipynb`
+
+Or run directly:
+```bash
+python3 kernel_run.py new
+```
+
+### Open an existing notebook
+```bash
+# Open a specific notebook
+more-compute your_notebook.ipynb
+
+# Or run directly
+python3 kernel_run.py your_notebook.ipynb
+
+# If no path provided, opens default notebook
+more-compute
+```
+
+## Features
+
+- **Interactive notebook interface** similar to Google Colab
+- **Support for both `.py` and `.ipynb` files**
+- **Real-time cell execution** with execution timing
+- **Magic commands** support:
+  - `!pip install package_name` - Install Python packages
+  - `!ls` - List directory contents
+  - `!pwd` - Print working directory
+  - `!any_shell_command` - Run any shell command
+- **Visual execution feedback**:
+  - ✅ Green check icon for successful execution
+  - ❌ Red X icon for failed execution
+  - Execution timing displayed for each cell
+- **Local development environment** - runs on your machine
+- **Web-based interface** accessible via localhost
+- **Cell management**:
+  - Add/delete cells
+  - Drag and drop to reorder
+  - Code and Markdown cell types
+
+## Usage Examples
+
+### Installing and Using Libraries
+```python
+# Install packages using magic commands (like Colab)
+!pip install pandas numpy matplotlib
+
+# Import and use them
+import pandas as pd
+import numpy as np
+import matplotlib.pyplot as plt
+
+# Create some data
+df = pd.DataFrame({
+    'x': np.range(10),
+    'y': np.random.randn(10)
+})
+
+print(df.head())
+```
+
+### Shell Commands
+```bash
+# List files
+!ls -la
+
+# Check current directory
+!pwd
+
+# Run any shell command
+!echo "Hello from the shell!"
+```
+
+### Data Analysis Example
+```python
+# Load data
+data = pd.read_csv('your_data.csv')
+
+# Analyze
+data.describe()
+
+# Plot
+plt.figure(figsize=(10, 6))
+plt.plot(data['x'], data['y'])
+plt.title('My Analysis')
+plt.show()
+```
+
+## Development
+
+To install in development mode:
+```bash
+pip install -e .
+```

more_compute-0.1.0.dist-info/RECORD

@@ -0,0 +1,26 @@
+kernel_run.py,sha256=tZwu-cmeqnBfzD7vme2o_-ZMFCWCZzAPBWwPfod2_ZY,9640
+more_compute-0.1.0.dist-info/licenses/LICENSE,sha256=0Ot-XIetYt06iay6IhtpJkruD-cLZtjyv7_aIEE-oSc,1073
+morecompute/__init__.py,sha256=pcMVq8Q7qb42AOn7tqgoZJOi3epDDBnEriiv2WVKnXY,87
+morecompute/cli.py,sha256=kVvzvPBqF8xO6UuhU_-TBn99nKwJ405R2mAS6zU0KBc,734
+morecompute/notebook.py,sha256=vSKSjaIZzUbDoPEk2yb4rmM9LA71o44WGqmFen0b70c,3114
+morecompute/process_worker.py,sha256=KsE3r-XpkYGuyO4w3t54VKkD51LfNHAZc3TYattMtrg,7185
+morecompute/server.py,sha256=sAZ9X4CJpd59ZDYjdu7LDYy_Nwd1gQYF9Dn5uRP_uDU,24071
+morecompute/execution/__init__.py,sha256=jPmBmq8BZWbUEY9XFSpqt5FkgX04uNS10WnUlr7Rnms,134
+morecompute/execution/__main__.py,sha256=pAWB_1bn99u8Gb-tVMSMI-NYvbYbDiwbn40L0a0djeA,202
+morecompute/execution/executor.py,sha256=Ngpd_PwQkgvX_ZMEcwGXPKY0OTKtwiI-lrv1ITPF9cA,19399
+morecompute/execution/worker.py,sha256=1XAFMUDbc2A1lO7xUKS9s69Vu6gOMOPdkr8w1pOoQU4,9634
+morecompute/services/pod_manager.py,sha256=eGZyLBiAXVk-rSdQJZOkw5scEHdofwqxNGUJIu78yI0,16684
+morecompute/services/prime_intellect.py,sha256=xHzfKgg8pIKikcMCzgz_5RvXC_LekeIR2gq2hDYRRm4,9129
+morecompute/static/styles.css,sha256=el_NtrUMbAUNSiMVBn1xlG70m3iPv7dyaIbWQMexhsY,19277
+morecompute/utils/__init__.py,sha256=VIxCL3S1pnjEs4cjKGZqZB68_P8FegdeMIqBjJhI5jQ,419
+morecompute/utils/cache_util.py,sha256=lVlXudHvtyvSo_kCSxORJrI85Jod8FrQLbI2f_JOIbA,661
+morecompute/utils/error_utils.py,sha256=e50WLFdD6ngIC30xAgrzdTYtD8tPOIFkKAAh_sPbK0I,11667
+morecompute/utils/notebook_util.py,sha256=3hH94dtXvhizRVTU9a2b38m_51Y4igoXpkjAXUqpVBQ,1353
+morecompute/utils/python_environment_util.py,sha256=l8WWWPwKbypknw8GwL22NXCji5i1FOy1vWG47J6og4g,7441
+morecompute/utils/special_commands.py,sha256=oCmAjKUsIOPIfobaJFDWhN2D5PLkKZtQozR3XAdIMe8,18653
+morecompute/utils/system_environment_util.py,sha256=32mQRubo0i4X61o-825T7m-eUSidcEp07qkInP1sWZA,4774
+more_compute-0.1.0.dist-info/METADATA,sha256=TTuzochhvAsXky6P2Jpq5nmULreWY9edtu0x5lBC0Wk,3817
+more_compute-0.1.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+more_compute-0.1.0.dist-info/entry_points.txt,sha256=xp7z9eRPNRM4oxkZZVlyXkhkSjN1AjoYI_B7qpDJ1bI,49
+more_compute-0.1.0.dist-info/top_level.txt,sha256=bK5t6sP_R9xshwmWbm19ctLWV3tln2qRyDQH1xb6OfU,23
+more_compute-0.1.0.dist-info/RECORD,,

more_compute-0.1.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (80.9.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

more_compute-0.1.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+more-compute = kernel_run:main

more_compute-0.1.0.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 MoreCompute Team
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

more_compute-0.1.0.dist-info/top_level.txt

@@ -0,0 +1,2 @@
+kernel_run
+morecompute

morecompute/__init__.py

@@ -0,0 +1,6 @@
+"""
+MoreCompute: An interactive, real-time Python notebook
+"""
+
+__version__ = "0.1.0"
+

morecompute/cli.py

@@ -0,0 +1,31 @@
+import click
+from pathlib import Path
+from .notebook import Notebook
+
+@click.group()
+def main():
+    """MoreCompute CLI."""
+
+
+@main.command()
+@click.argument('notebook_path', required=True)
+def new(notebook_path: str):
+    """Create a new .ipynb notebook."""
+    path = Path(notebook_path).expanduser().resolve()
+
+    if path.suffix != '.ipynb':
+        raise click.UsageError("Only .ipynb notebooks are supported right now.")
+
+    if path.exists():
+        raise click.ClickException(f"File '{path}' already exists")
+
+    path.parent.mkdir(parents=True, exist_ok=True)
+
+    notebook = Notebook(str(path))
+    notebook.save_to_file(str(path))
+
+    click.echo(f"✅ Created notebook at {path}")
+
+
+if __name__ == '__main__':
+    main()

morecompute/execution/__init__.py

@@ -0,0 +1,5 @@
+"""Code execution module - manages ZMQ-based code execution."""
+
+from .executor import NextZmqExecutor
+
+__all__ = ["NextZmqExecutor"]

morecompute/execution/__main__.py

@@ -0,0 +1,10 @@
+"""Entry point for ZMQ worker process.
+
+This allows running the worker via:
+    python -m morecompute.execution.worker
+"""
+
+from .worker import worker_main
+
+if __name__ == '__main__':
+    worker_main()