Update app.py

app.py

@@ -6,30 +6,17 @@ def main():
     # Display a placeholder for the message
     message_placeholder = st.empty()
 
-    # JavaScript code to send message to Streamlit
-    js_code = """
-    <script>
-    // Send a message to Streamlit
-    const sendMessageToStreamlit = () => {
-        const message = "Hello from JavaScript!";
-        window.parent.postMessage(message, "*");
-    };
-    // Call the function to send the message
-    sendMessageToStreamlit();
-    </script>
-    """
-
-    # Write the JavaScript code to the Streamlit app
-    st.write(js_code, unsafe_allow_html=True)
-
     # Function to handle messages received from JavaScript
     def receive_message(event):
-        message = event.newValue
-        # Display the received message
-        message_placeholder.write(f"Message received: {message}")
+        if event["type"] == "message":
+            message = event["data"]
+            # Display the received message
+            message_placeholder.write(f"Message received: {message}")
 
-    # Add event listener to listen for messages from JavaScript
-    st.script("window.addEventListener('message', receive_message);")
+    # Add event handler to listen for messages from JavaScript
+    st.add_report_ctx("message_ctx")
+    message_ctx = st.report_ctx.get_report_ctx().enqueue("message_ctx")
+    message_ctx.add_callback(receive_message)
 
 if __name__ == '__main__':
     main()