diff --git a/.bumpversion.toml b/.bumpversion.toml
index 8074db7..c288ef0 100644
--- a/.bumpversion.toml
+++ b/.bumpversion.toml
@@ -1,5 +1,5 @@
 [tool.bumpversion]
-current_version = "0.1.7"
+current_version = "0.1.8"
 parse = "(?P<major>\\d+)\\.(?P<minor>\\d+)\\.(?P<patch>\\d+)"
 serialize = ["{major}.{minor}.{patch}"]
 search = "{current_version}"
diff --git a/templates/index.html b/templates/index.html
index d4645d8..3587312 100644
--- a/templates/index.html
+++ b/templates/index.html
@@ -21,7 +21,8 @@
             background: black;
             color: white;
             font-family: Arial, sans-serif;
-            position: relative; /* So fixed elements inside work well */
+            position: relative;
+            /* So fixed elements inside work well */
         }
 
         .image-container {
@@ -50,8 +51,21 @@
             border-radius: 10px;
             max-width: 80vw;
             text-align: left;
+            max-height: 30vh;
+            /* NEW: limit height */
+            overflow-y: auto;
+            /* NEW: allow scrolling */
         }
 
+        /* Optional: Adjust height for smaller screens */
+        @media (max-width: 768px) {
+            .prompt {
+                max-height: 25vh;
+                /* even smaller on mobile */
+            }
+        }
+
+
         .button-group {
             display: flex;
             gap: 20px;
@@ -116,4 +130,4 @@
     <div class="version">v{{ version }}</div>
 </body>
 
-</html>
+</html>
\ No newline at end of file