Upload app.py

app.py

@@ -13,7 +13,7 @@ from PIL import Image, ImageEnhance
 import numpy as np
 import gspread
 from oauth2client.service_account import ServiceAccountCredentials
-
+from skimage.exposure import match_histograms
 
 # -------------------------
 # Global parameters
@@ -55,6 +55,24 @@ def match_brightness(source_img, target_img):
     adjusted = enhancer.enhance(factor)
     return adjusted
 
+def match_histograms_pil(img_reference, img_to_adjust):
+    """
+    Calque l'histogramme de `img_reference` sur `img_to_adjust`
+    (les deux images sont des objets PIL.Image).
+    Retourne une image PIL avec histogramme ajusté.
+    """
+    # Convertir les deux images en tableaux numpy
+    ref_array = np.array(img_reference)
+    adj_array = np.array(img_to_adjust)
+
+    # Ajuster l'histogramme
+    matched = match_histograms(adj_array, ref_array, channel_axis=-1)
+
+    # Reconvertir en image PIL
+    matched_img = Image.fromarray(np.uint8(matched))
+
+    return matched_img
+
 # -------------------------
 # Navigation via st.session_state
 # -------------------------
@@ -176,7 +194,8 @@ if st.session_state.page == "evaluation":
     img2 = Image.open(pair[1][1])
     img2 = img2.convert("L")
     
-    img2 = match_brightness(img1, img2)
+    # img2 = match_brightness(img1, img2)
+    img2 = match_histograms_pil(img1, img2)
     
     # Affichage dans deux colonnes avec radio buttons sous chacune
     col1, col2 = st.columns(2)