from library

app.py

@@ -5,6 +5,21 @@ import pubchempy as pcp
 from rdkit import Chem
 from rdkit.Chem import Draw
 import time
+import os
+import random
+
+@st.dialog("Select From Library")
+def lib_modal():
+    files = os.listdir("lib")
+    files = random.choices(files, k=50)
+    cols = st.columns(2)
+    for i, file in enumerate(files):
+        with cols[i % 2]:
+            st.image("lib/{}".format(file), use_container_width=True)
+            if st.button("Select", key=i):
+                st.session_state.uploaded_file = file
+                st.rerun()
+            st.markdown("---")
 
 st.title("3D2SMILES")
 st.markdown("""
@@ -26,9 +41,16 @@ uploaded_file = st.file_uploader("Upload an image", type=["png", "jpg", "jpeg",
 lib = st.columns(2)
 lib[0].markdown("Do not have an image? You can select a molecule from the library below.")
 from_library = lib[1].button("Select from Library")
+if from_library:
+    lib_modal()
 
 col1, col2 = st.columns(2)
-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.")
+if not "uploaded_file" in st.session_state:
+    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:
+    col1.checkbox("Contribute To Public Library", value=False, disabled=True, help="When using the library, this option is disabled. If you want to contribute to the library, please upload an image.")
+
+
 button = col2.button("Submit")
 if button:
     if uploaded_file:
@@ -42,7 +64,7 @@ if button:
         for i, (smiles, col) in enumerate(zip(options, cols)):
             cid = pcp.get_compounds(smiles, 'smiles')
             name = cid[0].synonyms[0]
-            col.markdown("## Option {}: {}".format(i + 1, name))
+            col.markdown(name)
             m = Chem.MolFromSmiles(smiles)
             img = Draw.MolToImage(m)
             col.image(img, use_container_width=False)

lib.py

@@ -0,0 +1,6 @@
+import streamlit as st
+import os
+import random
+
+
+