more-compute 0.3.3__py3-none-any.whl → 0.4.0__py3-none-any.whl

frontend/app/globals.css

@@ -643,7 +643,8 @@ body {
 }
 
 .markdown-rendered code {
-    background: #f3f4f6;
+    background: var(--mc-secondary);
+    color: var(--mc-text-color);
     padding: 2px 4px;
     border-radius: 3px;
     font-family: 'Fira', 'SF Mono', Monaco, 'Cascadia Code', 'Roboto Mono', Consolas, 'Courier New', monospace;
@@ -651,7 +652,7 @@ body {
 }
 
 .markdown-rendered pre {
-    background: #f9fafb;
+    background: var(--mc-secondary);
     padding: 12px;
     border-radius: 6px;
     margin: 12px 0;
@@ -665,7 +666,7 @@ body {
 }
 
 .markdown-rendered blockquote {
-    border-left: 3px solid #d1d5db;
+    border-left: 3px solid var(--mc-border);
     padding-left: 12px;
     margin: 12px 0;
     color: var(--mc-markdown-paragraph-color);
@@ -684,7 +685,7 @@ body {
 }
 
 .markdown-rendered .empty-markdown {
-    color: #9ca3af;
+    color: var(--mc-line-number-color);
     font-style: italic;
 }
 
@@ -697,19 +698,20 @@ body {
     flex-direction: column;
     align-items: center;
     font-size: 11px;
-    color: #6b7280;
+    color: var(--mc-line-number-color);
     width: 30px;
 }
 
 .execution-count {
     font-family: 'Fira', 'SF Mono', Monaco, 'Cascadia Code', 'Roboto Mono', Consolas, 'Courier New', monospace;
     font-size: 10px;
-    color: #9ca3af;
+    color: var(--mc-line-number-color);
 }
 
 .execution-time {
     font-size: 9px;
-    color: #d1d5db;
+    color: var(--mc-line-number-color);
+    opacity: 0.7;
     margin-top: 2px;
 }
 

frontend/app/layout.tsx

@@ -8,6 +8,7 @@ import PackagesPopup from "@/components/popups/PackagesPopup";
 import ComputePopup from "@/components/popups/ComputePopup";
 import MetricsPopup from "@/components/popups/MetricsPopup";
 import SettingsPopup from "@/components/popups/SettingsPopup";
+import ConfirmModal from "@/components/modals/ConfirmModal";
 import { ConnectionBanner } from "@/components/layout/ConnectionBanner";
 import {
   PodWebSocketProvider,
@@ -22,6 +23,7 @@ const POLL_MS = 3000;
 function AppContent({ children }: { children: React.ReactNode }) {
   const [appSettings, setAppSettings] = useState<NotebookSettings>(() => loadSettings());
   const [activePopup, setActivePopup] = useState<string | null>(null);
+  const [showRestartModal, setShowRestartModal] = useState(false);
   const { connectionState, gpuPods, connectingPodId } = usePodWebSocket();
 
   // Persistent metrics collection
@@ -84,6 +86,15 @@ function AppContent({ children }: { children: React.ReactNode }) {
     setActivePopup(null);
   };
 
+  const handleRestartKernel = () => {
+    // Send reset kernel command via WebSocket
+    const ws = new WebSocket('ws://127.0.0.1:3141/ws');
+    ws.onopen = () => {
+      ws.send(JSON.stringify({ type: 'reset_kernel' }));
+      setTimeout(() => ws.close(), 100);
+    };
+  };
+
   const renderPopup = () => {
     if (!activePopup) return null;
 
@@ -176,7 +187,28 @@ function AppContent({ children }: { children: React.ReactNode }) {
               id="kernel-status-dot"
               className="status-dot connecting"
             ></span>
-            <span id="kernel-status-text" className="status-text">
+            <span
+              id="kernel-status-text"
+              className="status-text"
+              data-original-text="Connecting..."
+              style={{
+                cursor: 'pointer',
+                transition: 'all 0.15s ease',
+              }}
+              onMouseEnter={(e) => {
+                const originalText = e.currentTarget.textContent || '';
+                e.currentTarget.setAttribute('data-original-text', originalText);
+                e.currentTarget.textContent = 'Restart Kernel';
+                e.currentTarget.style.color = 'var(--mc-primary)';
+              }}
+              onMouseLeave={(e) => {
+                const originalText = e.currentTarget.getAttribute('data-original-text') || 'Connecting...';
+                e.currentTarget.textContent = originalText;
+                e.currentTarget.style.color = '';
+              }}
+              onClick={() => setShowRestartModal(true)}
+              title="Click to restart kernel"
+            >
               Connecting...
             </span>
           </div>
@@ -197,6 +229,16 @@ function AppContent({ children }: { children: React.ReactNode }) {
           />
         </div>
       </div>
+      <ConfirmModal
+        isOpen={showRestartModal}
+        onClose={() => setShowRestartModal(false)}
+        onConfirm={handleRestartKernel}
+        title="Restart Kernel"
+        message="Are you sure you want to restart the kernel? All variables will be lost."
+        confirmLabel="Restart"
+        cancelLabel="Cancel"
+        isDangerous={true}
+      />
     </>
   );
 }

frontend/components/Notebook.tsx

@@ -276,10 +276,18 @@ function notebookReducer(
     case "EXECUTION_COMPLETE": {
       const payload = action.payload || {};
       const cell_index = payload.cell_index;
+
+      console.log(`[EXECUTION_COMPLETE] Processing completion for cell ${cell_index}`, payload);
+
       // Support both shapes: { result: {...} } and flat payload {...}
       const result = payload && payload.result ? payload.result : payload || {};
+
+      console.log(`[EXECUTION_COMPLETE] result.status=${result.status}, result.error=`, result.error);
+
+      // ALWAYS remove cell from executingCells when completion arrives
       const newExecuting = new Set(state.executingCells);
       newExecuting.delete(cell_index);
+
       return {
         ...state,
         executingCells: newExecuting,
@@ -310,8 +318,13 @@ function notebookReducer(
 
     case "EXECUTION_ERROR": {
       const { cell_index, error } = action.payload;
+
+      console.log(`[EXECUTION_ERROR] Processing error for cell ${cell_index}`);
+
+      // ALWAYS remove cell from executingCells when error arrives
       const newExecuting = new Set(state.executingCells);
       newExecuting.delete(cell_index);
+
       const normalizedError = normalizeError(error);
       return {
         ...state,
@@ -348,6 +361,8 @@ function notebookReducer(
           ...cell,
           outputs: [],
           execution_count: null,
+          execution_time: null,
+          error: null,
         })),
       };
 
@@ -435,6 +450,10 @@ export const Notebook: React.FC<NotebookProps> = ({
     dispatch({ type: "NOTEBOOK_UPDATED", payload: data });
   }, []);
 
+  const handleKernelRestarted = useCallback(() => {
+    dispatch({ type: "RESET_KERNEL" });
+  }, []);
+
   const handleHeartbeat = useCallback((data: any) => {
     // Heartbeat received - execution still in progress
     // Cell spinner already showing via executingCells set
@@ -473,7 +492,7 @@ export const Notebook: React.FC<NotebookProps> = ({
     wsRef.current = ws;
     handleKernelStatusUpdate("connecting");
 
-    ws.connect("ws://127.0.0.1:8000/ws")
+    ws.connect("ws://127.0.0.1:3141/ws")
       .then(() => {
         ws.loadNotebook(notebookName || "default");
       })
@@ -486,6 +505,7 @@ export const Notebook: React.FC<NotebookProps> = ({
     ws.on("disconnect", () => handleKernelStatusUpdate("disconnected"));
     ws.on("notebook_loaded", handleNotebookLoaded);
     ws.on("notebook_updated", handleNotebookUpdate);
+    ws.on("kernel_restarted", handleKernelRestarted);
     ws.on("execution_start", handleExecutionStart);
     ws.on("stream_output", handleStreamOutput);
     ws.on("execution_complete", handleExecutionComplete);
@@ -499,6 +519,7 @@ export const Notebook: React.FC<NotebookProps> = ({
     handleKernelStatusUpdate,
     handleNotebookLoaded,
     handleNotebookUpdate,
+    handleKernelRestarted,
     handleExecutionStart,
     handleStreamOutput,
     handleExecuteResult,

frontend/components/cell/CellButton.tsx

@@ -22,7 +22,8 @@ export const CellButton: React.FC<CellButtonProps> = ({
   const [isHovered, setIsHovered] = useState(false);
 
   const handleClick = (e: React.MouseEvent) => {
-    if (onClick && !disabled && !isLoading) {
+    if (onClick && !disabled) {
+      // Allow clicks even when loading (for stop/interrupt functionality)
       onClick(e);
     }
   };
@@ -34,7 +35,7 @@ export const CellButton: React.FC<CellButtonProps> = ({
       onClick={handleClick}
       onMouseEnter={() => setIsHovered(true)}
       onMouseLeave={() => setIsHovered(false)}
-      disabled={disabled || isLoading}
+      disabled={disabled}
       title={title}
       style={{
         width: '28px',
@@ -45,9 +46,9 @@ export const CellButton: React.FC<CellButtonProps> = ({
         display: 'flex',
         alignItems: 'center',
         justifyContent: 'center',
-        cursor: disabled || isLoading ? 'not-allowed' : 'pointer',
+        cursor: disabled ? 'not-allowed' : 'pointer',
         transition: 'background-color 0.15s ease',
-        opacity: disabled ? 0.5 : 1
+        opacity: disabled ? 0.5 : isLoading ? 0.8 : 1
       }}
     >
       {icon}

frontend/components/cell/MonacoCell.tsx

@@ -12,6 +12,7 @@ import {
   UpdateIcon,
   LinkBreak2Icon,
   PlayIcon,
+  StopIcon,
   ChevronUpIcon,
   ChevronDownIcon,
 } from "@radix-ui/react-icons";
@@ -269,13 +270,13 @@ export const MonacoCell: React.FC<CellProps> = ({
       onExecute(indexRef.current);
       setIsEditing(false);
     } else {
-      if (isExecuting) {
-        onInterrupt(indexRef.current);
-      } else {
-        onExecute(indexRef.current);
-      }
+      onExecute(indexRef.current);
     }
-  }, [cell.cell_type, isExecuting, onExecute, onInterrupt]);
+  }, [cell.cell_type, onExecute]);
+
+  const handleInterrupt = useCallback(() => {
+    onInterrupt(indexRef.current);
+  }, [onInterrupt]);
 
   const handleCellClick = () => {
     onSetActive(indexRef.current);
@@ -583,15 +584,26 @@ export const MonacoCell: React.FC<CellProps> = ({
         <div className="cell-hover-controls">
           <div className="cell-actions-right">
             {!isMarkdownWithContent && (
-              <CellButton
-                icon={<PlayIcon className="w-6 h-6" />}
-                onClick={(e) => {
-                  e.stopPropagation();
-                  handleExecute();
-                }}
-                title={isExecuting ? "Stop execution" : "Run cell"}
-                isLoading={isExecuting}
-              />
+              <>
+                <CellButton
+                  icon={<PlayIcon className="w-6 h-6" />}
+                  onClick={(e) => {
+                    e.stopPropagation();
+                    handleExecute();
+                  }}
+                  title="Run cell"
+                  disabled={isExecuting}
+                />
+                <CellButton
+                  icon={<StopIcon className="w-6 h-6" />}
+                  onClick={(e) => {
+                    e.stopPropagation();
+                    handleInterrupt();
+                  }}
+                  title="Stop execution"
+                  disabled={!isExecuting}
+                />
+              </>
             )}
             <CellButton
               icon={<ChevronUpIcon className="w-6 h-6" />}
@@ -599,8 +611,8 @@ export const MonacoCell: React.FC<CellProps> = ({
                 e.stopPropagation();
                 onMoveUp(indexRef.current);
               }}
-              title="Move cell up"
-              disabled={index === 0}
+              title={isExecuting ? "Cannot move while executing" : "Move cell up"}
+              disabled={isExecuting || index === 0}
             />
             <CellButton
               icon={<ChevronDownIcon className="w-6 h-6" />}
@@ -608,16 +620,25 @@ export const MonacoCell: React.FC<CellProps> = ({
                 e.stopPropagation();
                 onMoveDown(indexRef.current);
               }}
-              title="Move cell down"
-              disabled={index === totalCells - 1}
+              title={isExecuting ? "Cannot move while executing" : "Move cell down"}
+              disabled={isExecuting || index === totalCells - 1}
             />
             <CellButton
               icon={<LinkBreak2Icon className="w-5 h-5" />}
               onClick={(e) => {
                 e.stopPropagation();
-                onDelete(indexRef.current);
+                // If cell is executing, interrupt it first
+                if (isExecuting) {
+                  onInterrupt(indexRef.current);
+                  // Wait a bit for interrupt to complete, then delete
+                  setTimeout(() => {
+                    onDelete(indexRef.current);
+                  }, 500);
+                } else {
+                  onDelete(indexRef.current);
+                }
               }}
-              title="Delete cell"
+              title={isExecuting ? "Stop and delete cell" : "Delete cell"}
             />
           </div>
         </div>

frontend/components/output/ErrorDisplay.tsx

@@ -165,7 +165,20 @@ const TypedErrorDisplay: FC<{ error: ErrorOutput }> = ({ error }) => {
             margin: 0
           }}
         >
-          {error.traceback?.join('\n') || ''}
+          {/* Always show error name and message */}
+          {error.ename && (
+            <div style={{ fontWeight: 600, marginBottom: '4px' }}>
+              {error.ename}: {error.evalue}
+            </div>
+          )}
+          {/* Show traceback if available */}
+          {error.traceback && error.traceback.length > 0 && (
+            <div>{error.traceback.join('\n')}</div>
+          )}
+          {/* Fallback if no traceback */}
+          {(!error.traceback || error.traceback.length === 0) && !error.ename && (
+            <div>An unknown error occurred</div>
+          )}
         </div>
       </div>
     </div>

frontend/contexts/PodWebSocketContext.tsx

@@ -95,7 +95,7 @@ export const PodWebSocketProvider: React.FC<PodWebSocketProviderProps> = ({ chil
       wsRef.current = null;
     }
 
-    const wsUrl = 'ws://127.0.0.1:8000/ws';
+    const wsUrl = 'ws://127.0.0.1:3141/ws';
     const ws = new WebSocket(wsUrl);
 
     ws.onopen = () => {

frontend/lib/websocket.ts

@@ -53,12 +53,12 @@ export class WebSocketService {
     };
   }
 
-  connect(url: string = 'ws://localhost:8000'): Promise<void> {
+  connect(url: string = 'ws://localhost:3141'): Promise<void> {
     return new Promise((resolve, reject) => {
       // For development, connect directly to the backend WebSocket
       const wsUrl = process.env.NODE_ENV === 'production'
         ? '/ws'
-        : 'ws://localhost:8000/ws';
+        : 'ws://localhost:3141/ws';
 
       // Use native WebSocket for FastAPI compatibility
       const ws = new WebSocket(wsUrl);

frontend/next.config.mjs

@@ -4,11 +4,11 @@ const nextConfig = {
     return [
       {
         source: '/ws/:path*',
-        destination: 'http://localhost:8000/ws/:path*',
+        destination: 'http://localhost:3141/ws/:path*',
       },
       {
         source: '/api/:path*',
-        destination: 'http://localhost:8000/api/:path*',
+        destination: 'http://localhost:3141/api/:path*',
       },
     ];
   },

kernel_run.py

@@ -33,8 +33,8 @@ class NotebookLauncher:
     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"))
+            # Force a stable port (default 3141); if busy, ask to free it
+            chosen_port = int(os.getenv("MORECOMPUTE_PORT", "3141"))
             self._ensure_port_available(chosen_port)
             cmd = [
                 sys.executable,
@@ -406,21 +406,40 @@ def build_parser() -> argparse.ArgumentParser:
 
 def ensure_notebook_exists(notebook_path: Path):
     if notebook_path.exists():
-        if notebook_path.suffix != '.ipynb':
+        # File exists, check extension
+        if notebook_path.suffix == '.ipynb':
             raise ValueError(
-                f"Error: '{notebook_path}' is not a notebook file.\n"
-                f"Notebook files must have a .ipynb extension.\n"
-                f"Example: more-compute {notebook_path}.ipynb"
+                f"Error: MoreCompute only supports .py notebooks.\n\n"
+                f"Convert your notebook with:\n"
+                f"  more-compute convert {notebook_path.name} -o {notebook_path.stem}.py\n\n"
+                f"Then open with:\n"
+                f"  more-compute {notebook_path.stem}.py"
+            )
+        elif notebook_path.suffix != '.py':
+            raise ValueError(
+                f"Error: '{notebook_path}' is not a Python notebook file.\n"
+                f"Notebook files must have a .py extension.\n"
+                f"Example: more-compute {notebook_path.stem}.py"
             )
         return
 
-    if notebook_path.suffix != '.ipynb':
+    # File doesn't exist, create it
+    if notebook_path.suffix == '.ipynb':
+        raise ValueError(
+            f"Error: MoreCompute only supports .py notebooks.\n\n"
+            f"Convert your notebook with:\n"
+            f"  more-compute convert {notebook_path.name} -o {notebook_path.stem}.py\n\n"
+            f"Or create a new notebook:\n"
+            f"  more-compute new"
+        )
+
+    if notebook_path.suffix != '.py':
         raise ValueError(
-            f"Error: '{notebook_path}' does not have the .ipynb extension.\n"
-            f"Notebook files must end with .ipynb\n\n"
+            f"Error: '{notebook_path}' does not have the .py extension.\n"
+            f"Notebook files must end with .py\n\n"
             f"Did you mean?\n"
-            f"  more-compute {notebook_path}.ipynb\n\n"
-            f"Or to create a new notebook with timestamp:\n"
+            f"  more-compute {notebook_path}.py\n\n"
+            f"Or to create a new notebook:\n"
             f"  more-compute new"
         )
 
@@ -431,14 +450,18 @@ def ensure_notebook_exists(notebook_path: Path):
 
 def print_help():
     """Print concise help message"""
-    print(f"""Usage: more-compute [OPTIONS] [NOTEBOOK]
+    print(f"""Usage: more-compute [OPTIONS] [COMMAND] [NOTEBOOK]
 
-MoreCompute - Jupyter notebooks with GPU compute
+MoreCompute - Python notebooks with GPU compute
 
 Getting started:
 
   * more-compute new              create a new notebook with timestamp
-  * more-compute notebook.ipynb   open or create notebook.ipynb
+  * more-compute notebook.py      open or create notebook.py
+
+Commands:
+  convert NOTEBOOK -o OUTPUT      convert .ipynb to .py format
+    Example: more-compute convert notebook.ipynb -o notebook.py
 
 Options:
   --version, -v                   Show version and exit
@@ -446,12 +469,72 @@ Options:
   --help, -h                      Show this message and exit
 
 Environment variables:
-  MORECOMPUTE_PORT                Backend port (default: 8000)
+  MORECOMPUTE_PORT                Backend port (default: 3141)
   MORECOMPUTE_FRONTEND_PORT       Frontend port (default: 2718)
   MORECOMPUTE_NOTEBOOK_PATH       Default notebook path
+
+Note: MoreCompute uses .py notebooks (not .ipynb). Convert existing notebooks with:
+  more-compute convert notebook.ipynb -o notebook.py
 """)
 
 def main(argv=None):
+    # Handle convert command before argparse (to avoid parsing -o flag)
+    argv_to_check = argv if argv is not None else sys.argv[1:]
+    if len(argv_to_check) > 0 and argv_to_check[0] == "convert":
+        # Parse convert arguments
+        if len(sys.argv) < 3:
+            print("Error: convert command requires input file")
+            print("\nUsage:")
+            print("  more-compute convert notebook.ipynb         # -> notebook.py")
+            print("  more-compute convert notebook.py            # -> notebook.ipynb")
+            print("  more-compute convert notebook.ipynb -o out.py")
+            sys.exit(1)
+
+        input_file = Path(sys.argv[2])
+        output_file = None
+
+        # Parse -o flag
+        if len(sys.argv) >= 5 and sys.argv[3] == "-o":
+            output_file = Path(sys.argv[4])
+        else:
+            # Auto-detect output extension based on input
+            if input_file.suffix == '.ipynb':
+                output_file = input_file.with_suffix('.py')
+            elif input_file.suffix == '.py':
+                output_file = input_file.with_suffix('.ipynb')
+            else:
+                print(f"Error: Unsupported file type: {input_file.suffix}")
+                print("Supported: .ipynb, .py")
+                sys.exit(1)
+
+        if not input_file.exists():
+            print(f"Error: File not found: {input_file}")
+            sys.exit(1)
+
+        # Perform conversion based on input type
+        if input_file.suffix == '.ipynb':
+            from morecompute.utils.notebook_converter import convert_ipynb_to_py
+            try:
+                convert_ipynb_to_py(input_file, output_file)
+                sys.exit(0)
+            except Exception as e:
+                print(f"Error converting notebook: {e}")
+                sys.exit(1)
+        elif input_file.suffix == '.py':
+            from morecompute.utils.notebook_converter import convert_py_to_ipynb
+            try:
+                convert_py_to_ipynb(input_file, output_file)
+                sys.exit(0)
+            except Exception as e:
+                print(f"Error converting notebook: {e}")
+                sys.exit(1)
+        else:
+            print(f"Error: Can only convert .ipynb or .py files")
+            print(f"Got: {input_file.suffix}")
+            sys.exit(1)
+
+
+    # Parse arguments for non-convert commands
     parser = build_parser()
     args = parser.parse_args(argv)
 
@@ -462,10 +545,11 @@ def main(argv=None):
 
     raw_notebook_path = args.notebook_path
 
+    # Handle "new" command
     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"
+        raw_notebook_path = f"notebook_{timestamp}.py"
         print(f"Creating new notebook: {raw_notebook_path}")
 
     notebook_path_env = os.getenv("MORECOMPUTE_NOTEBOOK_PATH")
@@ -477,7 +561,13 @@ def main(argv=None):
         sys.exit(0)
 
     notebook_path = Path(raw_notebook_path).expanduser().resolve()
-    ensure_notebook_exists(notebook_path)
+
+    try:
+        ensure_notebook_exists(notebook_path)
+    except ValueError as e:
+        # Print clean error message without traceback
+        print(str(e), file=sys.stderr)
+        sys.exit(1)
 
     launcher = NotebookLauncher(
         notebook_path=notebook_path,