Update mtdna_classifier.py

mtdna_classifier.py

@@ -43,7 +43,6 @@ def get_info_from_accession(accession):
 
     # Return the values, even if they are empty strings
     return pubmedID, isolate
-print(get_info_from_accession("KU131308"))
 # Step 2: Get doi link to access the paper
 def get_doi_from_pubmed_id(pubmed_id):
     cmd = f'{os.environ["HOME"]}/edirect/esummary -db pubmed -id {pubmed_id} -format medline | grep -i "AID"'
@@ -208,6 +207,7 @@ def classify_sample_location(accession):
   keyword, context, location, qa_result, haplo_result = "", "", "", "", ""
   # Step 1: get pubmed id and isolate
   pubmedID, isolate = get_info_from_accession(accession)
+  print(pubmedID, isolate)  
   if not pubmedID:
     return {"error": f"Could not retrieve PubMed ID for accession {accession}"}
   if not isolate: