better UI

app.py

@@ -10,6 +10,7 @@ import random
 
 @st.dialog("Select From Library")
 def lib_modal():
+    # if st.button("Confirm", key="confirm"):
     files = os.listdir("lib")
     files = random.choices(files, k=50)
     cols = st.columns(2)
@@ -18,9 +19,15 @@ def lib_modal():
             st.image("lib/{}".format(file), use_container_width=True)
             if st.button("Select", key=i):
                 st.session_state.uploaded_file = file
+                st.session_state.show_modal = False
                 st.rerun()
             st.markdown("---")
 
+
+
+if "show_modal" not in st.session_state:
+    st.session_state.show_modal = True
+
 st.title("3D2SMILES")
 st.markdown("""
 This app generates SMILES strings from images of molecules ball-and-stick models. 
@@ -45,9 +52,12 @@ if mode == "Upload File":
     col1.checkbox("Contribute To Public Library", value=True, help="If checked, images will be included in the PUBLIC library, and the image will be reviewed by our team and used for model training. When checked, do not upload any sensitive or personal data.")
 else:
     uploaded_file = None
+    st.warning("You are accessing a user-uploaded library. These images are unverified, unmoderated, and may include inappropriate content or incorrectly assembled files. Proceed at your own risk. For inquiries, contact us.")
     from_library = st.button("Select from Library")
     if from_library:
-        lib_modal()
+        if st.session_state.show_modal:
+            lib_modal()
+
     if "uploaded_file" in st.session_state:
         st.markdown("You have selected: {}".format(st.session_state.uploaded_file))