Update mtdna_backend.py

mtdna_backend.py

@@ -170,15 +170,29 @@ async def summarize_results(accession, stop_flag=None, niche_cases=None):
     cached = check_known_output(accession, niche_cases)
     if cached:
         print(f"✅ Using cached result for {accession}")
-        return [[
+        if niche_cases:
+            return [[
             cached["Sample ID"] or "unknown",
             cached["Predicted Country"] or "unknown",
             cached["Country Explanation"] or "unknown",
             cached["Predicted Sample Type"] or "unknown",
             cached["Sample Type Explanation"] or "unknown",
+            cached["Predicted "+niche_cases[0]] or "unknown",
+            cached[niche_cases[0] + " Explanation"] or "unknown",    
             cached["Sources"] or "No Links",
             cached["Time cost"]
         ]]
+
+        else:
+            return [[
+                cached["Sample ID"] or "unknown",
+                cached["Predicted Country"] or "unknown",
+                cached["Country Explanation"] or "unknown",
+                cached["Predicted Sample Type"] or "unknown",
+                cached["Sample Type Explanation"] or "unknown",
+                cached["Sources"] or "No Links",
+                cached["Time cost"]
+            ]]
     # only run when nothing in the cache  
     try:
         print("try gemini pipeline: ",accession)
@@ -645,7 +659,8 @@ def check_known_output(accession, niche_cases=None):
                 if niche_col not in _known_samples_cache.columns:
                     print(f"⚠️ Niche column '{niche_col}' not found in known_samples. Skipping cache.")
                     return None
-                output_niche = row.get("Predicted " + niche_cases[0], "").strip().lower()
+                output_niche = row.get(niche_col, "").strip().lower()
+                print("output niche: ", output_niche)
                 if country and country.lower() not in ["","unknown"] and sample_type and sample_type.lower() not in ["","unknown"] and output_niche and output_niche.lower() not in ["","unknown"]:
                     print(f"🎯 Found {accession} in cache")
                     return row.to_dict()