Update app.py

app.py

@@ -601,41 +601,41 @@ def attention_check1():
         key="att1_radio"
     )
 
-    # 1. دکمه HTML واقعی با JavaScript
-    html_code = """
+        # 1. تزریق JavaScript برای ارتباط دوطرفه
+    js_code = """
     <script>
-    function handleContinue() {
-        // ارسال داده به Streamlit
-        parent.document.dispatchEvent(new CustomEvent('continueGreenClicked'));
+    function handleClick() {
+        // روش 1: ارسال به sessionState
+        Streamlit.setComponentValue("btn_clicked");
+        
+        // روش 2: یا از طریق postMessage
+        // window.parent.postMessage("continue_green_clicked", "*");
     }
     </script>
     
-    <button onclick="handleContinue()"
+    <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(html_code, height=50)
+    html(js_code, height=60)
 
-    # 2. دریافت رویداد در Streamlit
-    if st.session_state.get("continue_green_clicked"):
+    # 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  # ریست وضعیت
 
-    # کد JavaScript برای دریافت رویداد (در بخش اصلی برنامه)
-    st.components.v1.html("""
-    <script>
-    document.addEventListener('continueGreenClicked', function() {
-        Streamlit.setComponentValue("continue_green_clicked");
-    });
-    </script>
-    """)
+# 3. مقداردهی اولیه session state
+if "btn_clicked" not in st.session_state:
+    st.session_state.btn_clicked = False
 
 
 def attention_check2():