diff --git a/src/uncloseai-modal-builtin.css b/src/uncloseai-modal-builtin.css
index 4a786df..f7d27a9 100644
--- a/src/uncloseai-modal-builtin.css
+++ b/src/uncloseai-modal-builtin.css
@@ -1122,13 +1122,21 @@ dialog#uncloseai-embedded-modal[data-theme="dark"] .uncloseai-chat-box {
 /* Code blocks within AI messages - allow horizontal scroll, preserve all whitespace */
 .ai-message pre {
   overflow-x: auto;
-  max-width: 100%;
-  box-sizing: border-box;
+  max-width: 340px;
 }
 
 .ai-message code {
-  max-width: 100%;
-  box-sizing: border-box;
+  max-width: 340px;
+}
+
+@media (min-width: 768px) {
+  .ai-message pre {
+    max-width: 720px;
+  }
+
+  .ai-message code {
+    max-width: 720px;
+  }
 }
 
 .message-text {
diff --git a/src/uncloseai-modal-pico.css b/src/uncloseai-modal-pico.css
index b511d37..80d7ab0 100644
--- a/src/uncloseai-modal-pico.css
+++ b/src/uncloseai-modal-pico.css
@@ -1195,13 +1195,21 @@ dialog#uncloseai-embedded-modal[data-theme="dark"] .uncloseai-chat-box {
 /* Code blocks within AI messages - allow horizontal scroll, preserve all whitespace */
 .ai-message pre {
   overflow-x: auto;
-  max-width: 100%;
-  box-sizing: border-box;
+  max-width: 340px;
 }
 
 .ai-message code {
-  max-width: 100%;
-  box-sizing: border-box;
+  max-width: 340px;
+}
+
+@media (min-width: 768px) {
+  .ai-message pre {
+    max-width: 720px;
+  }
+
+  .ai-message code {
+    max-width: 720px;
+  }
 }
 
 .message-text {