Update app.py

app.py

@@ -173,6 +173,8 @@ with st.spinner("Téléchargement Sentinel-2 + calcul NDWI…"):
 
     rgb = np.where(mask_rgb[..., None] > 0, rgb, np.nan)
     rgb = normalize_rgb(rgb)
+    rgb_raw = rgb.copy()  # même image true color, sans overlay
+
 
     eps = 1e-6
     ndwi_11 = (b8a - b11) / (b8a + b11 + eps)
@@ -190,13 +192,26 @@ with st.spinner("Téléchargement Sentinel-2 + calcul NDWI…"):
 left, right = st.columns([1.2, 1.0], gap="large")
 
 with left:
-    st.subheader("Overlay stress hydrique sur vraie couleur")
-    fig = plt.figure(figsize=(8, 8))
-    plt.imshow(rgb)
-    plt.imshow(stress, cmap="Reds", alpha=0.40, vmin=0, vmax=1)
-    plt.axis("off")
-    plt.title(f"Stress = NDWI < p{p} (seuil {thr:.3f}) — {swir_choice}")
-    st.pyplot(fig, clear_figure=True)
+    st.subheader("Vraie couleur : raw vs stress")
+
+    c1, c2 = st.columns(2, gap="medium")
+
+    with c1:
+        st.caption("Raw (true color)")
+        fig_raw = plt.figure(figsize=(6, 6))
+        plt.imshow(rgb_raw)
+        plt.axis("off")
+        st.pyplot(fig_raw, clear_figure=True)
+
+    with c2:
+        st.caption("Overlay stress hydrique")
+        fig = plt.figure(figsize=(6, 6))
+        plt.imshow(rgb)
+        plt.imshow(stress, cmap="Reds", alpha=0.40, vmin=0, vmax=1)
+        plt.axis("off")
+        plt.title(f"Stress = NDWI < p{p} (seuil {thr:.3f})")
+        st.pyplot(fig, clear_figure=True)
+
 
 with right:
     st.subheader("Cartes NDWI")