send height

app.py

@@ -126,3 +126,25 @@ if __name__ == "__main__":
             with st.chat_message("assistant"):
                 st.markdown(st.session_state.messages[i]["content"])
 
+import streamlit.components.v1 as components
+components.html(
+    """
+    <script>
+      function sendHeightWhenReady() {
+        const el = window.parent.document.getElementsByClassName('st-emotion-cache-b95f0i')[0];
+        if (el) {
+          const height = el.scrollHeight;
+          console.log("Sending height to parent:", height);
+          window.parent.parent.postMessage({ type: 'setHeight', height: height }, '*');
+        } else {
+          // Retry in 100ms until the element appears
+          setTimeout(sendHeightWhenReady, 1000);
+        }
+      }
+
+      window.onload = sendHeightWhenReady;
+      window.addEventListener('resize', sendHeightWhenReady);
+      setInterval(sendHeightWhenReady, 1000);
+    </script>
+    """
+)