fixed step headline not showing

app.py

@@ -169,7 +169,7 @@ def format_call_info(step, step_index):
         }}
     </style>
 
-    <h2>Step {step_index + 1}: {analysis.get('headline', 'N/A')}</h2>
+    <h2>Step {step_index + 1}: {analysis.get('headline', '')}</h2>
 
     <h3>Call Metadata</h3>
     <ul>
@@ -270,7 +270,7 @@ with gr.Blocks() as demo:
                 if not analyzed_traces or task_id not in analyzed_traces:
                     return gr.Dropdown(choices=[], label="Select Step", value="No data available.")
                 steps = analyzed_traces[task_id]['steps']
-                return gr.Dropdown(choices=[(f"Step {i+1}", i) for i in range(len(steps))], label="Select Step", value=0)
+                return gr.Dropdown(choices=[(f"Step {i+1}", i) for i in range(len(steps))], label="Select Step", value=0), format_call_info(steps[0], 0)
 
             def update_raw_call_details(agent_name, task_id, step_index):
                 analyzed_traces = get_analyzed_traces(agent_name, "usaco")
@@ -298,7 +298,7 @@ with gr.Blocks() as demo:
                                     outputs=[raw_task_dropdown, raw_step_dropdown, raw_call_details])
             raw_task_dropdown.change(update_raw_step_dropdown, 
                                     inputs=[raw_agent_dropdown, raw_task_dropdown], 
-                                    outputs=[raw_step_dropdown])
+                                    outputs=[raw_step_dropdown, raw_call_details])
             raw_step_dropdown.change(update_raw_call_details, 
                                     inputs=[raw_agent_dropdown, raw_task_dropdown, raw_step_dropdown], 
                                     outputs=[raw_call_details])