more-compute 0.3.0__py3-none-any.whl → 0.3.2__py3-none-any.whl

frontend/app/globals.css

@@ -480,6 +480,7 @@ body {
 
 .cell:hover {
     box-shadow: 2px 4px 0 var(--mc-secondary-hover);
+    z-index: 11; /* Ensure hovered cell appears above add-cell-line (z-index: 10) */
 }
 
 .cell.active {

frontend/components/cell/MonacoCell.tsx

@@ -580,9 +580,9 @@ export const MonacoCell: React.FC<CellProps> = ({
         data-cell-index={index}
       >
         {/* Hover Controls */}
-        {!isMarkdownWithContent && (
-          <div className="cell-hover-controls">
-            <div className="cell-actions-right">
+        <div className="cell-hover-controls">
+          <div className="cell-actions-right">
+            {!isMarkdownWithContent && (
               <CellButton
                 icon={<PlayIcon className="w-6 h-6" />}
                 onClick={(e) => {
@@ -592,35 +592,35 @@ export const MonacoCell: React.FC<CellProps> = ({
                 title={isExecuting ? "Stop execution" : "Run cell"}
                 isLoading={isExecuting}
               />
-              <CellButton
-                icon={<ChevronUpIcon className="w-6 h-6" />}
-                onClick={(e) => {
-                  e.stopPropagation();
-                  onMoveUp(indexRef.current);
-                }}
-                title="Move cell up"
-                disabled={index === 0}
-              />
-              <CellButton
-                icon={<ChevronDownIcon className="w-6 h-6" />}
-                onClick={(e) => {
-                  e.stopPropagation();
-                  onMoveDown(indexRef.current);
-                }}
-                title="Move cell down"
-                disabled={index === totalCells - 1}
-              />
-              <CellButton
-                icon={<LinkBreak2Icon className="w-5 h-5" />}
-                onClick={(e) => {
-                  e.stopPropagation();
-                  onDelete(indexRef.current);
-                }}
-                title="Delete cell"
-              />
-            </div>
+            )}
+            <CellButton
+              icon={<ChevronUpIcon className="w-6 h-6" />}
+              onClick={(e) => {
+                e.stopPropagation();
+                onMoveUp(indexRef.current);
+              }}
+              title="Move cell up"
+              disabled={index === 0}
+            />
+            <CellButton
+              icon={<ChevronDownIcon className="w-6 h-6" />}
+              onClick={(e) => {
+                e.stopPropagation();
+                onMoveDown(indexRef.current);
+              }}
+              title="Move cell down"
+              disabled={index === totalCells - 1}
+            />
+            <CellButton
+              icon={<LinkBreak2Icon className="w-5 h-5" />}
+              onClick={(e) => {
+                e.stopPropagation();
+                onDelete(indexRef.current);
+              }}
+              title="Delete cell"
+            />
           </div>
-        )}
+        </div>
 
         {/* Cell Content */}
         <div

frontend/components/popups/FilterPopup.tsx

@@ -84,8 +84,8 @@ const FilterPopup: React.FC<FilterPopupProps> = ({
           top: "50%",
           left: "50%",
           transform: "translate(-50%, -50%)",
-          backgroundColor: "white",
-          border: "1px solid #d1d5db",
+          backgroundColor: "var(--mc-cell-background)",
+          border: "1px solid var(--mc-border)",
           borderRadius: "8px",
           padding: "16px",
           width: "320px",
@@ -110,7 +110,7 @@ const FilterPopup: React.FC<FilterPopupProps> = ({
               fontSize: "14px",
               fontWeight: 600,
               margin: 0,
-              color: "#111827",
+              color: "var(--mc-text-color)",
             }}
           >
             Filter
@@ -119,7 +119,7 @@ const FilterPopup: React.FC<FilterPopupProps> = ({
             onClick={handleClearAll}
             style={{
               fontSize: "11px",
-              color: "#3b82f6",
+              color: "var(--mc-primary)",
               background: "none",
               border: "none",
               cursor: "pointer",
@@ -142,9 +142,9 @@ const FilterPopup: React.FC<FilterPopupProps> = ({
             width: "100%",
             padding: "8px 10px",
             borderRadius: "6px",
-            border: "1px solid #d1d5db",
-            backgroundColor: "white",
-            color: "#111827",
+            border: "1px solid var(--mc-border)",
+            backgroundColor: "var(--mc-input-background)",
+            color: "var(--mc-text-color)",
             fontSize: "12px",
             marginBottom: "12px",
             cursor: "pointer",
@@ -166,9 +166,9 @@ const FilterPopup: React.FC<FilterPopupProps> = ({
             width: "100%",
             padding: "8px 10px",
             borderRadius: "6px",
-            border: "1px solid #d1d5db",
-            backgroundColor: "white",
-            color: "#111827",
+            border: "1px solid var(--mc-border)",
+            backgroundColor: "var(--mc-input-background)",
+            color: "var(--mc-text-color)",
             fontSize: "12px",
             marginBottom: "12px",
             boxSizing: "border-box",
@@ -182,7 +182,7 @@ const FilterPopup: React.FC<FilterPopupProps> = ({
             overflowY: "auto",
             marginBottom: "16px",
             maxHeight: "240px",
-            border: "1px solid #e5e7eb",
+            border: "1px solid var(--mc-border)",
             borderRadius: "6px",
             padding: "4px",
           }}
@@ -200,12 +200,12 @@ const FilterPopup: React.FC<FilterPopupProps> = ({
                     padding: "8px 6px",
                     cursor: "pointer",
                     fontSize: "12px",
-                    color: "#374151",
+                    color: "var(--mc-text-color)",
                     borderRadius: "4px",
                     transition: "background-color 0.15s",
                   }}
                   onMouseEnter={(e) =>
-                    (e.currentTarget.style.backgroundColor = "#f3f4f6")
+                    (e.currentTarget.style.backgroundColor = "var(--mc-secondary)")
                   }
                   onMouseLeave={(e) =>
                     (e.currentTarget.style.backgroundColor = "transparent")
@@ -249,12 +249,12 @@ const FilterPopup: React.FC<FilterPopupProps> = ({
                       padding: "8px 6px",
                       cursor: "pointer",
                       fontSize: "12px",
-                      color: "#374151",
+                      color: "var(--mc-text-color)",
                       borderRadius: "4px",
                       transition: "background-color 0.15s",
                     }}
                     onMouseEnter={(e) =>
-                      (e.currentTarget.style.backgroundColor = "#f3f4f6")
+                      (e.currentTarget.style.backgroundColor = "var(--mc-secondary)")
                     }
                     onMouseLeave={(e) =>
                       (e.currentTarget.style.backgroundColor = "transparent")
@@ -303,12 +303,12 @@ const FilterPopup: React.FC<FilterPopupProps> = ({
                       padding: "8px 6px",
                       cursor: "pointer",
                       fontSize: "12px",
-                      color: "#374151",
+                      color: "var(--mc-text-color)",
                       borderRadius: "4px",
                       transition: "background-color 0.15s",
                     }}
                     onMouseEnter={(e) =>
-                      (e.currentTarget.style.backgroundColor = "#f3f4f6")
+                      (e.currentTarget.style.backgroundColor = "var(--mc-secondary)")
                     }
                     onMouseLeave={(e) =>
                       (e.currentTarget.style.backgroundColor = "transparent")
@@ -352,12 +352,12 @@ const FilterPopup: React.FC<FilterPopupProps> = ({
                       padding: "8px 6px",
                       cursor: "pointer",
                       fontSize: "12px",
-                      color: "#374151",
+                      color: "var(--mc-text-color)",
                       borderRadius: "4px",
                       transition: "background-color 0.15s",
                     }}
                     onMouseEnter={(e) =>
-                      (e.currentTarget.style.backgroundColor = "#f3f4f6")
+                      (e.currentTarget.style.backgroundColor = "var(--mc-secondary)")
                     }
                     onMouseLeave={(e) =>
                       (e.currentTarget.style.backgroundColor = "transparent")
@@ -390,9 +390,9 @@ const FilterPopup: React.FC<FilterPopupProps> = ({
               padding: "8px 16px",
               fontSize: "12px",
               borderRadius: "6px",
-              border: "1px solid #d1d5db",
-              backgroundColor: "white",
-              color: "#374151",
+              border: "1px solid var(--mc-border)",
+              backgroundColor: "var(--mc-secondary)",
+              color: "var(--mc-text-color)",
               cursor: "pointer",
               fontWeight: 500,
             }}
@@ -410,8 +410,8 @@ const FilterPopup: React.FC<FilterPopupProps> = ({
               fontSize: "12px",
               borderRadius: "6px",
               border: "none",
-              backgroundColor: "#3b82f6",
-              color: "white",
+              backgroundColor: "var(--mc-primary)",
+              color: "var(--mc-button-foreground)",
               cursor: "pointer",
               fontWeight: 500,
             }}

frontend/components/popups/PackagesPopup.tsx

@@ -70,9 +70,6 @@ const PackagesPopup: React.FC<PackagesPopupProps> = ({ onClose }) => {
     return packages.filter(p => p.name.toLowerCase().includes(q));
   }, [packages, query]);
 
-  if (loading) return <div className="packages-list">Loading...</div>;
-  if (error) return <div className="packages-list">{error}</div>;
-
   return (
     <div className="packages-container">
       <div className="packages-toolbar">
@@ -98,12 +95,26 @@ const PackagesPopup: React.FC<PackagesPopupProps> = ({ onClose }) => {
           <div className="col-version">Version</div>
         </div>
         <div className="packages-list">
-          {filtered.map((pkg) => (
-            <div key={`${pkg.name}@${pkg.version}`} className="package-row">
-              <div className="col-name package-name">{pkg.name}</div>
-              <div className="col-version package-version">{pkg.version}</div>
+          {loading ? (
+            <div style={{ padding: '20px', textAlign: 'center', color: 'var(--mc-text-color)' }}>
+              Loading packages...
+            </div>
+          ) : error ? (
+            <div style={{ padding: '20px', textAlign: 'center', color: '#dc2626' }}>
+              {error}
+            </div>
+          ) : filtered.length === 0 ? (
+            <div style={{ padding: '20px', textAlign: 'center', color: 'var(--mc-text-color)', opacity: 0.6 }}>
+              {query ? 'No packages found' : 'No packages installed'}
             </div>
-          ))}
+          ) : (
+            filtered.map((pkg) => (
+              <div key={`${pkg.name}@${pkg.version}`} className="package-row">
+                <div className="col-name package-name">{pkg.name}</div>
+                <div className="col-version package-version">{pkg.version}</div>
+              </div>
+            ))
+          )}
         </div>
       </div>
     </div>

kernel_run.py

@@ -24,6 +24,7 @@ class NotebookLauncher:
         self.debug = debug
         self.notebook_path = notebook_path
         self.is_windows = platform.system() == "Windows"
+        self.cleaning_up = False  # Flag to prevent multiple cleanup calls
         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)
@@ -262,8 +263,14 @@ class NotebookLauncher:
             self.cleanup()
             sys.exit(1)
 
-    def cleanup(self):
+    def cleanup(self, force=False):
         """Clean up processes on exit"""
+        if self.cleaning_up:
+            return  # Already cleaning up, don't run again
+        self.cleaning_up = True
+
+        timeout = 0.5 if force else 2  # Shorter timeout on force exit
+
         if self.frontend_process:
             try:
                 if self.is_windows:
@@ -274,11 +281,14 @@ class NotebookLauncher:
                         stderr=subprocess.DEVNULL
                     )
                 else:
-                    self.frontend_process.terminate()
-                    try:
-                        self.frontend_process.wait(timeout=5)
-                    except subprocess.TimeoutExpired:
-                        self.frontend_process.kill()
+                    if force:
+                        self.frontend_process.kill()  # Force kill immediately
+                    else:
+                        self.frontend_process.terminate()
+                        try:
+                            self.frontend_process.wait(timeout=timeout)
+                        except subprocess.TimeoutExpired:
+                            self.frontend_process.kill()
             except Exception:
                 pass
 
@@ -292,11 +302,14 @@ class NotebookLauncher:
                         stderr=subprocess.DEVNULL
                     )
                 else:
-                    self.backend_process.terminate()
-                    try:
-                        self.backend_process.wait(timeout=5)
-                    except subprocess.TimeoutExpired:
-                        self.backend_process.kill()
+                    if force:
+                        self.backend_process.kill()  # Force kill immediately
+                    else:
+                        self.backend_process.terminate()
+                        try:
+                            self.backend_process.wait(timeout=timeout)
+                        except subprocess.TimeoutExpired:
+                            self.backend_process.kill()
             except Exception:
                 pass
 
@@ -316,8 +329,8 @@ class NotebookLauncher:
                 print("[CTRL-C AGAIN TO EXIT]")
             else:
                 print("\n        Thanks for using MoreCompute!\n")
-                self.cleanup()
-                sys.exit(0)
+                self.cleanup(force=True)  # Force immediate cleanup
+                os._exit(0)  # Hard exit without raising SystemExit
 
         # Windows signal handling is different
         if not self.is_windows:

{more_compute-0.3.0.dist-info → more_compute-0.3.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: more-compute
-Version: 0.3.0
+Version: 0.3.2
 Summary: An interactive notebook environment for local and GPU computing
 Home-page: https://github.com/DannyMang/MORECOMPUTE
 Author: MoreCompute Team

{more_compute-0.3.0.dist-info → more_compute-0.3.2.dist-info}/RECORD

@@ -1,4 +1,4 @@
-kernel_run.py,sha256=XAgUmZO9EVPXMlQe8RO8w4s_NveB9U7J6S-CyF12D0o,15847
+kernel_run.py,sha256=sQchzBOWgynXOKkSMp8T8BAkljw5CqDFrHX2Y0U6yto,16526
 frontend/.DS_Store,sha256=uQeHnkKyuTF1AVax3NPqtN0uCH6XNXAxL9Nkb6Q9cGw,8196
 frontend/.gitignore,sha256=IH4mX_SQH5rZ-W2M4IUw4E-fxgCBVHKmbQpEYJbWVM0,480
 frontend/README.md,sha256=YLVf9995r3JZD5UkII5GZCvDK9wXXNrUE0loHA4vlY8,1450
@@ -14,13 +14,13 @@ frontend/styling_README.md,sha256=jvoXKUzJv0rEIU40gz23Z6Xugy4To8P0N9szg_hKrqw,56
 frontend/tailwind.config.ts,sha256=eP9nVaAuyYo46vGQfCyWbo25_pr2hW830fs1Itcix9Q,620
 frontend/tsconfig.json,sha256=7SvBlRBYmuXAlAteRQTGwEE7ooWuNaPUrZ219dOo61E,598
 frontend/app/favicon.ico,sha256=K4rS0zRVqPc2_DqOv48L3qiEitTA20iigzvQ-c13WTI,25931
-frontend/app/globals.css,sha256=zx2VGZhPHqrFSwCsW667gJ80BkEwL8E7J_IJNoMuYGQ,33738
+frontend/app/globals.css,sha256=0k9WgoXRHIQjsI5kVSSkKLfYJs04FYOhOY8JvQrSulg,33823
 frontend/app/layout.tsx,sha256=VMkhvePu0hV0w-huEmTocTr7lZ6DF_7eidrhzCCMwzc,7604
 frontend/app/page.tsx,sha256=p-DgDv8xDnwcRfDJY4rtfSQ2VdYwbnP3G-otWqDR1l0,256
 frontend/components/Notebook.tsx,sha256=JImlThoX4F363yShOqdYDjIIO3i1uqfCLRkDlzurinQ,21851
 frontend/components/cell/AddCellButton.tsx,sha256=ZC2Vck0JIRDxGYhYv3LPYAdKDo13U6008WG_-XoPlIM,1012
 frontend/components/cell/CellButton.tsx,sha256=j3lRStluVtRXjZOhi9isS3XKsRmM0WTn5SaFDM6GuYg,1386
-frontend/components/cell/MonacoCell.tsx,sha256=YunwOlyeRsFqx-sHowhctqMV0RmhcQ1glSl5LO0s7NY,23914
+frontend/components/cell/MonacoCell.tsx,sha256=EqsajBZuQpoOWlNVG7OmmoIyY4pUfW5obiBwpniuQbw,23862
 frontend/components/layout/ConnectionBanner.tsx,sha256=-m77wKCFpeRJ_AQwnM38jLwCY5vfpqE846xeXmT3p8A,870
 frontend/components/layout/Sidebar.tsx,sha256=kxgO2mXX11xI6rX478cUqqQ3xB40jlby21y27GTJSLU,1551
 frontend/components/modals/ConfirmModal.tsx,sha256=3WgXy_wmR8FHZPwzMevfZHFXa0blR4v_SbKG3d5McT4,3652
@@ -30,10 +30,10 @@ frontend/components/output/CellOutput.tsx,sha256=KLRzwEchvBoeoai8TbabKEwO4tHmog3
 frontend/components/output/ErrorDisplay.tsx,sha256=kYfip5zikcFIuN7WW47RVjoqwDujC1q_yMbgHrHl0E0,4996
 frontend/components/output/MarkdownRenderer.tsx,sha256=RtZ5yNRxDXIh_NkNsNiy30wMGIW7V1gfhsjecgMdc80,3341
 frontend/components/popups/ComputePopup.tsx,sha256=B7AwwjpJ9_nNI5qSN53jdR2tAulB5yl7spjMXjH4yDM,44021
-frontend/components/popups/FilterPopup.tsx,sha256=4kx9txg8cWeC6XHlh0pu0-BAfZkLTDYEU93fMdzn86M,13818
+frontend/components/popups/FilterPopup.tsx,sha256=KRIzRvVxEF47ELWIAvrqv7BZQYmEo_zOX8Nbpz0zEik,14133
 frontend/components/popups/FolderPopup.tsx,sha256=V2tDAbztvNIUyPWtFiwjeIoCmFGQyDosQgST_JsAzLo,5215
 frontend/components/popups/MetricsPopup.tsx,sha256=ublYbOQ-pSU_w48F10uFjDYctysEwmri9lrji5YHIC4,6034
-frontend/components/popups/PackagesPopup.tsx,sha256=faeXL7hGFIEUBU6PUrTIr9P5Lw-81GGKWK1ldnIEDcc,3675
+frontend/components/popups/PackagesPopup.tsx,sha256=2DnwNVN8kwC3ff1UN_7cl6qLwAFE9F9H8hYwl7M3jTM,4141
 frontend/components/popups/SettingsPopup.tsx,sha256=rGwYxjkADnyTtpP0MjVpFPBsGoT1eENN2z7nWjLuzFE,2773
 frontend/contexts/PodWebSocketContext.tsx,sha256=QyEg1Msyum7h3T6F3j6pn4Ds5sXFT_9WYu5_0YZ2IUI,8256
 frontend/lib/api.ts,sha256=L6Js3K7RcPCLfVVUz-bKX5hLAVFKCfUSHAKSHk2keAk,11026
@@ -64,9 +64,9 @@ frontend/public/fonts/Fira.ttf,sha256=dbSM4W7Drd9n_EkfXq8P31KuxbjZ1wtuFiZ8aFebvT
 frontend/public/fonts/Tiempos.woff2,sha256=h83bJKvAK301wXCMIvK7ZG5j0H2K3tzAfgo0yk4z8OE,13604
 frontend/public/fonts/VeraMono.ttf,sha256=2kKB3H2xej385kpiztkodcWJU0AFXsi6JKORTrl7NJ0,49224
 frontend/types/notebook.ts,sha256=v23RaZe6H3lU5tq6sqnJDPxC2mu0NZFDCJfiN0mgvSs,1359
-more_compute-0.3.0.dist-info/licenses/LICENSE,sha256=0Ot-XIetYt06iay6IhtpJkruD-cLZtjyv7_aIEE-oSc,1073
+more_compute-0.3.2.dist-info/licenses/LICENSE,sha256=0Ot-XIetYt06iay6IhtpJkruD-cLZtjyv7_aIEE-oSc,1073
 morecompute/__init__.py,sha256=pcMVq8Q7qb42AOn7tqgoZJOi3epDDBnEriiv2WVKnXY,87
-morecompute/__version__.py,sha256=VrXpHDu3erkzwl_WXrqINBm9xWkcyUy53IQOj042dOs,22
+morecompute/__version__.py,sha256=vNiWJ14r_cw5t_7UDqDQIVZvladKFGyHH2avsLpN7Vg,22
 morecompute/cli.py,sha256=kVvzvPBqF8xO6UuhU_-TBn99nKwJ405R2mAS6zU0KBc,734
 morecompute/notebook.py,sha256=KEcv0eOEh9N7bPVGoRuKJb47G9MmpQ5zz1B6Dm58w18,4651
 morecompute/process_worker.py,sha256=KsE3r-XpkYGuyO4w3t54VKkD51LfNHAZc3TYattMtrg,7185
@@ -74,7 +74,7 @@ morecompute/server.py,sha256=mdN80iBUixq4KUQfu53aoVsNZ6DM9RaBXc-lwt4bxFk,40285
 morecompute/execution/__init__.py,sha256=jPmBmq8BZWbUEY9XFSpqt5FkgX04uNS10WnUlr7Rnms,134
 morecompute/execution/__main__.py,sha256=pAWB_1bn99u8Gb-tVMSMI-NYvbYbDiwbn40L0a0djeA,202
 morecompute/execution/executor.py,sha256=BesGdx10HSGj8PyPrekdQZzc0tXoqidznEaFITOq8JM,20554
-morecompute/execution/worker.py,sha256=4yZc_Z4A_nhCv_Lv8MaseSe7KYdXPYK5yDeQeDZoVwM,16004
+morecompute/execution/worker.py,sha256=-_-KD89LVA1DmGa1fTqKekttTOGX_JMtUdO02-SV5DE,16388
 morecompute/models/__init__.py,sha256=VLJ5GWi2uTNiZBdvl-ipSbmA6EL8FZHZ5oq-rJmm9z0,640
 morecompute/models/api_models.py,sha256=-ydvi9SeTfdoY9oVPNQS4by-kQGSknx6BHhGD8E2tpo,4553
 morecompute/services/data_manager.py,sha256=c4GKucetMM-VPNbHyzce6bZRvFfmz8kTd5RppLjoLVc,14329
@@ -95,8 +95,8 @@ morecompute/utils/shell_utils.py,sha256=fGFLhQLZU-lmMGALbbS-fKPkhQtmMhZ1FkgQ3Teo
 morecompute/utils/special_commands.py,sha256=IyF9MTINMNNo3P_f56Q6yS_P2HNjA9vohYVxEV7KYnc,17331
 morecompute/utils/system_environment_util.py,sha256=32mQRubo0i4X61o-825T7m-eUSidcEp07qkInP1sWZA,4774
 morecompute/utils/zmq_util.py,sha256=tx7-iS04UN69OFtBzkxcEnRhT7xtI9EzRnrZ_nsH_O0,1889
-more_compute-0.3.0.dist-info/METADATA,sha256=86In89hNjbgE5pKBac7gAQ9YKQD-hVbL3b9IecnskMo,3631
-more_compute-0.3.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-more_compute-0.3.0.dist-info/entry_points.txt,sha256=xp7z9eRPNRM4oxkZZVlyXkhkSjN1AjoYI_B7qpDJ1bI,49
-more_compute-0.3.0.dist-info/top_level.txt,sha256=Tamm6ADzjwaQa1z27O7Izcyhyt9f0gVjMv1_tC810aI,32
-more_compute-0.3.0.dist-info/RECORD,,
+more_compute-0.3.2.dist-info/METADATA,sha256=qO93-kelqgnC3wWl2NzLe_JyMfOnGtBJ3wPY5WqadrI,3631
+more_compute-0.3.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+more_compute-0.3.2.dist-info/entry_points.txt,sha256=xp7z9eRPNRM4oxkZZVlyXkhkSjN1AjoYI_B7qpDJ1bI,49
+more_compute-0.3.2.dist-info/top_level.txt,sha256=Tamm6ADzjwaQa1z27O7Izcyhyt9f0gVjMv1_tC810aI,32
+more_compute-0.3.2.dist-info/RECORD,,

morecompute/__version__.py

@@ -1 +1 @@
-__version__ = "0.3.0"
+__version__ = "0.3.2"

morecompute/execution/worker.py

@@ -332,7 +332,14 @@ def worker_main():
                 if lines:
                     last = lines[-1].strip()
                     # Skip comments and empty lines
-                    if last and not last.startswith('#'):
+                    if not last or last.startswith('#'):
+                        last = None
+                    # Skip orphaned closing brackets from multi-line expressions
+                    # e.g., the ')' from: model = func(\n    arg1,\n    arg2\n)
+                    elif last.lstrip().startswith(')') or last.lstrip().startswith('}') or last.lstrip().startswith(']'):
+                        last = None
+
+                    if last:
                         # Check if it looks like a statement (assignment, import, etc)
                         is_statement = False