Update app.py

app.py

@@ -1,3 +1,4 @@
+import streamlit as st
 import requests
 
 def get_genome_from_ncbi(accession_number):
@@ -8,7 +9,7 @@ def get_genome_from_ncbi(accession_number):
     if response.status_code == 200:
         return response.text
     else:
-        print(f"Failed to retrieve genome for {accession_number}. Please check the accession number.")
+        st.warning(f"Failed to retrieve genome for {accession_number}. Please check the accession number.")
         return None
 
 def extract_sequences(genome_data):
@@ -23,14 +24,14 @@ def extract_sequences(genome_data):
     return start_sequence, end_sequence
 
 def main():
-    # Prompt user for the input file name
-    input_file = input("Enter the name of the input text file (with extension): ")
-    output_start_file = "starting_sequences.txt"  # Output file for starting 55 base pairs
-    output_end_file = "ending_sequences.txt"  # Output file for ending 114 base pairs
+    st.title("NCBI Genome Sequence Extractor")
 
-    try:
-        with open(input_file, 'r') as file:
-            accession_numbers = file.read().splitlines()  # Read lines into a list
+    # File uploader for the user to upload the input file
+    uploaded_file = st.file_uploader("Upload a text file containing accession numbers", type="txt")
+    
+    if uploaded_file is not None:
+        # Read and split accession numbers from the uploaded file
+        accession_numbers = uploaded_file.read().decode("utf-8").splitlines()
 
         # Prepare lists to store sequences
         starting_sequences = []
@@ -43,20 +44,13 @@ def main():
                 starting_sequences.append(f"{accession_number}: {start_sequence}")
                 ending_sequences.append(f"{accession_number}: {end_sequence}")
 
-        # Write starting sequences to file
-        with open(output_start_file, 'w') as file:
-            for seq in starting_sequences:
-                file.write(seq + '\n')
-
-        # Write ending sequences to file
-        with open(output_end_file, 'w') as file:
-            for seq in ending_sequences:
-                file.write(seq + '\n')
-
-        print(f"Sequences extracted and saved to {output_start_file} and {output_end_file}.")
+        # Write starting sequences to a file for download
+        start_sequences_text = "\n".join(starting_sequences)
+        st.download_button("Download Starting Sequences", start_sequences_text, file_name="starting_sequences.txt", mime="text/plain")
 
-    except FileNotFoundError:
-        print("Error: The specified file was not found. Please check the file name and try again.")
+        # Write ending sequences to a file for download
+        end_sequences_text = "\n".join(ending_sequences)
+        st.download_button("Download Ending Sequences", end_sequences_text, file_name="ending_sequences.txt", mime="text/plain")
 
 if __name__ == "__main__":
     main()