Update app.py

app.py

@@ -67,6 +67,10 @@ def perform_blast_analysis(sequence):
         except Exception as e:
             st.error("An error occurred during protein analysis. Please try again later.")
 
+import streamlit as st
+
+# ... (keep all previous imports and functions)
+
 def update(sequence, word1, word2, word3, sequence_length):
     headers = {
         'Content-Type': 'application/x-www-form-urlencoded',
@@ -113,13 +117,18 @@ def update(sequence, word1, word2, word3, sequence_length):
         Enjoy exploring the world of protein sequences! Share your high-confidence protein images with us on X [*@WandsAI*](https://x.com/wandsai)!
         """)
 
-        if st.button('Analyze Protein'):
-            perform_blast_analysis(sequence)
+        st.session_state.show_analyze_button = True
 
     except requests.exceptions.RequestException as e:
         st.error(f"An error occurred while calling the API: {str(e)}")
         st.write("Please try again later or contact support if the issue persists.")
 
+# Initialize session state variables
+if 'sequence' not in st.session_state:
+    st.session_state.sequence = None
+if 'show_analyze_button' not in st.session_state:
+    st.session_state.show_analyze_button = False
+
 st.title("Word-Seeded Protein Sequence Generator and Structure Predictor")
 
 st.sidebar.subheader("Generate Sequence from Words")
@@ -131,10 +140,16 @@ sequence_length = st.sidebar.number_input("Sequence Length", min_value=50, max_v
 if st.sidebar.button('Generate and Predict'):
     if word1 and word2 and word3:
         sequence = generate_sequence_from_words([word1, word2, word3], sequence_length)
+        st.session_state.sequence = sequence
         st.sidebar.text_area("Generated Sequence", sequence, height=100)
         st.sidebar.info("Note: The same words and sequence length will always produce the same sequence.")
         
         with st.spinner("Predicting protein structure... This may take a few minutes."):
             update(sequence, word1, word2, word3, sequence_length)
     else:
-        st.sidebar.warning("Please enter all three words to generate a sequence.")
+        st.sidebar.warning("Please enter all three words to generate a sequence.")
+
+# Show the Analyze Protein button if a sequence has been generated
+if st.session_state.show_analyze_button:
+    if st.button('Analyze Protein'):
+        perform_blast_analysis(st.session_state.sequence)