Upload data_generator.py

data_generator.py

@@ -87,4 +87,21 @@ def compute_ratios(s: FinancialSnapshot) -> Dict:
         "roe": roe,
         "debt_to_equity": dte,
         "profit_margin": pm,
-    }
+    }
+
+# ------------------------------------------------------------
+# Health classification (required by Medium grader)
+# ------------------------------------------------------------
+
+def classify_health(roe: float, dte: float, pm: float) -> str:
+    """
+    Classifies financial health based on ratio thresholds.
+    Matches logic expected by grader_medium.
+    """
+
+    if roe > 0.15 and dte < 1.0 and pm > 0.2:
+        return "Healthy"
+    elif roe > 0.05 and dte < 2.0 and pm > 0.1:
+        return "At Risk"
+    else:
+        return "Critical"