Update my_pages/rashomon_effect.py

my_pages/rashomon_effect.py

@@ -6,6 +6,17 @@ from utils import go_to
 plt.style.use('dark_background')
 
 def render():
+    st.markdown(
+        """
+        <style>
+        button[kind="primary"] {
+            background: green!important;
+        }
+        </style>
+        """,
+        unsafe_allow_html=True,
+    )
+    
     st.markdown(
         """
         <div style='text-align: center; font-size:18px; color:gray;'>
@@ -66,8 +77,9 @@ def render():
         return fig
 
 
-    highlight_point = None
-    if "highlight_point" in st.session_state:
+    graph_selected, highlight_point = None, None
+    if "graph_selected" in st.session_state:
+        graph_selected = st.session_state.graph_selected
         highlight_point = st.session_state.highlight_point
 
     # Top scatter plot (centered to match smaller width)
@@ -78,18 +90,33 @@ def render():
     col1, col2, col3, col4, col5 = st.columns([0.5, 1, 1, 1, 0.5])
     with col2:
         st.pyplot(plot_scatter(income, credit, colors, boundary_type="vertical", highlight_point=highlight_point))
-        if st.button("Choose Model 1"):
+        if graph_selected=="vertical":
+            button_click_v = st.button("Choose Model 1", type="primary")
+        else:
+            button_click_v = st.button("Choose Model 1")
+        if button_click_v:
             st.session_state.highlight_point = (32, 902)
+            st.session_state.graph_selected = "vertical"
             st.rerun()
     with col3:
         st.pyplot(plot_scatter(income, credit, colors, boundary_type="slant", highlight_point=highlight_point))
-        if st.button("Choose Model 2"):
-            st.session_state.highlight_point = (97, 370)
+        if graph_selected=="slant":
+            button_click_s = st.button("Choose Model 2", type="primary")
+        else:
+            button_click_s = st.button("Choose Model 2")
+        if button_click_s:
+            st.session_state.highlight_point = (32, 902)
+            st.session_state.graph_selected = "slant"
             st.rerun()
     with col4:
         st.pyplot(plot_scatter(income, credit, colors, boundary_type="horizontal", highlight_point=highlight_point))
-        if st.button("Choose Model 3"):
-            st.session_state.highlight_point = (32, 902)
+        if graph_selected=="horizontal":
+            button_click_h = st.button("Choose Model 3", type="primary")
+        else:
+            button_click_h = st.button("Choose Model 3")
+        if button_click_h:
+            st.session_state.highlight_point = (97, 370)
+            st.session_state.graph_selected = "horizontal"
             st.rerun()
 
     st.markdown("---")