Upload viz.py

utils/viz.py

@@ -37,7 +37,7 @@ def create_task_success_heatmap(df, benchmark_name):
     fig.update_layout(
         xaxis_title='Task ID',
         height=total_height,
-        width=1150,
+        # width=1150,
         yaxis=dict(
             autorange='reversed',
             showticklabels=True,  # Show y-axis tick labels (agent names)
@@ -171,7 +171,7 @@ def create_scatter_plot(df, x: str, y: str, x_label: str = None, y_label: str =
     fig.update_xaxes(rangemode="tozero")
 
     fig.update_layout(
-    width = 1150,
+    # width = 1150,
     height = 600,
     xaxis_title = x_label,
     yaxis_title = y_label,
@@ -263,10 +263,10 @@ def create_flow_chart(steps):
         # Wrap text to improve readability
         wrapped_description = '<br>'.join(textwrap.wrap(description, width=90, max_lines=20))
         wrapped_assessment = '<br>'.join(textwrap.wrap(assessment, width=90, max_lines=10))
-        wrapped_outline = textwrap.shorten(step_headline, width=30, placeholder='')
+        wrapped_outline = textwrap.shorten(step_headline, width=50, placeholder='')
         wrapped_outline = '' if wrapped_outline == '' else f": {wrapped_outline}"
 
-        node_text_outline = '' if wrapped_outline == '' else f":<br>{textwrap.shorten(step_headline, width=30, placeholder='')}"
+        node_text_outline = '' if wrapped_outline == '' else f":<br>{'<br>'.join(textwrap.wrap(step_headline, width=30, placeholder=''))}"
         node_text.append(f"Step {i+1}{node_text_outline}")
         
         # Create formatted hover text without indentation