Update app.py

app.py

@@ -3,6 +3,7 @@ import pandas as pd
 import primer3
 from Bio import SeqIO
 import os
+from io import StringIO
 
 # Ensure the 'temp' directory exists for saving temporary files
 temp_dir = "temp"
@@ -13,8 +14,13 @@ st.set_page_config(page_title="PCR Primer Design", page_icon="🧬", layout="wid
 uploaded_file = st.file_uploader("Upload a GenBank file", type=['gb', 'gbk'])
 
 def extract_features_from_genbank(genbank_content, feature_types=['CDS', 'tRNA', 'gene']):
-    """Extracts specified features from a GenBank content."""
-    record = SeqIO.read(genbank_content, "genbank")
+    """Extracts specified features from GenBank content."""
+    # Convert binary stream (if present) to a StringIO object for text handling
+    if isinstance(genbank_content, bytes):
+        text_stream = StringIO(genbank_content.decode("utf-8"))
+    else:
+        text_stream = genbank_content
+    record = SeqIO.read(text_stream, "genbank")
     features = {ftype: [] for ftype in feature_types}
     for feature in record.features:
         if feature.type in feature_types:
@@ -43,7 +49,8 @@ def design_primers_for_region(sequence, start, end, num_to_return=5):
     )
 
 if uploaded_file is not None:
-    features, record = extract_features_from_genbank(uploaded_file)
+    genbank_content = StringIO(uploaded_file.getvalue().decode("utf-8"))
+    features, record = extract_features_from_genbank(genbank_content)
     feature_type = st.selectbox('Select feature type:', ['CDS', 'tRNA', 'gene'])
 
     # Create a selection box for genes based on chosen feature