Update app.py

app.py

@@ -1,9 +1,6 @@
 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 datetime import datetime
 import gspread
 from google.oauth2.service_account import Credentials
@@ -603,23 +600,13 @@ def attention_check1():
         key="att1_radio"
     )
 
-    # استفاده از کامپوننت اختصاصی
-    custom_button("ادامه", key="continue_btn")
-
-    # بررسی کلیک
-    if st.session_state.get("continue_btn_clicked"):
-        if not answer:
-            st.warning("لطفاً یک گزینه را انتخاب کنید")
+    if st.button("ادامه", key="att1_btn"):
+        if answer:
+            st.session_state.attention_check1 = answer  # ذخیره پاسخ در session_state
+            st.session_state.current_page = "random_likert_questions"
+            st.rerun()
         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()
+            st.warning("لطفاً یک گزینه را انتخاب کنید")
 
 
 def attention_check2():
@@ -914,14 +901,4 @@ 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()