Update app.py

app.py

@@ -47,12 +47,9 @@ cloud_button_style = """
 st.markdown(cloud_button_style, unsafe_allow_html=True)
 
 def cloud_button(label, key):
-    button_html = f'<button type="button" class="cloud-button">{label}</button>'
-    button_click = st.button(label, key=key, use_container_width=True)
-    if button_click:
-        st.session_state[key] = not st.session_state.get(key, False)
+    button_html = f'<button onclick="javascript:window.parent.postMessage({{\'setSessionState\': {{\'{key}\': true}}}}, \'*\');" class="cloud-button">{label}</button>'
+    st.markdown(button_html, unsafe_allow_html=True)
     return st.session_state.get(key, False)
-
         
 # Step 1: Clone the Dataset Repository
 repo = Repository(