Update app.py

app.py

@@ -2,6 +2,9 @@ import streamlit as st
 import folium
 from streamlit_folium import folium_static
 from streamlit.components.v1 import html
+import custom_button
+from streamlit.runtime.scriptrunner import RerunData, RerunException
+from streamlit.source_util import get_pages
 from datetime import datetime
 import gspread
 from google.oauth2.service_account import Credentials
@@ -601,41 +604,23 @@ def attention_check1():
         key="att1_radio"
     )
 
-        # 1. تزریق JavaScript برای ارتباط دوطرفه
-    js_code = """
-    <script>
-    function handleClick() {
-        // روش 1: ارسال به sessionState
-        Streamlit.setComponentValue("btn_clicked");
-        
-        // روش 2: یا از طریق postMessage
-        // window.parent.postMessage("continue_green_clicked", "*");
-    }
-    </script>
-    
-    <button onclick="handleClick()"
-            style="background-color:#28a745; color:white; border:none;
-                   border-radius:8px; padding:10px 20px; font-weight:bold;
-                   width:100%; cursor:pointer; margin-top:10px;">
-        ادامه
-    </button>
-    """
-    html(js_code, height=60)
-
-    # 2. بررسی کلیک
-    if st.session_state.get("btn_clicked"):
-        if answer:
-            st.session_state.attention_check1 = answer
-            st.session_state.current_page = "random_likert_questions"
-            time.sleep(0.1)  # تأخیر کوتاه برای اطمینان
-            st.rerun()
-        else:
-            st.warning("لطفاً یک گزینه را انتخاب کنید")
-            st.session_state.btn_clicked = False  # ریست وضعیت
+    # استفاده از کامپوننت اختصاصی
+    custom_button("ادامه", key="continue_btn")
 
-# 3. مقداردهی اولیه session state
-if "btn_clicked" not in st.session_state:
-    st.session_state.btn_clicked = False
+    # بررسی کلیک
+    if st.session_state.get("continue_btn_clicked"):
+        if not answer:
+            st.warning("لطفاً یک گزینه را انتخاب کنید")
+        else:
+            st.session_state.attention_check1 = answer
+            # تغییر صفحه به صورت دستی
+            def update_page():
+                pages = get_pages("")
+                pages["random_likert_questions"].is_active = True
+                pages["current_page"].is_active = False
+                raise RerunException(RerunData())
+            
+            update_page()
 
 
 def attention_check2():
@@ -930,4 +915,14 @@ def main():
     pages[st.session_state.current_page]()
 
 if __name__ == "__main__":
+    # دریافت پیام از دکمه
+    if "continue_btn_clicked" not in st.session_state:
+        st.session_state.continue_btn_clicked = False
+        
+    # دریافت پیام از iframe
+    from streamlit.web.server.websocket_headers import _get_websocket_headers
+    headers = _get_websocket_headers()
+    if headers and headers.get("Referer"):
+        if "continue_btn_clicked" in headers.get("Referer"):
+            st.session_state.continue_btn_clicked = True
     main()