fix

Author: misrasaurabh1

codeflash/cli_cmds/console.py

@@ -129,6 +129,9 @@ def code_print(
 
 spinners = cycle(SPINNER_TYPES)
 
+# Track whether a progress bar is already active to prevent nested Live displays
+_progress_bar_active = False
+
 
 @contextmanager
 def progress_bar(
@@ -138,28 +141,38 @@ def progress_bar(
 
     If revert_to_print is True, falls back to printing a single logger.info message
     instead of showing a progress bar.
+
+    If a progress bar is already active, yields a dummy task ID to avoid Rich's
+    LiveError from nested Live displays.
     """
+    global _progress_bar_active
+
     if is_LSP_enabled():
         lsp_log(LspTextMessage(text=message, takes_time=True))
         yield
         return
 
-    if revert_to_print:
-        logger.info(message)
+    if revert_to_print or _progress_bar_active:
+        if revert_to_print:
+            logger.info(message)
 
         # Create a fake task ID since we still need to yield something
         yield DummyTask().id
     else:
-        progress = Progress(
-            SpinnerColumn(next(spinners)),
-            *Progress.get_default_columns(),
-            TimeElapsedColumn(),
-            console=console,
-            transient=transient,
-        )
-        task = progress.add_task(message, total=None)
-        with progress:
-            yield task
+        _progress_bar_active = True
+        try:
+            progress = Progress(
+                SpinnerColumn(next(spinners)),
+                *Progress.get_default_columns(),
+                TimeElapsedColumn(),
+                console=console,
+                transient=transient,
+            )
+            task = progress.add_task(message, total=None)
+            with progress:
+                yield task
+        finally:
+            _progress_bar_active = False
 
 
 @contextmanager

codeflash/optimization/function_optimizer.py

@@ -369,9 +369,7 @@ def _process_candidates(
                 logger.info(success_msg.format(len(candidates), self.candidate_len))
 
             callback()
-        # Call get_next_candidate() OUTSIDE the progress_bar context to avoid
-        # nested progress bars (Rich only allows one active Live display at a time)
-        return self.get_next_candidate()
+            return self.get_next_candidate()
 
     def _filter_refined_candidates(self, candidates: list[OptimizedCandidate]) -> list[OptimizedCandidate]:
         """We generate a weighted ranking based on the runtime and diff lines and select the best of valid optimizations to be tested."""