Update app.py

app.py

@@ -803,40 +803,18 @@ def enhanced_likert_scale(question_data):
     
     for i in range(scale):
         with cols[i + 2]:  # شروع از ستون دوم (پس از برچسب‌ها)
-            button_html = f"""
-            <button class="likert-button" data-value="{options[i]}" 
-                    style="margin: 0 5px; width: 30px; height: 30px; border-radius: 50%; padding: 0; 
-                           text-align: center; line-height: 30px; background-color: #fff; 
-                           border: 2px solid #6a0dad; color: #6a0dad;"
-                    onclick="parent.window.parent.postMessage({{type: 'button_click', value: {options[i]}}}, '*')">
-                {options[i]}
-            </button>
-            <script>
-                window.addEventListener('message', function(event) {{
-                    if (event.data.type === 'button_click') {{
-                        parent.window.parent.postMessage({{type: 'update_session', key: '{key}', value: event.data.value}}, '*');
-                    }}
-                }});
-                // بررسی اولیه برای مقدار فعلی
-                if ({selected_value} === {int(options[i])}) {{
-                    document.querySelector('.likert-button[data-value="{options[i]}"]').disabled = false;
-                    document.querySelector('.likert-button[data-value="{options[i]}"]').style.backgroundColor = '#6a0dad';
-                    document.querySelector('.likert-button[data-value="{options[i]}"]').style.color = '#fff';
-                }}
-            </script>
-            """
-            st.components.v1.html(button_html, height=40)
+            button_key = f"{key}_btn_{i}"
+            is_selected = selected_value == int(options[i])
+            st.markdown(f'<div style="text-align: center;">', unsafe_allow_html=True)
+            if st.button(options[i], key=button_key, disabled=selected_value != 0 and not is_selected):
+                st.session_state[key] = int(options[i])
+            st.markdown(f'</div>', unsafe_allow_html=True)
 
     # نمایش برچسب سمت چپ
     with cols[1]:
         st.markdown(f"<div style='text-align:left; font-size:14px; font-weight:bold; color:#6a0dad; direction: rtl;'>{labels[1]}</div>", 
                     unsafe_allow_html=True)
 
-    # به‌روزرسانی session_state بر اساس پیام
-    if "button_click_value" in st.session_state and st.session_state["button_click_value"] is not None:
-        st.session_state[key] = st.session_state["button_click_value"]
-        del st.session_state["button_click_value"]
-
     # نمایش پاسخ انتخاب‌شده
     if st.session_state[key] != 0:
         st.markdown(f"<p style='text-align:right; color:#6a0dad; direction: rtl;'>پاسخ شما: {st.session_state[key]}</p>", 
@@ -856,7 +834,7 @@ def enhanced_likert_scale(question_data):
             direction: rtl;
         }
         /* استایل خاص برای دکمه‌های لیکرت */
-        .likert-button {
+        [data-testid="stButton"] {
             margin: 0 5px !important;
             width: 30px !important;
             height: 30px !important;
@@ -869,7 +847,7 @@ def enhanced_likert_scale(question_data):
             color: #6a0dad !important;
             cursor: pointer !important;
         }
-        .likert-button:disabled {
+        [data-testid="stButton"]:disabled {
             background-color: #6a0dad !important;
             color: #fff !important;
         }
@@ -878,7 +856,7 @@ def enhanced_likert_scale(question_data):
             .likert-separator {
                 width: 90%;
             }
-            [data-testid="stHorizontalBlock"] .likert-button {
+            [data-testid="stHorizontalBlock"] [data-testid="stButton"] {
                 margin: 0 2px !important;
             }
         }
@@ -895,6 +873,7 @@ def enhanced_likert_scale(question_data):
     """, unsafe_allow_html=True)
 
     return st.session_state.get(key)
+    
 def create_ride_map():
     """ایجاد نقشه سفر با Folium - نسخه اصلاح شده با مناطق عمومی"""
     # نقاط تقریبی برای مناطق عمومی جنوب و غرب تهران