update retrieval

app.py

@@ -151,15 +151,23 @@ def retrieval():
         st.markdown('\n\n\n\n Plot of protein to be added soon. \n\n\n\n')
         
         selected_encoder = st.selectbox(
-            'Select encoder for protein target',('SeqVec')
+            'Select encoder for protein target',('None', 'SeqVec')
         )
-        from bio_embeddings.embed import SeqVecEmbedder
-        encoder = SeqVecEmbedder()
-        embeddings = encoder.embed_batch([sequence])
-        for emb in embeddings:
-            embedding = encoder.reduce_per_protein(emb)
-            break
-
+        if selected_encoder == 'SeqVec':
+            from bio_embeddings.embed import SeqVecEmbedder
+            encoder = SeqVecEmbedder()
+            embeddings = encoder.embed_batch([sequence])
+            for emb in embeddings:
+                embedding = encoder.reduce_per_protein(emb)
+                break
+        else: 
+            st.write('Only SeqVec is currently available to encode protein structures.')
+            embedding = None
+        if embedding is not None:
+            st.write(f'{selected_encoder} embedding')
+            st.write(embedding)
+        
+    st.markdown('### Retrieval')
     st.write('TODO HyperPCM predicts the QSAR model for the given protein target.')
 
     col1, col2 = st.columns(2)