Update ajcc_api.py

ajcc_api.py

@@ -54,31 +54,40 @@ def stage_cancer(cancer_type: str, T: str, N: str, M: str) -> str:
     N = normalize_tnm(N)
     M = normalize_tnm(M)
 
-    match cancer_type:
-        case "breast":
-            return stage_breast(T, N, M)
-        case "lung" | "nsclc" | "non-small cell lung" | "non small cell lung":
-            return stage_nsclc(T, N, M)
-        case "colon" | "rectal" | "colorectal":
-            return stage_colorectal(T, N, M)
-        case "stomach" | "gastric":
-            return stage_stomach(T, N, M)
-        case "pancreas" | "pancreatic":
-            return stage_pancreas(T, N, M)
-        case "liver" | "hcc" | "hepatocellular":
-            return stage_liver(T, N, M)
-        case "melanoma":
-            return stage_melanoma(T, N, M)
-        case "prostate":
-            return stage_prostate(T, N, M)
-        case "kidney" | "renal":
-            return stage_kidney(T, N, M)
-        case "bladder":
-            return stage_bladder(T, N, M)
-        case _:
-            # General fallback AJCC rule
-            return stage_general(T, N, M)
+    if cancer_type == "breast":
+        return stage_breast(T, N, M)
 
+    elif cancer_type in ("lung", "nsclc", "non-small cell lung", "non small cell lung"):
+        return stage_nsclc(T, N, M)
+
+    elif cancer_type in ("colon", "rectal", "colorectal"):
+        return stage_colorectal(T, N, M)
+
+    elif cancer_type in ("stomach", "gastric"):
+        return stage_stomach(T, N, M)
+
+    elif cancer_type in ("pancreas", "pancreatic"):
+        return stage_pancreas(T, N, M)
+
+    elif cancer_type in ("liver", "hcc", "hepatocellular"):
+        return stage_liver(T, N, M)
+
+    elif cancer_type == "melanoma":
+        return stage_melanoma(T, N, M)
+
+    elif cancer_type == "prostate":
+        return stage_prostate(T, N, M)
+
+    elif cancer_type in ("kidney", "renal"):
+        return stage_kidney(T, N, M)
+
+    elif cancer_type == "bladder":
+        return stage_bladder(T, N, M)
+
+    else:
+        return stage_general(T, N, M)
+
+    
 
 # ---------------------------------------------------------------------
 # General fallback AJCC rule (for unknown cancer types)