Update app.py

app.py

@@ -20,14 +20,7 @@ PROMPT = """
 4. その日の計画: {plan}.
 5. 日本語で100語以内で説明してください。
 """
-html_code = """
-<script>
-window.addEventListener('message', (_ev) => {
-  const elmIframe = document.querySelector('#parentframe');
-  elmIframe.style.height = _ev.data + 'px';
-});
-</script>
-"""
+
 
 def init_page():
     st.set_page_config(
@@ -57,7 +50,16 @@ def init_chain():
     output_parser = StrOutputParser()
     chain = prompt | llm | output_parser
     return chain
-
+    
+def send_height():
+    st.components.v1.html("""
+        <script>
+        window.onload = () => {
+            const height = document.body.scrollHeight;
+            window.parent.postMessage({ type: 'resize', height: height }, '*');
+        };
+        </script>
+    """, height=0)
        
 
 def main():
@@ -66,7 +68,7 @@ def main():
     st.markdown("""<style>.st-emotion-cache-15ecox0 { display: none !important; }
     @media (max-width: 50.5rem) {.st-emotion-cache-13ln4jf {max-width: calc(0rem + 100vw);}}
     </style>""",unsafe_allow_html=True,)
-    components.html(html_code, height=0)
+    send_height()
     chain = init_chain()
     if chain:
         feeling = st.selectbox(