Update app.py

app.py

@@ -1,6 +1,10 @@
 import streamlit as st
 from streamlit.components.v1 import html
 
+# Debugging Headers and Cookies
+st.write("Headers:", st.context.headers)
+st.write("Cookies:", st.context.cookies)
+
 # Define JavaScript code
 javascript_code = """
 <script>
@@ -57,4 +61,46 @@ st.markdown(
     });
 
     // Dispatch data from CustomEvent to Streamlit
-    
+    document.addEventListener('StreamlitResult', (e) => {
+        const data = e.detail;
+        window.parent.postMessage({ type: 'streamlit:result', data: data }, '*');
+    });
+    </script>
+    """,
+    unsafe_allow_html=True,
+)
+
+# Use an iframe for communication
+if "js_result" not in st.session_state:
+    st.session_state["js_result"] = ""
+
+try:
+    # Listen for result data from JavaScript
+    received_result = html(
+        """
+        <script>
+        window.addEventListener('message', (event) => {
+            if (event.data && event.data.type === 'streamlit:result') {
+                const data = event.data.data;
+                // Send the result back to Streamlit
+                document.dispatchEvent(new CustomEvent('PythonReceivedResult', { detail: data }));
+            }
+        });
+
+        document.addEventListener('PythonReceivedResult', (e) => {
+            const received = e.detail;
+            document.getElementById('result-placeholder').innerText = received;
+        });
+        </script>
+        <div id="result-placeholder"></div>
+        """,
+        height=0,
+    )
+
+    # If there's a new result, update it in session state
+    if received_result and received_result != st.session_state["js_result"]:
+        st.session_state["js_result"] = received_result
+        result_placeholder.write(f"Result from JavaScript: {received_result}")
+
+except Exception as e:
+    st.error(f"An error occurred: {e}")