Update app.py

app.py

@@ -1202,7 +1202,8 @@ def explanation_questions():
         {
             "key": "price_increase_radio",
             "label": "آیا به نظر شما در این سفر افزایش قیمت نسبت به حالت طبیعی وجود داشته است؟",
-            "options": ["بله", "خیر", "مطمئن نیستم"]
+            "options": ["بله", "خیر", "مطمئن نیستم"],
+            "logical_key": "price_increase"
         },
         {
             "key": "pricing_method_radio",
@@ -1212,17 +1213,20 @@ def explanation_questions():
                 "به صورت خودکار توسط هوش مصنوعی و الگوریتم‌ها",
                 "ترکیبی از هر دو روش",
                 "نظری ندارم"
-            ]
+            ],
+            "logical_key": "pricing_method"
         },
         {
             "key": "trust_radio",
-            "label": "آیا شما به تصمیم گیری هایی که توسط هوش مصنوعی انجام می شود اعتماد دارید؟",
-            "options": ["بله", "خیر", "نظری ندارم"]
+            "label": "آیا شما به تصمیم‌گیری‌هایی که توسط هوش مصنوعی انجام می‌شود اعتماد دارید؟",
+            "options": ["بله", "خیر", "نظری ندارم"],
+            "logical_key": "trust"
         },
         {
             "key": "explanation_received_radio",
             "label": "آیا برای قیمت پیشنهادی این سفر، توضیحی به شما ارائه شد؟",
-            "options": ["بله", "خیر"]
+            "options": ["بله", "خیر"],
+            "logical_key": "explanation_received"
         },
         {
             "key": "explanation_type_radio",
@@ -1232,65 +1236,49 @@ def explanation_questions():
                 "شامل سناریوهای جایگزین که می‌توانستند قیمت متفاوتی ایجاد کنند",
                 "توضیحی دریافت نکردم"
             ],
+            "logical_key": "explanation_type",
             "condition": lambda: st.session_state.get("explanation_received_radio") == "بله"
         }
     ]
 
-    # مقدار اولیه مرحله
+    # مرحله فعلی
     if "explanation_step" not in st.session_state:
         st.session_state.explanation_step = 0
 
-    current_step = st.session_state.explanation_step
+    step = st.session_state.explanation_step
 
-    if current_step < len(questions):
-        q = questions[current_step]
-
-        # اگر شرط خاصی داره (مثل سوال پنجم)، بررسیش کن
-        if "condition" in q and not q["condition"]():
-            st.session_state.explanation_step += 1
+    # وقتی همه سوالات تمام شدند
+    if step >= len(questions):
+        if st.button("اتمام و ادامه"):
+            st.session_state.current_page = "demographic"
             st.rerun()
-            return
-
-        # نمایش سوال فعلی
-        st.radio(
-            q["label"],
-            q["options"],
-            index=None,
-            key=q["key"]
-        )
+        return
 
-        if st.button("ادامه", key=f"continue_{current_step}"):
-            response = st.session_state.get(q["key"])
-            if response is None:
-                st.warning("لطفاً یک گزینه را انتخاب کنید")
-            else:
-                # 🟣 ذخیره پاسخ فعلی در session_state
-                key_mapping = {
-                    "trust_radio": "trust",
-                    "pricing_method_radio": "pricing_method",
-                    "price_increase_radio": "price_increase",
-                    "explanation_received_radio": "explanation_received",
-                    "explanation_type_radio": "explanation_type"
-                }
-                logical_key = key_mapping.get(q["key"])
-                if logical_key:
-                    st.session_state[logical_key] = response
-        
-                st.session_state.explanation_step += 1
-                st.rerun()
-       
-    else:
-        # مرحله آخر: ذخیره داده‌ها و رفتن به فرم دموگرافیک
-        st.session_state.trust = st.session_state.get("trust_radio")
-        st.session_state.pricing_method = st.session_state.get("pricing_method_radio")
-        st.session_state.price_increase = st.session_state.get("price_increase_radio")
-        st.session_state.explanation_received = st.session_state.get("explanation_received_radio")
-        st.session_state.explanation_type = (
-            st.session_state.get("explanation_type_radio") if st.session_state.explanation_received == "بله" else "N/A"
-        )
+    current_q = questions[step]
 
-        st.session_state.current_page = "demographic"
+    # اگر شرط خاصی داره (مثل سوال پنجم)، بررسیش کن
+    if "condition" in current_q and not current_q["condition"]():
+        st.session_state.explanation_step += 1
         st.rerun()
+        return
+
+    # نمایش سوال
+    response = st.radio(
+        current_q["label"],
+        current_q["options"],
+        index=None,
+        key=current_q["key"]
+    )
+
+    if st.button("ادامه", key=f"continue_{step}"):
+        if response is None:
+            st.warning("لطفاً یک گزینه را انتخاب کنید")
+        else:
+            # ذخیره پاسخ فعلی
+            logical_key = current_q["logical_key"]
+            st.session_state[logical_key] = response
+            st.session_state.explanation_step += 1
+            st.rerun()
         
 def demographic_form():
     """فرم اطلاعات دموگرافیک"""