VyLala commited on
Commit
fba43f1
·
verified ·
1 Parent(s): df3e4ea

Update mtdna_backend.py

Browse files
Files changed (1) hide show
  1. mtdna_backend.py +3 -0
mtdna_backend.py CHANGED
@@ -25,6 +25,7 @@ import threading
25
  #@lru_cache(maxsize=3600)
26
  def pipeline_classify_sample_location_cached(accession,stop_flag=None, save_df=None):
27
  print("inside pipeline_classify_sample_location_cached, and [accession] is ", [accession])
 
28
  return pipeline.pipeline_with_gemini([accession],stop_flag=stop_flag, save_df=save_df)
29
 
30
  # Count and suggest final location
@@ -170,6 +171,7 @@ def summarize_results(accession, stop_flag=None):
170
  return None
171
 
172
  save_df = pd.DataFrame(data[1:], columns=data[0])
 
173
  outputs = pipeline_classify_sample_location_cached(accession, stop_flag, save_df)
174
  if stop_flag is not None and stop_flag.value:
175
  print(f"🛑 Skipped {accession} mid-pipeline.")
@@ -538,6 +540,7 @@ def save_batch_output(all_rows, output_type, summary_text=None, flag_text=None):
538
  # print(f"⚠️ Failed to load known samples from Google Sheets: {e}")
539
  # return None
540
  def check_known_output(accession):
 
541
  try:
542
  # ✅ Load credentials from Hugging Face secret
543
  creds_dict = json.loads(os.environ["GCP_CREDS_JSON"])
 
25
  #@lru_cache(maxsize=3600)
26
  def pipeline_classify_sample_location_cached(accession,stop_flag=None, save_df=None):
27
  print("inside pipeline_classify_sample_location_cached, and [accession] is ", [accession])
28
+ print("len of save df: ", len(save_df))
29
  return pipeline.pipeline_with_gemini([accession],stop_flag=stop_flag, save_df=save_df)
30
 
31
  # Count and suggest final location
 
171
  return None
172
 
173
  save_df = pd.DataFrame(data[1:], columns=data[0])
174
+ print("before pipeline, len of save df: ", len(save_df))
175
  outputs = pipeline_classify_sample_location_cached(accession, stop_flag, save_df)
176
  if stop_flag is not None and stop_flag.value:
177
  print(f"🛑 Skipped {accession} mid-pipeline.")
 
540
  # print(f"⚠️ Failed to load known samples from Google Sheets: {e}")
541
  # return None
542
  def check_known_output(accession):
543
+ print("inside check known output function")
544
  try:
545
  # ✅ Load credentials from Hugging Face secret
546
  creds_dict = json.loads(os.environ["GCP_CREDS_JSON"])