Consecutive Analyses (#1274)

Author: ahuang11

lumen/ai/coordinator.py

@@ -36,7 +36,7 @@
 from .utils import (
     fuse_messages, log_debug, mutate_user_message, stream_details,
 )
-from .views import LumenOutput
+from .views import AnalysisOutput, LumenOutput
 
 if TYPE_CHECKING:
     from panel.chat.step import ChatStep
@@ -362,6 +362,13 @@ async def run_demo(event):
     async def _add_analysis_suggestions(self):
         pipeline = self._memory["pipeline"]
         current_analysis = self._memory.get("analysis")
+
+        # Clear current_analysis unless the last message is the same AnalysisOutput
+        if current_analysis and self.interface.objects:
+            last_message_obj = self.interface.objects[-1].object
+            if not (isinstance(last_message_obj, AnalysisOutput) and last_message_obj.analysis is current_analysis):
+                current_analysis = None
+
         allow_consecutive = getattr(current_analysis, '_consecutive_calls', True)
         applicable_analyses = []
         for analysis in self._analyses:

lumen/ai/views.py

@@ -288,6 +288,11 @@ class AnalysisOutput(LumenOutput):
     def __init__(self, **params):
         if not params['analysis'].autorun:
             params['active'] = 0
+
+        # Set title based on analysis name if not provided
+        if 'title' not in params or params['title'] is None:
+            params['title'] = type(params['analysis']).__name__
+
         super().__init__(**params)
         controls = self.analysis.controls()
         if controls is not None or not self.analysis.autorun:
@@ -310,7 +315,7 @@ def __init__(self, **params):
                 )
                 self._main.insert(1, ('Config', pn.Column(controls, run_button)))
             with discard_events(self):
-                self._main.active = 1 if self.analysis.autorun else 0
+                self._main.active = 0 if self.analysis.autorun else 1
         self._rendered = True
 
     async def _rerun(self, event):
@@ -324,7 +329,7 @@ async def _rerun(self, event):
             spec = view.to_spec()
             self.param.update(
                 spec=yaml.dump(spec),
-                active=2
+                active=0
             )