Update app.py

app.py

@@ -3,21 +3,20 @@ import streamlit as st
 def main():
     st.title('Message Receiver')
 
-    # JavaScript code to handle messages received from the iframe
+    # Display a placeholder for the received message
+    received_message_placeholder = st.empty()
+
+    # JavaScript code to handle messages received from the parent iframe
     js_code = """
     <script>
-    // Function to handle messages received from the iframe
-    const receiveMessageFromIframe = (event) => {
-        // Check the origin of the message to ensure security
-        if (event.origin !== 'http://origin_of_your_iframe') {
-            return;
-        }
+    // Function to handle messages received from the parent iframe
+    const receiveMessageFromParent = (event) => {
         // Update the Streamlit title with the received message
-        document.title = 'Message Receiver - ' + event.data;
+        document.title = 'Message Receiver - ' + event.data.message;
     };
 
-    // Add event listener to listen for messages from the iframe
-    window.addEventListener('message', receiveMessageFromIframe);
+    // Add event listener to listen for messages from the parent iframe
+    window.addEventListener('message', receiveMessageFromParent);
     </script>
     """