Update app.py

app.py

@@ -496,30 +496,11 @@ def welcome_page():
     """, unsafe_allow_html=True)
     
     if st.button("شروع پرسشنامه", key="start_btn", type="primary"):
-        st.session_state.current_page = "contact"
+        st.session_state.current_page = "scenario_explanation"  
         st.session_state.start_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
         st.rerun()
 
 
-def user_contact():
-    """راه ارتباطی ساده"""
-    st.markdown("""
-    <div style="text-align: center; margin-bottom: 30px;">
-        <h3>📩 راه ارتباطی شما (اختیاری)</h3>
-        <p>در صورت تمایل به شرکت در قرعه‌کشی می‌توانید آیدی تلگرام، شماره تماس یا ایمیل خود را وارد کنید:</p>
-    </div>
-    """, unsafe_allow_html=True)
-    
-    contact_info = st.text_input(
-        "راه ارتباطی (اختیاری)",
-        placeholder="مثال: @username یا 09123456789 یا example@email.com",
-        key="user_contact_input"
-    )
-
-    if st.button("ادامه", key="continue_btn", type="primary"):
-        st.session_state.user_contact = contact_info
-        st.session_state.current_page = "demographic"
-        st.rerun()
         
 def demographic_form():
     """فرم اطلاعات دموگرافیک"""
@@ -545,8 +526,28 @@ def demographic_form():
                 "ride_frequency": ride_frequency,
                 "related_education_job": related_education_job
             }
-            st.session_state.current_page = "scenario_explanation"
-            st.rerun()  
+            st.session_state.current_page = "contact"
+            st.rerun() 
+
+def user_contact():
+    """راه ارتباطی ساده"""
+    st.markdown("""
+    <div style="text-align: center; margin-bottom: 30px;">
+        <h3>📩 راه ارتباطی شما (اختیاری)</h3>
+        <p>در صورت تمایل به شرکت در قرعه‌کشی می‌توانید آیدی تلگرام، شماره تماس یا ایمیل خود را وارد کنید:</p>
+    </div>
+    """, unsafe_allow_html=True)
+    
+    contact_info = st.text_input(
+        "راه ارتباطی (اختیاری)",
+        placeholder="مثال: @username یا 09123456789 یا example@email.com",
+        key="user_contact_input"
+    )
+
+    if st.button("ثبت نهایی", key="final_submit", type="primary"):  # تغییر متن دکمه
+        st.session_state.user_contact = contact_info
+        st.session_state.current_page = "thank_you"  # رفتن به صفحه تشکر
+        st.rerun()
   
 def scenario_explanation():
     """توضیح سناریو"""
@@ -796,8 +797,8 @@ def explanation_questions():
                 index=None,
                 key="explanation_type_radio"
             )
-    
-    if st.button("ثبت پاسخ‌ها", type="primary", key="submit_explanation"):
+
+    if st.button("ادامه", type="primary", key="submit_explanation"):
         # استفاده از مقادیر مستقیماً از session_state
         st.session_state.trust = st.session_state.get("trust_radio")
         st.session_state.pricing_method = st.session_state.get("pricing_method_radio")
@@ -845,10 +846,9 @@ def explanation_questions():
                 save_data[q["label"]] = st.session_state.get(q["key"], None)
             
             if save_to_sheet(save_data):
-                st.session_state.current_page = "thank_you"
+                st.session_state.current_page = "demographic"  # تغییر مسیر به demographic
                 st.rerun()
-            else:
-                st.error("خطا در ذخیره‌سازی داده‌ها. لطفاً دوباره تلاش کنید.")
+
                 
 def thank_you_page():
     """صفحه تشکر"""
@@ -873,16 +873,17 @@ def main():
         st.session_state.demographic_data = None
         st.session_state.price_accepted = 0
 
-    pages = {
+        
+        pages = {
         "welcome": welcome_page,
-        "contact": user_contact,
-        "demographic": demographic_form,
         "scenario_explanation": scenario_explanation,
         "map_view": map_view,
         "attention_check1": attention_check1,
         "random_likert_questions": random_likert_questions,
         "attention_check2": attention_check2,
         "explanation_questions": explanation_questions,
+        "demographic": demographic_form,  # دموگرافیک قبل از کانتکت
+        "contact": user_contact,         # کانتکت در انتها
         "thank_you": thank_you_page
     }