fix

static/style.css

@@ -7,9 +7,12 @@ html {
 #chat-messages .message-content pre {
     overflow-x: auto;
     white-space: pre;
-    max-width: 662px;
     position: relative;
-    border-radius: 8px;
+    max-width: 685px;
+    border-radius: 0px 0px 8px 8px;
+    background-color: #2c2c2c;
+    margin-top: 0;
+    /* Remove top margin to align with header */
 }
 
 #chat-messages .message-content pre code {
@@ -18,10 +21,24 @@ html {
     white-space: inherit;
 }
 
+.code-header {
+    display: flex;
+    justify-content: space-between;
+    align-items: center;
+    background-color: #515151;
+    border-radius: 8px 8px 0 0;
+    /* Rounded corners on top */
+    padding: 5px 10px 10px 10px;
+    margin-bottom: -8px;
+    /* Overlap with pre element */
+}
+
 .copy-button {
-    position: absolute;
-    top: 5px;
-    right: 5px;
+    background: none;
+    border: none;
+    cursor: pointer;
+    font-size: 14px;
+    color: #ffffff;
 }
 
 .content {
@@ -30,6 +47,7 @@ html {
     color: #ffffff;
 }
 
+
 /* Style for the overall chat container */
 
 #chat-messages {