Update app.py

app.py

@@ -1,4 +1,6 @@
 import streamlit as st
+import folium
+from streamlit_folium import folium_static
 import random
 from datetime import datetime
 import gspread
@@ -7,94 +9,97 @@ import os
 import json
 
 # تنظیمات اولیه
-st.set_page_config(layout="wide")
-
-# ========== بخش سناریوها ==========
-SCENARIOS = {
-    "control": {
-        "title": "سناریوی کنترل (بدون توضیح)",
-        "description": "شما قصد دارید سفری را از مبدأ X به مقصد Y درخواست کنید. قیمت پیشنهادی این سفر طبق تصمیم هوش مصنوعی ۱۰۰ هزار تومان است."
-    },
-    "input_based": {
-        "title": "سناریوی مبتنی بر ورودی",
-        "description": """شما قصد دارید سفری را از مبدأ X به مقصد Y درخواست کنید. قیمت پیشنهادی این سفر طبق تصمیم هوش مصنوعی ۱۰۰ هزار تومان است.
-        
-        سازمان این توضیحات را به شما ارائه می‌دهد:
-        این قیمت بر اساس چند عامل تعیین شده است:
-        • سطح تقاضا در منطقه: زیاد (+)
-        • تعداد رانندگان فعال: کم (+)
-        • زمان روز: ساعت اوج ترافیک (+)
-        • شرایط جوی: هوای بارانی(++)
-
-        به این دلیل، قیمت این سفر بالاتر از حد معمول است."""
-    },
-    "counterfactual": {
-        "title": "سناریوی پادواقعی",
-        "description": """شما قصد دارید سفری را از مبدأ X به مقصد Y درخواست کنید. قیمت پیشنهادی این سفر طبق تصمیم هوش مصنوعی ۱۰۰ هزار تومان است.
-        
-        سازمان این توضیحات را به شما ارائه می‌دهد:
-        اگر این سفر را ۳۰ دقیقه زودتر یا دیرتر درخواست می‌کردید، به دلیل سطح تقاضای کمتر، رانندگان فعال بیش‌تر، زمان بهتر روز و شرایط جوی بهتر، ممکن بود قیمت ۱۵٪ کمتر باشد.
-        اگر تعداد رانندگان فعال در منطقه شما دو برابر بود، قیمت‌گذاری نوسانی اعمال نمی‌شد."""
-    }
-}
-
-# ========== بخش نقشه ==========
-def show_map():
-    # می‌توانید از folium یا تصاویر استاتیک استفاده کنید
-    st.image("map_placeholder.png", caption="مسیر سفر از X به Y")
-
-# ========== بخش جمع‌آوری داده‌ها ==========
-def get_demographic_data():
-    with st.form("demographic_form"):
-        st.header("اطلاعات دموگرافیک")
-        age = st.number_input("سن", min_value=18, max_value=100)
-        gender = st.selectbox("جنسیت", ["مرد", "زن", "سایر"])
-        education = st.selectbox("تحصیلات", ["دیپلم", "لیسانس", "فوق لیسانس", "دکترا"])
-        ride_frequency = st.selectbox("دفعات استفاده از سرویس‌های اشتراک سفر در ماه", ["کمتر از 5 بار", "5-10 بار", "بیش از 10 بار"])
-        
-        if st.form_submit_button("ذخیره اطلاعات دموگرافیک"):
-            return {
-                "age": age,
-                "gender": gender,
-                "education": education,
-                "ride_frequency": ride_frequency
-            }
-    return None
-
-# ========== پرسشنامه عدالت قیمتی ==========
-def fairness_questionnaire():
-    st.header("پرسشنامه ادراک انصاف قیمتی")
-    
-    fairness = st.slider("به نظر شما این قیمت چقدر منصفانه است؟ (1 کاملاً ناعادلانه - 7 کاملاً عادلانه)", 1, 7)
-    transparency = st.slider("چقدر توضیحات ارائه شده در مورد قیمت را شفاف می‌دانید؟ (1 کاملاً نامشخص - 7 کاملاً شفاف)", 1, 7)
-    satisfaction = st.slider("چقدر از این قیمت‌گذاری راضی هستید؟ (1 کاملاً ناراضی - 7 کاملاً راضی)", 1, 7)
-    
-    return {
-        "fairness": fairness,
-        "transparency": transparency,
-        "satisfaction": satisfaction
-    }
-
+st.set_page_config(layout="wide", page_title="تحلیل انصاف قیمتی", page_icon="📊")
 
 # ========== تنظیمات دیتا ==========
 SHEET_ID = "1mmdWAyOCYq4yXMgP53Duq712AnlqZWLkfIo76JqM7wM"
 SHEET_NAME = "Condition1"
 
-# ========== توابع ==========
+# ========== استایل‌های سفارشی ==========
+st.markdown("""
+<style>
+    .ride-header {
+        background-color: #3d5a80;
+        padding: 15px;
+        border-radius: 10px;
+        color: white;
+        margin-bottom: 20px;
+    }
+    .price-box {
+        border: 2px solid #3d5a80;
+        border-radius: 10px;
+        padding: 15px;
+        margin: 15px 0;
+        text-align: center;
+        font-size: 24px;
+        font-weight: bold;
+    }
+    .explanation-box {
+        background-color: #f8f9fa;
+        border-radius: 10px;
+        padding: 15px;
+        margin: 15px 0;
+        border-right: 5px solid #3d5a80;
+    }
+    .survey-section {
+        background-color: #f0f2f6;
+        border-radius: 10px;
+        padding: 20px;
+        margin-top: 30px;
+    }
+</style>
+""", unsafe_allow_html=True)
+
+# ========== توابع اصلی ==========
+def create_ride_map():
+    """ایجاد نقشه سفر با Folium"""
+    start_point = [35.6892, 51.3890]  # تهران
+    end_point = [35.8400, 50.9391]    # کرج
+    
+    m = folium.Map(location=[35.7, 51.3], zoom_start=11)
+    folium.Marker(start_point, popup="<b>مبدأ:</b> میدان انقلاب", icon=folium.Icon(color="green")).add_to(m)
+    folium.Marker(end_point, popup="<b>مقصد:</b> میدان امام خمینی", icon=folium.Icon(color="red")).add_to(m)
+    folium.PolyLine([start_point, end_point], color="blue", weight=2.5).add_to(m)
+    return m
+
+def generate_random_price():
+    """تولید قیمت تصادفی"""
+    return random.randint(80, 200) * 1000  # قیمت به تومان
+
+def show_explanation(exp_type):
+    """نمایش توضیحات قیمت"""
+    explanations = {
+        "control": [],
+        "input": [
+            "• سطح تقاضا در منطقه: زیاد (+)",
+            "• تعداد رانندگان فعال: کم (+)",
+            "• زمان روز: ساعت اوج ترافیک (+)",
+            "• شرایط جوی: هوای بارانی (++)"
+        ],
+        "counterfactual": [
+            "اگر این سفر را ۳۰ دقیقه زودتر یا دیرتر درخواست می‌کردید، قیمت ۱۵٪ کمتر می‌شد",
+            "اگر تعداد رانندگان فعال دو برابر بود، قیمت‌گذاری نوسانی اعمال نمی‌شد"
+        ]
+    }
+    
+    if exp_type != "control":
+        with st.expander("🔍 توضیحات قیمت", expanded=True):
+            st.markdown("<div class='explanation-box'><h4>علت قیمت گذاری:</h4>", unsafe_allow_html=True)
+            for item in explanations.get(exp_type, []):
+                st.markdown(f"<p>{item}</p>", unsafe_allow_html=True)
+            st.markdown("</div>", unsafe_allow_html=True)
+    return exp_type
+
+# ========== توابع مدیریت داده‌ها ==========
 def get_credentials():
     """دریافت اعتبارنامه از Secrets"""
     try:
-        # دریافت اطلاعات از secrets Hugging Face
         service_account_json = os.environ.get('GCP_SERVICE_ACCOUNT')
-        
         if not service_account_json:
             st.error("مقدار GCP_SERVICE_ACCOUNT در محیط یافت نشد")
             return None
             
-        # تبدیل رشته JSON به دیکشنری
         service_account_info = json.loads(service_account_json)
-        
-        # ایجاد اعتبارنامه
         creds = Credentials.from_service_account_info(
             service_account_info,
             scopes=[
@@ -106,13 +111,6 @@ def get_credentials():
     except Exception as e:
         st.error(f"خطا در دریافت اعتبارنامه: {str(e)}")
         return None
-        
-
-
-
-
-
-# ========== ذخیره در Google Sheets ==========
 
 def save_to_sheet(data):
     try:
@@ -126,7 +124,8 @@ def save_to_sheet(data):
             
         row_data = [
             datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
-            data.get("scenario"),
+            data.get("scenario_type"),
+            data.get("price"),
             data.get("age"),
             data.get("gender"),
             data.get("education"),
@@ -143,36 +142,111 @@ def save_to_sheet(data):
         st.error(f"خطا در ذخیره‌سازی: {str(e)}")
         return False
 
+# ========== بخش‌های فرم ==========
+def demographic_form():
+    """فرم اطلاعات دموگرافیک"""
+    with st.form("demographic"):
+        st.header("📝 اطلاعات دموگرافیک")
+        age = st.number_input("سن", min_value=18, max_value=100)
+        gender = st.selectbox("جنسیت", ["مرد", "زن", "سایر"])
+        education = st.selectbox("تحصیلات", ["دیپلم", "لیسانس", "فوق لیسانس", "دکترا"])
+        ride_frequency = st.selectbox("دفعات استفاده از سرویس‌های اشتراک سفر در ماه", 
+                                    ["کمتر از 5 بار", "5-10 بار", "بیش از 10 بار"])
+        
+        if st.form_submit_button("ذخیره اطلاعات"):
+            return {
+                "age": age,
+                "gender": gender,
+                "education": education,
+                "ride_frequency": ride_frequency
+            }
+    return None
 
-
-
-# ========== اجرای اصلی برنامه ==========
-def main():
-    if 'current_scenario' not in st.session_state:
-        st.session_state.current_scenario = random.choice(list(SCENARIOS.keys()))
-    
-    scenario = SCENARIOS[st.session_state.current_scenario]
+def fairness_questionnaire():
+    """پرسشنامه ادراک انصاف"""
+    st.header("📊 پرسشنامه ادراک انصاف قیمتی")
     
-    st.title(scenario["title"])
-    show_map()
-    st.markdown(scenario["description"])
+    st.markdown("**لطفاً با توجه به قیمت و توضیحات ارائه شده به سوالات زیر پاسخ دهید:**")
     
-    demographic_data = get_demographic_data()
+    fairness = st.slider("به نظر شما این قیمت چقدر منصفانه است؟ (1 کاملاً ناعادلانه - 7 کاملاً عادلانه)", 1, 7)
+    transparency = st.slider("چقدر توضیحات ارائه شده در مورد قیمت را شفاف می‌دانید؟ (1 کاملاً نامشخص - 7 کاملاً شفاف)", 1, 7)
+    satisfaction = st.slider("چقدر از این قیمت‌گذاری راضی هستید؟ (1 کاملاً ناراضی - 7 کاملاً راضی)", 1, 7)
     
-    if demographic_data:
-        fairness_data = fairness_questionnaire()
+    return {
+        "fairness": fairness,
+        "transparency": transparency,
+        "satisfaction": satisfaction
+    }
+
+# ========== رابط کاربری اصلی ==========
+def main():
+    # مدیریت وضعیت
+    if 'scenario_type' not in st.session_state:
+        st.session_state.scenario_type = random.choice(["control", "input", "counterfactual"])
+    if 'price' not in st.session_state:
+        st.session_state.price = generate_random_price()
+    if 'demographic_data' not in st.session_state:
+        st.session_state.demographic_data = None
+    if 'survey_completed' not in st.session_state:
+        st.session_state.survey_completed = False
+
+    # نمایش هدر
+    st.markdown("""
+    <div class="ride-header">
+        <h2>تحلیل انصاف قیمتی در پلتفرم‌های اشتراک سفر</h2>
+        <p>این تحقیق به بررسی تأثیر توضیحات قیمت بر ادراک انصاف می‌پردازد</p>
+    </div>
+    """, unsafe_allow_html=True)
+
+    # اگر اطلاعات دموگرافیک تکمیل نشده
+    if not st.session_state.demographic_data:
+        data = demographic_form()
+        if data:
+            st.session_state.demographic_data = data
+            st.rerun()
+        return
+
+    # اگر پرسشنامه تکمیل نشده
+    if not st.session_state.survey_completed:
+        # نمایش سناریو
+        st.markdown("### 🗺️ مسیر سفر شما")
+        folium_static(create_ride_map(), width=800, height=400)
         
-        if st.button("ارسال پاسخ‌ها"):
-            all_data = {
-                "scenario": st.session_state.current_scenario,
-                **demographic_data,
-                **fairness_data
-            }
+        # نمایش قیمت
+        st.markdown(f"""
+        <div class="price-box">
+            قیمت پیشنهادی: {st.session_state.price:,} تومان
+        </div>
+        """, unsafe_allow_html=True)
+        
+        # نمایش توضیحات (بر اساس نوع سناریو)
+        show_explanation(st.session_state.scenario_type)
+        
+        # پرسشنامه
+        with st.container():
+            st.markdown("<div class='survey-section'>", unsafe_allow_html=True)
+            survey_data = fairness_questionnaire()
+            st.markdown("</div>", unsafe_allow_html=True)
             
-            if save_to_sheet(all_data):
-                st.success("پاسخ‌های شما با موفقیت ثبت شد. با تشکر از مشارکت شما!")
-                st.session_state.clear()
-                st.rerun()
+            if st.button("ارسال پاسخ‌ها", type="primary"):
+                all_data = {
+                    "scenario_type": st.session_state.scenario_type,
+                    "price": st.session_state.price,
+                    **st.session_state.demographic_data,
+                    **survey_data
+                }
+                
+                if save_to_sheet(all_data):
+                    st.session_state.survey_completed = True
+                    st.rerun()
+    else:
+        # صفحه تشکر
+        st.success("✅ پاسخ‌های شما با موفقیت ثبت شد. با تشکر از مشارکت شما در این تحقیق!")
+        st.balloons()
+        
+        if st.button("بازگشت به ابتدا"):
+            st.session_state.clear()
+            st.rerun()
 
 if __name__ == "__main__":
     main()