Update app.py

app.py

@@ -142,18 +142,48 @@ def load_results():
     Load latest artifacts from artifacts/py and artifacts/r.
     Returns values in the order used by the Gradio outputs.
     """
+
+    # --------------------
     # Python artifacts
-    py_metrics = _safe_read_json(BASE_DIR / "artifacts" / "py" / "metrics" / "metrics.json") or {}
-    py_conf = str(BASE_DIR / "artifacts" / "py" / "figures" / "confusion_matrix.png")
-    py_roc = str(BASE_DIR / "artifacts" / "py" / "figures" / "roc_curve.png")
-    py_fi = _safe_read_csv(BASE_DIR / "artifacts" / "py" / "tables" / "feature_importances.csv") or pd.DataFrame()
-    py_pred = _safe_read_csv(BASE_DIR / "artifacts" / "py" / "tables" / "test_predictions.csv", nrows=50) or pd.DataFrame()
+    # --------------------
+    py_metrics = _safe_read_json(BASE_DIR / "artifacts" / "py" / "metrics" / "metrics.json")
+    if py_metrics is None:
+        py_metrics = {}
+
+    py_conf_path = BASE_DIR / "artifacts" / "py" / "figures" / "confusion_matrix.png"
+    py_roc_path = BASE_DIR / "artifacts" / "py" / "figures" / "roc_curve.png"
+
+    py_fi = _safe_read_csv(BASE_DIR / "artifacts" / "py" / "tables" / "feature_importances.csv")
+    if py_fi is None:
+        py_fi = pd.DataFrame()
+
+    py_pred = _safe_read_csv(BASE_DIR / "artifacts" / "py" / "tables" / "test_predictions.csv", nrows=50)
+    if py_pred is None:
+        py_pred = pd.DataFrame()
 
+    # --------------------
     # R artifacts
-    r_metrics = _safe_read_json(BASE_DIR / "artifacts" / "r" / "metrics" / "metrics.json") or {}
-    r_roc = str(BASE_DIR / "artifacts" / "r" / "figures" / "roc_curve.png")
-    r_coef = _safe_read_csv(BASE_DIR / "artifacts" / "r" / "tables" / "coefficients.csv", nrows=50) or pd.DataFrame()
-    r_pred = _safe_read_csv(BASE_DIR / "artifacts" / "r" / "tables" / "test_predictions.csv", nrows=50) or pd.DataFrame()
+    # --------------------
+    r_metrics = _safe_read_json(BASE_DIR / "artifacts" / "r" / "metrics" / "metrics.json")
+    if r_metrics is None:
+        r_metrics = {}
+
+    r_roc_path = BASE_DIR / "artifacts" / "r" / "figures" / "roc_curve.png"
+
+    r_coef = _safe_read_csv(BASE_DIR / "artifacts" / "r" / "tables" / "coefficients.csv", nrows=50)
+    if r_coef is None:
+        r_coef = pd.DataFrame()
+
+    r_pred = _safe_read_csv(BASE_DIR / "artifacts" / "r" / "tables" / "test_predictions.csv", nrows=50)
+    if r_pred is None:
+        r_pred = pd.DataFrame()
+
+    # Gradio Image(type="filepath") works best with:
+    # - a string path if the file exists
+    # - None if it does not exist
+    py_conf = str(py_conf_path) if py_conf_path.exists() else None
+    py_roc = str(py_roc_path) if py_roc_path.exists() else None
+    r_roc = str(r_roc_path) if r_roc_path.exists() else None
 
     return py_metrics, r_metrics, py_conf, py_roc, r_roc, py_fi, r_coef, py_pred, r_pred