Update app.py

app.py

@@ -36,43 +36,54 @@ def render_mol(pdb):
 
 def perform_blast_analysis(sequence):
     st.subheader('Protein Analysis')
-    with st.spinner("Analyzing generated protein... This may take a several minutes. Stay tuned!"):
+    with st.spinner("Analyzing generated protein... This may take several minutes. Stay tuned!"):
         progress_bar = st.progress(0)
+        
+        # Slow down the progress bar
         for i in range(100):
             progress_bar.progress(i + 1)
-            time.sleep(0.1)  # Simulate analysis time
+            time.sleep(0.3)  # Increased sleep time to 0.3 seconds
         
-        try:
-            record = SeqRecord(Seq(sequence), id='random_protein')
-            result_handle = NCBIWWW.qblast("blastp", "swissprot", record.seq)
-            
-            blast_record = NCBIXML.read(result_handle)
-            
-            if blast_record.alignments:
-                alignment = blast_record.alignments[0]  # Get the top hit
-                hsp = alignment.hsps[0]  # Get the first (best) HSP
+        max_retries = 3
+        for attempt in range(max_retries):
+            try:
+                Entrez.email = "your_email@example.com"  # Replace with your email
+                record = SeqRecord(Seq(sequence), id='random_protein')
+                result_handle = NCBIWWW.qblast("blastp", "swissprot", record.seq, expect=10, hitlist_size=1)
                 
-                # Extract protein name and organism
-                title_parts = alignment.title.split('|')
-                protein_name = title_parts[-1].strip()
-                organism = title_parts[-2].split('OS=')[-1].split('OX=')[0].strip()
+                blast_record = NCBIXML.read(result_handle)
                 
-                # Calculate identity percentage
-                identity_percentage = (hsp.identities / alignment.length) * 100
+                if blast_record.alignments:
+                    alignment = blast_record.alignments[0]  # Get the top hit
+                    hsp = alignment.hsps[0]  # Get the first (best) HSP
+                    
+                    # Extract protein name and organism
+                    title_parts = alignment.title.split('|')
+                    protein_name = title_parts[-1].strip()
+                    organism = title_parts[-2].split('OS=')[-1].split('OX=')[0].strip()
+                    
+                    # Calculate identity percentage
+                    identity_percentage = (hsp.identities / alignment.length) * 100
+                    
+                    st.write(f"**Top Match:** {protein_name}")
+                    st.write(f"**Organism:** {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!")
                 
-                st.write(f"**Top Match:** {protein_name}")
-                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}")
-                
-                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)}")
-            st.write("Please try again later or contact support if the issue persists.")
+                break  # If successful, exit the retry loop
+            
+            except Exception as e:
+                if attempt < max_retries - 1:
+                    st.warning(f"Attempt {attempt + 1} failed. Retrying...")
+                    time.sleep(random.uniform(1, 3))  # Add a random delay between retries
+                else:
+                    st.error(f"An error occurred during protein analysis after {max_retries} attempts: {str(e)}")
+                    st.write("Please try again later BLAST could be experiencing server issues.")
 
 def update(sequence, word1, word2, word3, sequence_length):
     headers = {