Upload modal_dev_env.py

modal_dev_env.py

@@ -0,0 +1,138 @@
+import os
+import subprocess
+import time
+import signal
+import sys
+
+import modal
+
+app = modal.App("dev-environment")
+
+# Build optimized image with JupyterLab and code-server
+image = (
+    modal.Image.debian_slim(python_version="3.12")
+    .apt_install(
+        "curl", "git", "wget", "vim", "htop",
+        "build-essential",  # for compiling Python packages
+        "nodejs", "npm"
+    )
+    .pip_install(
+        "jupyterlab>=4.0.0",
+        "ipython",
+        "matplotlib", "numpy", "pandas", "scipy",  # common data science libs
+        "requests", "python-dotenv"
+    )
+    # Install code-server with specific version for stability
+    .run_commands(
+        "curl -fsSL https://code-server.dev/install.sh | sh",
+        # Install useful VS Code extensions
+        "mkdir -p /root/.local/share/code-server/extensions"
+    )
+    # Create working directory
+    .run_commands("mkdir -p /workspace")
+)
+
+@app.function(
+    image=image,
+    cpu=2,
+    memory=4096,  # 4GB RAM
+    timeout=6 * 3600,  # 6 hours (more reasonable than 24h)
+    max_containers=1,
+    # Mount for persistent storage (optional)
+    # volumes={"/workspace": modal.Volume.from_name("dev-workspace", create_if_missing=True)}
+)
+def serve_dev_environment():
+    """Launch Jupyter Lab and VS Code development environment"""
+    
+    # Set up signal handler for graceful shutdown
+    def signal_handler(sig, frame):
+        print("\nShutting down development environment...")
+        sys.exit(0)
+    
+    signal.signal(signal.SIGINT, signal_handler)
+    signal.signal(signal.SIGTERM, signal_handler)
+    
+    # Change to workspace directory
+    os.chdir("/workspace")
+    
+    # Expose ports to the public internet
+    with modal.forward(8888) as jupyter_tunnel, \
+         modal.forward(8080) as vscode_tunnel:
+        
+        print("🚀 Development Environment Ready!")
+        print("=" * 50)
+        print(f"📊 Jupyter Lab  → {jupyter_tunnel.url}")
+        print(f"💻 VS Code      → {vscode_tunnel.url}")
+        print("=" * 50)
+        print("💡 Tip: Use Ctrl+C to stop the environment")
+        print()
+        
+        # Launch JupyterLab with better configuration
+        jupyter_process = subprocess.Popen([
+            "jupyter", "lab",
+            "--no-browser",
+            "--allow-root",
+            "--ip=0.0.0.0",
+            "--port=8888",
+            "--ServerApp.allow_origin='*'",
+            "--ServerApp.allow_remote_access=True",
+            "--ServerApp.disable_check_xsrf=True",
+            "--ServerApp.token=''",  # No token for easier access
+            "--ServerApp.password=''",
+            "--ServerApp.root_dir=/workspace"
+        ], env={**os.environ}, cwd="/workspace")
+        
+        # Launch code-server with better configuration
+        vscode_process = subprocess.Popen([
+            "code-server",
+            "--bind-addr", "0.0.0.0:8080",
+            "--auth", "none",
+            "--disable-telemetry",
+            "--disable-update-check",
+            "/workspace"  # Open workspace directory
+        ], env={**os.environ}, cwd="/workspace")
+        
+        try:
+            # Health check loop with process monitoring
+            while True:
+                # Check if processes are still running
+                if jupyter_process.poll() is not None:
+                    print("⚠️  Jupyter Lab process died, restarting...")
+                    jupyter_process = subprocess.Popen([
+                        "jupyter", "lab", "--no-browser", "--allow-root",
+                        "--ip=0.0.0.0", "--port=8888",
+                        "--ServerApp.allow_origin='*'",
+                        "--ServerApp.allow_remote_access=True",
+                        "--ServerApp.token=''", "--ServerApp.password=''"
+                    ], env={**os.environ}, cwd="/workspace")
+                
+                if vscode_process.poll() is not None:
+                    print("⚠️  VS Code process died, restarting...")
+                    vscode_process = subprocess.Popen([
+                        "code-server", "--bind-addr", "0.0.0.0:8080",
+                        "--auth", "none", "--disable-telemetry", "/workspace"
+                    ], env={**os.environ}, cwd="/workspace")
+                
+                time.sleep(30)  # Check every 30 seconds instead of 10 minutes
+                
+        except KeyboardInterrupt:
+            print("\n🛑 Stopping development environment...")
+            jupyter_process.terminate()
+            vscode_process.terminate()
+            
+            # Wait for graceful shutdown
+            try:
+                jupyter_process.wait(timeout=5)
+                vscode_process.wait(timeout=5)
+            except subprocess.TimeoutExpired:
+                jupyter_process.kill()
+                vscode_process.kill()
+
+@app.local_entrypoint()
+def main():
+    """Start the development environment"""
+    print("🔄 Starting development environment...")
+    serve_dev_environment.remote()
+
+if __name__ == "__main__":
+    main()