Update app.py

app.py

@@ -419,7 +419,6 @@ def save_to_sheet(data):
         spreadsheet = client.open_by_key(SHEET_ID)
         worksheet = spreadsheet.worksheet(SHEET_NAME)
         
-        # آماده‌سازی داده‌ها برای ذخیره‌سازی
         row_data = [
             data.get("start_time", ""),  # زمان شروع
             data.get("scenario_type", ""),
@@ -435,6 +434,10 @@ def save_to_sheet(data):
             data.get("attention_check1", ""),
             data.get("attention_check2", ""),
             
+            # سوالات جدید
+            data.get("pricing_method", ""),
+            data.get("price_increase", ""),
+            
             # سوالات informational (5 گزینه‌ای)
             data.get("informational_1", ""),
             data.get("informational_2", ""),
@@ -717,40 +720,74 @@ def explanation_questions():
     """, unsafe_allow_html=True)
     st.markdown("### 📋 سوالات تکمیلی")
 
-    # سوال اول
-    explanation_received = st.radio(
-        "آیا برای قیمت پیشنهادی این سفر، توضیحی به شما ارائه شد؟",
-        ["بله", "خیر"],
+    # سوال جدید 1: روش قیمت‌گذاری
+    pricing_method = st.radio(
+        "به نظر شما پلتفرم قیمت را چگونه تعیین می‌کند؟",
+        [
+            "به صورت دستی توسط تیم پلتفرم",
+            "به صورت خودکار توسط هوش مصنوعی و الگوریتم‌ها",
+            "ترکیبی از هر دو روش",
+            "نظری ندارم"
+        ],
         index=None,
-        key="explanation_received"
+        key="pricing_method"
     )
-    
-    # سوال دوم (فقط اگر پاسخ بله باشد)
+
+    # سوال جدید 2: افزایش قیمت
+    price_increase = None
+    if pricing_method is not None:  # فقط اگر به سوال اول پاسخ داده شده باشد
+        price_increase = st.radio(
+            "آیا به نظر شما در این سفر افزایش قیمت نسبت به حالت طبیعی وجود داشته است؟",
+            ["بله", "خیر", "مطمئن نیستم"],
+            index=None,
+            key="price_increase"
+        )
+
+    # سوالات قبلی
+    explanation_received = None
     explanation_type = None
-    if explanation_received == "بله":
-        explanation_type = st.radio(
-            "اگر توضیحی دریافت کردید، این توضیح بیشتر به کدام مورد شباهت داشت؟",
-            [
-                "بر اساس عواملی که در قیمت‌گذاری لحاظ شده‌اند",
-                "شامل سناریوهای جایگزین که می‌توانستند قیمت متفاوتی ایجاد کنند",
-                "توضیحی دریافت نکردم"
-            ],
+    if price_increase is not None:  # فقط اگر به سوال دوم پاسخ داده شده باشد
+        explanation_received = st.radio(
+            "آیا برای قیمت پیشنهادی این سفر، توضیحی به شما ارائه شد؟",
+            ["بله", "خیر"],
             index=None,
-            key="explanation_type"
+            key="explanation_received"
         )
+        
+        # سوال دوم (فقط اگر پاسخ بله باشد)
+        if explanation_received == "بله":
+            explanation_type = st.radio(
+                "اگر توضیحی دریافت کردید، این توضیح بیشتر به کدام مورد شباهت داشت؟",
+                [
+                    "بر اساس عواملی که در قیمت‌گذاری لحاظ شده‌اند",
+                    "شامل سناریوهای جایگزین که می‌توانستند قیمت متفاوتی ایجاد کنند",
+                    "توضیحی دریافت نکردم"
+                ],
+                index=None,
+                key="explanation_type"
+            )
     
     if st.button("ثبت پاسخ‌ها", type="primary"):
-        if explanation_received is None:
-            st.warning("لطفاً به سوال اول پاسخ دهید")
+        # بررسی کامل بودن پاسخ‌ها
+        if pricing_method is None:
+            st.warning("لطفاً به سوال روش قیمت‌گذاری پاسخ دهید")
+        elif price_increase is None:
+            st.warning("لطفاً به سوال افزایش قیمت پاسخ دهید")
+        elif explanation_received is None:
+            st.warning("لطفاً به سوال دریافت توضیح پاسخ دهید")
         elif explanation_received == "بله" and explanation_type is None:
-            st.warning("لطفاً به سوال دوم پاسخ دهید")
+            st.warning("لطفاً به سوال نوع توضیح پاسخ دهید")
         else:
+            # ذخیره پاسخ‌های جدید در session_state
+            st.session_state.pricing_method = pricing_method
+            st.session_state.price_increase = price_increase
+            
+            # ذخیره پاسخ‌های قبلی
             st.session_state.explanation_data = {
                 "explanation_received": explanation_received,
                 "explanation_type": explanation_type if explanation_received == "بله" else "N/A"
             }
 
-            
             # جمع‌آوری تمام داده‌ها برای ذخیره‌سازی
             end_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
             start_time = datetime.strptime(st.session_state.start_time, "%Y-%m-%d %H:%M:%S")
@@ -766,6 +803,8 @@ def explanation_questions():
                 "price_accepted": st.session_state.get("price_accepted", 0),
                 "attention_check1": st.session_state.attention_check1,
                 "attention_check2": st.session_state.attention_check2, 
+                "pricing_method": pricing_method,
+                "price_increase": price_increase,
                 "explanation_received": explanation_received,
                 "explanation_type": explanation_type if explanation_received == "بله" else "N/A",
                 **st.session_state.demographic_data
@@ -780,8 +819,7 @@ def explanation_questions():
                 st.rerun()
             else:
                 st.error("خطا در ذخیره‌سازی داده‌ها. لطفاً دوباره تلاش کنید.")
-
-
+                
 def thank_you_page():
     """صفحه تشکر"""
     st.success("✅ پاسخ‌های شما با موفقیت ثبت شد. با تشکر از مشارکت شما در این تحقیق!")