Update phase/interpret.py

modified entire interface; defined state rules for every component

phase/interpret.py

@@ -1,25 +1,46 @@
-def assign_typology(sorted_components, high_th=0.4, th=0.2):
+def assign_typology(components, high_th=0.4, th=0.2):
+
+    # define state rules for every components
+    STATE_RULES = {
+        "PCR (Participation Coverage Ratio)": [
+            (0.5, "Low Alert"),
+            (0.2, "Medium Alert"),
+            (0.0, "High Alert")
+        ],
+        "SPI-A (Silent Participation Inference - Silent Non-Adoption)": [
+            (0.7, "High Alert"),
+            (0.3, "Medium Alert"),
+            (0.0, "Low Alert")
+        ],
+        "SPI-B (Silent Participation Inference - Silent Adoption)": [
+            (0.5, "High Alert"),
+            (0.2, "Medium Alert"),
+            (0.0, "Low Alert")
+        ],
+        "PEG (Participation-to-Expression Gap)": [
+            (0.6, "High Alert"),
+            (0.3, "Medium Alert"),
+            (0.0, "Low Alert")
+        ]
+    }
     
-    high_impact = []
-    medium_impact = []
-    low_impact = []
+    buckets = {
+        "High Alert": [],
+        "Medium Alert": [],
+        "Low Alert": []
+    }
 
-    for component, data in sorted_components.items():
-        entry = {
-            "component": component,
-            "value": data["value"],
-            "description": data.get("description", "")
-        }
+    for component, data in components.items():
+        value = data["Value"]
+        description = data.get("Description", "")
 
-        if abs(data["value"]) >= high_th:
-            high_impact.append(entry)
-        elif abs(data["value"]) >= th:
-            medium_impact.append(entry)
-        else:
-            low_impact.append(entry)
+        for threshold, state in STATE_RULES[component]:
+            if value >= threshold:
+                buckets[state].append({
+                    "Component": component,
+                    "Value": value,
+                    "Description": description
+                })
+                break
 
-    return {
-        "high": high_impact,
-        "medium": medium_impact,
-        "low": low_impact
-    }
+    return buckets