headline fix

utils/viz.py

@@ -123,19 +123,19 @@ def create_flow_chart(steps):
         assessment = analysis.get('assessment', 'No assessment available.')
         success = analysis.get('success', True)  # Assuming True if not specified
         action_type = analysis.get('action_type', 'other')  # Default to 'other' if not specified
-        step_outline = analysis.get('step_outline', '')
+        step_headline = analysis.get('headline', '')
         
         # Set node color and shape based on attributes
         node_colors.append(color_map[success])
         node_shapes.append(shape_map.get(action_type, 'circle'))
         
         # Wrap text to improve readability
-        wrapped_description = '<br>'.join(textwrap.wrap(description, width=50))
-        wrapped_assessment = '<br>'.join(textwrap.wrap(assessment, width=50))
-        wrapped_outline = textwrap.shorten(step_outline, width=30, placeholder='')
+        wrapped_description = '<br>'.join(textwrap.wrap(description, width=90, max_lines=10))
+        wrapped_assessment = '<br>'.join(textwrap.wrap(assessment, width=90, max_lines=10))
+        wrapped_outline = textwrap.shorten(step_headline, width=30, placeholder='')
         wrapped_outline = '' if wrapped_outline == '' else f": {wrapped_outline}"
 
-        node_text_outline = '' if wrapped_outline == '' else f":<br>{textwrap.shorten(step_outline, width=30, placeholder='')}"
+        node_text_outline = '' if wrapped_outline == '' else f":<br>{textwrap.shorten(step_headline, width=30, placeholder='')}"
         node_text.append(f"Step {i+1}{node_text_outline}")
         
         # Create formatted hover text without indentation