Maintain save_image state across modality switches

app.py

@@ -26,6 +26,9 @@ if "grascii" not in st.session_state:
 if "alternatives" not in st.session_state:
     st.session_state["alternatives"] = {}
 
+if "save_image" in st.session_state:
+    st.session_state["save_image"] = st.session_state["save_image"]
+
 if st.session_state["report_submitted"]:
     st.toast("Thanks for the report!")
     st.session_state["report_submitted"] = False

search.py

@@ -70,7 +70,8 @@ def write_grascii_search():
                         """,
                 )
                 save = st.checkbox(
-                    "Save images I upload for potential inclusion in open-source datasets used to train and improve models"
+                    "Save images I upload for potential inclusion in open-source datasets used to train and improve models",
+                    key="save_image",
                 )
 
             if image_data: