Reuse canonical chat context

src/app/view/chat.rs

@@ -86,6 +86,11 @@ impl App {
                     {
                         continue;
                     }
+                    if message.user
+                        && let Some(stats) = message.generation_stats
+                    {
+                        messages = messages.push(generation_summary(stats));
+                    }
                     let label = if message.user {
                         "You"
                     } else if message.tool {
@@ -193,9 +198,6 @@ impl App {
                     } else {
                         message_body
                     });
-                    if let Some(stats) = message.generation_stats {
-                        messages = messages.push(generation_summary(stats));
-                    }
                 }
                 if let Some(activity) = &self.activity {
                     messages = messages.push(
@@ -388,6 +390,7 @@ fn generation_summary(stats: crate::app::generation::GenerationStats) -> Element
         rule::horizontal(1),
     ]
     .spacing(5)
+    .padding(Padding::ZERO.right(24))
     .into()
 }