Try to change style when embedding, so that the height is not fixed.

app.py

@@ -23,13 +23,20 @@ context = read_dataset(dataset)
 st.markdown(
     """
     <style>
-    .scrollable-div {
-        height: 200px;
-        width: 100%;
-        overflow-y: auto;
-        padding: 10px;
-        border: 1px solid #ccc;
-    }
+        .scrollable-div {
+            height: 200px;
+            width: 100%;
+            overflow-y: auto;
+            padding: 10px;
+            border: 1px solid #ccc;
+        }
+    
+       .block-container {
+            padding-top: 1rem;
+            padding-bottom: 0rem;
+            padding-left: 5rem;
+            padding-right: 5rem;
+        }
     </style>
     """,
     unsafe_allow_html=True