Update app.py

app.py

@@ -14,39 +14,9 @@ import urllib.parse
 
 urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
 
-# Set page config and apply dark theme
 st.set_page_config(layout='wide')
-st.markdown("""
-    <style>
-    body {
-        color: #fff;
-        background-color: #0e1117;
-    }
-    .stApp {
-        background-color: #0e1117;
-    }
-    .stTextInput > div > div > input {
-        color: #fff;
-        background-color: #262730;
-    }
-    .stNumberInput > div > div > input {
-        color: #fff;
-        background-color: #262730;
-    }
-    .stTextArea > div > div > textarea {
-        color: #fff;
-        background-color: #262730;
-    }
-    .stButton > button {
-        color: #fff;
-        background-color: #0e1117;
-        border: 1px solid #fff;
-    }
-    </style>
-    """, unsafe_allow_html=True)
-
-st.title('🔮 GenPro2 Protein Generator, Structure Predictor, and Analysis Tool')
-st.write('GenPro2 is an end-to-end protein sequence generator, structure predictor, and analysis tool based [*ESMFold*](https://esmatlas.com/about) and the ESM-2 language model.')
+st.sidebar.title('🔮 GenPro2 Protein Generator, Structure Predictor, and Analysis Tool')
+st.sidebar.write('GenPro2 is an end-to-end protein sequence generator, structure predictor, and analysis tool based [*ESMFold*](https://esmatlas.com/about) and the ESM-2 language model.')
 
 def generate_sequence_from_words(words, length):
     seed = ' '.join(words).encode('utf-8')
@@ -94,10 +64,11 @@ def perform_blast_analysis(sequence):
                 st.write(f"**Organism Code:** {organism}")
                 st.write(f"**Sequence Identity:** {identity_percentage:.2f}%")
                
+                
                 # Fetch protein function (if available)
                 if hasattr(alignment, 'description') and alignment.description:
                     st.write(f"**Potential Function:** {alignment.description}")
-            else:
+                
                 st.write("No significant matches found. This might be a unique protein sequence!")
         except Exception as e:
             st.error(f"An error occurred during protein analysis: {str(e)}")
@@ -150,29 +121,25 @@ if 'show_analyze_button' not in st.session_state:
 if 'structure_info' not in st.session_state:
     st.session_state.structure_info = None
 
-# Main layout
-st.subheader("Generate Sequence from Words")
-col1, col2, col3 = st.columns(3)
-with col1:
-    word1 = st.text_input("Word 1")
-with col2:
-    word2 = st.text_input("Word 2")
-with col3:
-    word3 = st.text_input("Word 3")
+st.title("Word-Seeded Protein Sequence Generator, Structure Predictor, and Analysis Tool")
 
-sequence_length = st.number_input("Sequence Length", min_value=50, max_value=400, value=100, step=10)
+st.sidebar.subheader("Generate Sequence from Words")
+word1 = st.sidebar.text_input("Word 1")
+word2 = st.sidebar.text_input("Word 2")
+word3 = st.sidebar.text_input("Word 3")
+sequence_length = st.sidebar.number_input("Sequence Length", min_value=50, max_value=400, value=100, step=10)
 
-if st.button('Generate and Predict'):
+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.text_area("Generated Sequence", sequence, height=100)
-        st.info("Note: The same words and sequence length will always produce the same 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.warning("Please enter all three words to generate a sequence.")
+        st.sidebar.warning("Please enter all three words to generate a sequence.")
 
 # Display structure information if available
 if st.session_state.structure_info:
@@ -181,14 +148,14 @@ if st.session_state.structure_info:
     render_mol(info['pdb_string'])
     
     st.subheader('plDDT Confidence Score')
-    st.write('plDDT is a benchmark for scoring the confidence level in protein folding predictions based on a scale from 0-100%. 70% or more is good!')
+    st.write('plDDT is a bench mark for scoring the confidence level in protein folding predictions based on a scale from 0-100%. 70% or more is good!')
     plddt_score = int(info["b_value"] * 100)
     st.info(f'Your plDDT score is: {plddt_score}%')
  
     st.subheader("Share your unique protein on X(Twitter)")
     
     st.markdown("""
-    <div style='background-color: #262730; padding: 10px; border-radius: 5px; font-size: 0.8em;'>
+    <div style='background-color: #e6f2ff; padding: 10px; border-radius: 5px; font-size: 0.8em;'>
     <ol>
         <li>Take a screenshot of the protein structure above.</li>
         <li>Click the 'Share on X' button below to open a pre-filled post with your protein seed-words and score.</li>