Update app.py

app.py

@@ -12,6 +12,17 @@ def get_genome_from_ncbi(accession_number):
         st.error("Failed to retrieve genome. Please check the accession number.")
         return None
 
+def extract_sequences(genome_data):
+    # Extract the sequence part from the FASTA format
+    lines = genome_data.splitlines()
+    sequence = ''.join(lines[1:])  # Join all lines except the first (which is the header)
+    
+    # Get the first 55 and last 114 base pairs
+    start_sequence = sequence[:55]  # First 55 base pairs
+    end_sequence = sequence[-114:]   # Last 114 base pairs
+
+    return start_sequence, end_sequence
+
 def main():
     st.title("Download Genome from NCBI")
     st.write("Enter the accession number to download the corresponding genome.")
@@ -22,8 +33,15 @@ def main():
         if accession_number:
             genome_data = get_genome_from_ncbi(accession_number)
             if genome_data:
-                st.text_area("Genome Data", genome_data, height=300)
-                st.download_button("Download Genome", genome_data, file_name=f"{accession_number}.fasta", mime="text/plain")
+                start_sequence, end_sequence = extract_sequences(genome_data)
+
+                # Display the sequences
+                st.text_area("Starting 55 Base Pairs", start_sequence, height=50)
+                st.text_area("Ending 114 Base Pairs", end_sequence, height=100)
+
+                # Prepare content for download
+                download_content = f"Starting 55 Base Pairs:\n{start_sequence}\n\nEnding 114 Base Pairs:\n{end_sequence}\n"
+                st.download_button("Download Sequences", download_content, file_name=f"{accession_number}_sequences.txt", mime="text/plain")
         else:
             st.warning("Please enter an accession number.")