small bugs

app.py

@@ -45,13 +45,13 @@ def about_page():
         """
     )
 
-    st.image('hyper-dti.png')
+    st.image('hyper-dti.png', caption='Overview of HyperPCM architecture.')
 
 
 def predict_dti():
     st.markdown('## Predict drug-target interaction')
 
-    st.write('In the future this page will display the predicted interaction betweek the given drug compounds and protein target by the HyperPCM mdoel.')
+    st.warning('In the future this page will display the predicted interaction betweek the given drug compounds and protein target by the HyperPCM mdoel.')
     
     col1, col2 = st.columns(2)
     
@@ -108,7 +108,7 @@ def predict_dti():
         prot_col1, prot_col2 = st.columns(2)
 
         with prot_col1:
-            sequence = st.text_input('Enter the amino-acid sequence of the query protein target', value='HXHVWPVQDAKARFSEFLDACITEGPQIVSRRGAEEAVLVPIGEWRRLQAAA', placeholder='HXHVWPVQDAKARFSEFLDACITEGPQIVSRRGAEEAVLVPIGEWRRLQAAA')
+            sequence = st.text_input('Enter query amino-acid sequence', value='HXHVWPVQDAKARFSEFLDACITEGPQIVSRRGAEEAVLVPIGEWRRLQAAA', placeholder='HXHVWPVQDAKARFSEFLDACITEGPQIVSRRGAEEAVLVPIGEWRRLQAAA')
         
             if sequence:
                 st.error('Visualization comming soon...')
@@ -117,7 +117,7 @@ def predict_dti():
             selected_encoder = st.selectbox(
                 'Select encoder for protein target',('None', 'SeqVec', 'UniRep', 'ESM-1b', 'ProtT5')
             )
-            st.image('protein_encoder.png')
+        
             if sequence: 
                 if selected_encoder == 'SeqVec':
                     with st.spinner('Encoding in progress...'):
@@ -152,14 +152,15 @@ def predict_dti():
                             break
                 else: 
                     prot_embedding = None
-                    st.warning('Chosen encoder above...')
+                    st.image('protein_encoder.png')
+                    st.warning('Choose encoder above...')
                     
                 if prot_embedding is not None:
                     st.image('protein_encoder_done.png')
                     st.success('Encoding complete.')
 
-    if not drug_embedding or not prot_embedding:
-        st.error('Waiting for both drug and target embeddings to be computed...')
+    if drug_embedding is None or prot_embedding is None:
+        st.warning('Waiting for both drug and target embeddings to be computed...')
     else:
         st.warning('In the future inference will be run with HyperPCM on the given drug compound and protein target...')
     
@@ -170,7 +171,7 @@ def retrieval():
     st.write('In the furute this page will retrieve the top-k drug compounds that are predicted to have the highest activity toward the given protein target from either the Lenselink or Davis datasets.')
 
     st.markdown('### Target')
-    sequence = st.text_input('Enter the amino-acid sequence of the query protein target', value='HXHVWPVQDAKARFSEFLDACITEGPQIVSRRGAEEAVLVPIGEWRRLQAAA', placeholder='HXHVWPVQDAKARFSEFLDACITEGPQIVSRRGAEEAVLVPIGEWRRLQAAA')
+    sequence = st.text_input('Enter query amino-acid sequence', value='HXHVWPVQDAKARFSEFLDACITEGPQIVSRRGAEEAVLVPIGEWRRLQAAA', placeholder='HXHVWPVQDAKARFSEFLDACITEGPQIVSRRGAEEAVLVPIGEWRRLQAAA')
     
     if sequence:
         col1, col2 = st.columns(2)
@@ -178,8 +179,6 @@ def retrieval():
             st.error('Visualization coming soon...')
         
         with col2:
-            st.image('protein_encoder.png')
-
             with st.spinner('Encoding in progress...'):
                 from bio_embeddings.embed import SeqVecEmbedder
                 encoder = SeqVecEmbedder()