diff --git a/webui/templates/visualize.html b/webui/templates/visualize.html
index 1fe2076..36eb296 100644
--- a/webui/templates/visualize.html
+++ b/webui/templates/visualize.html
@@ -30,10 +30,13 @@
         display: flex;
         gap: 20px;
         align-items: flex-start;
+        justify-content: space-between;
+        margin-right: 20px;
     }
 
     .column {
-        flex: 0 0 auto;
+        flex: 1;
+        max-width: 80vw;
         padding: 20px;
         background-color: #f8f8f8;
         border: 1px solid #ddd;
@@ -41,6 +44,14 @@
         box-shadow: 2px 2px 5px rgba(0, 0, 0, 0.1);
     }
 
+    .column:first-child {
+        flex: 0 0 30%;
+    }
+
+    .column:last-child {
+        flex: 0 0 65%;
+    }
+
     .generated-text-container {
         margin-bottom: 20px;
     }
@@ -91,6 +102,14 @@
         padding: 5px;
         margin-bottom: 5px;
         min-height: 100px;
+        width: 95%;
+        max-width: 95%;
+        overflow-x: auto;
+        /* Add horizontal scroll for long text */
+        white-space: pre-wrap;
+        /* Preserve line breaks and wrap text */
+        word-break: break-all;
+        /* Force break words even without spaces */
     }
 
     .index-display {