Update app.py

app.py

@@ -94,6 +94,18 @@ st.markdown("""
         display: inline-block;
         margin-left: 10px;
     }
+    
+    .explanation-title {
+        color: #6a0dad;
+        font-weight: bold;
+        margin-top: 20px;
+    }
+    
+    .explanation-item {
+        margin: 10px 0;
+        padding-right: 15px;
+        border-right: 3px solid #6a0dad;
+    }
 </style>
 """, unsafe_allow_html=True)
 
@@ -145,9 +157,9 @@ def show_explanation(exp_type):
     }
     
     if exp_type != "control":
-        st.markdown("<p style='color: #6a0dad; font-weight: bold; margin-top: 20px;'>علت قیمت گذاری:</p>", unsafe_allow_html=True)
+        st.markdown("<p class='explanation-title'>علت قیمت گذاری:</p>", unsafe_allow_html=True)
         for item in explanations.get(exp_type, []):
-            st.markdown(f"<p style='margin: 10px 0; padding-right: 15px; border-right: 3px solid #6a0dad;'>• {item}</p>", unsafe_allow_html=True)
+            st.markdown(f"<p class='explanation-item'>• {item}</p>", unsafe_allow_html=True)
 
 # ========== توابع مدیریت داده‌ها ==========
 def get_credentials():
@@ -402,23 +414,27 @@ def thank_you_page():
         st.rerun()
 
 # ========== مدیریت وضعیت و صفحه‌بندی ==========
-if 'current_page' not in st.session_state:
-    st.session_state.current_page = "welcome"
-    st.session_state.scenario_type = random.choice(["control", "input", "counterfactual"])
-    st.session_state.price = 200000  # قیمت ثابت 200 هزار تومان
+def main():
+    if 'current_page' not in st.session_state:
+        st.session_state.current_page = "welcome"
+        st.session_state.scenario_type = random.choice(["control", "input", "counterfactual"])
+        st.session_state.price = 200000  # قیمت ثابت 200 هزار تومان
+        st.session_state.demographic_data = None
+        st.session_state.transparency_answers = []
+        st.session_state.fairness_answers = []
 
-# نمایش صفحه فعلی
-pages = {
-    "welcome": welcome_page,
-    "demographic": demographic_form,
-    "scenario_explanation": scenario_explanation,
-    "map_view": map_view,
-    "transparency_questions": transparency_questions,
-    "fairness_questions": fairness_questions,
-    "thank_you": thank_you_page
-}
+    # نمایش صفحه فعلی
+    pages = {
+        "welcome": welcome_page,
+        "demographic": demographic_form,
+        "scenario_explanation": scenario_explanation,
+        "map_view": map_view,
+        "transparency_questions": transparency_questions,
+        "fairness_questions": fairness_questions,
+        "thank_you": thank_you_page
+    }
 
-pages[st.session_state.current_page]()
+    pages[st.session_state.current_page]()
 
 if __name__ == "__main__":
     main()