Update src/streamlit_app.py

src/streamlit_app.py

@@ -512,7 +512,7 @@ if page == "Home":
         st.info("Dataset not ready or no image selected yet.")
     else:
         st.sidebar.image(
-            str(selected_image_path),
+            selected_image_path.as_posix(),
             caption="Selected Image",
             use_container_width=True,
         )
@@ -520,12 +520,12 @@ if page == "Home":
         col1, col2 = st.columns([1, 1])
         with col1:
             st.subheader("Chosen Image")
-            st.image(str(selected_image_path), use_container_width=True)
+            st.image(selected_image_path.as_posix(), use_container_width=True)
 
         with col2:
             st.subheader("Prediction Output")
             if st.button("Predict"):
-                result = predict_image(str(selected_image_path))
+                result = predict_image(selected_image_path.as_posix())
                 if result.get("error"):
                     st.error(result["error"])
                 else:
@@ -551,11 +551,11 @@ elif page == "Training Report":
     if CONFUSION_IMAGE_PATH.exists():
         left, center, right = st.columns([1, 3, 1])
         with center:
-            st.image(str(CONFUSION_IMAGE_PATH), width=700)
+            st.image(CONFUSION_IMAGE_PATH.as_posix(), use_container_width=True)
     else:
-        st.info(
-            f"Confusion matrix image not found at `{CONFUSION_IMAGE_PATH}`. "
-            "Save your PNG there to display it here."
+        st.warning(
+            "Confusion matrix image not found. "
+            "Place `confusion_matrix_normalized.png` under `src/reports/`."
         )
 
 # -------------------------------------------------------------------