Update app.py

app.py

@@ -1,34 +1,31 @@
 import streamlit as st
 import requests
 
-# Install required packages
-import os
-os.system("pip install requests streamlit")
-
-def get_genome_from_uniprot(uniprot_id):
-    url = f"https://www.uniprot.org/uniprot/{uniprot_id}.fasta"
+def get_genome_from_ncbi(accession_number):
+    # Construct the URL for NCBI GenBank
+    url = f"https://www.ncbi.nlm.nih.gov/sviewer/viewer.cgi?db=nuccore&id={accession_number}&report=fasta&retmode=text"
     response = requests.get(url)
     
     if response.status_code == 200:
         return response.text
     else:
-        st.error("Failed to retrieve genome. Please check the UniProt ID.")
+        st.error("Failed to retrieve genome. Please check the accession number.")
         return None
 
 def main():
-    st.title("Download Genome from UniProtKB")
-    st.write("Enter the unique UniProt ID to download the corresponding genome.")
+    st.title("Download Genome from NCBI")
+    st.write("Enter the accession number to download the corresponding genome.")
 
-    uniprot_id = st.text_input("UniProt ID")
+    accession_number = st.text_input("Accession Number")
 
     if st.button("Get Genome"):
-        if uniprot_id:
-            genome_data = get_genome_from_uniprot(uniprot_id)
+        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"{uniprot_id}.fasta", mime="text/plain")
+                st.download_button("Download Genome", genome_data, file_name=f"{accession_number}.fasta", mime="text/plain")
         else:
-            st.warning("Please enter a UniProt ID.")
+            st.warning("Please enter an accession number.")
 
 if __name__ == "__main__":
     main()