Update app.py

app.py

@@ -4,14 +4,16 @@ import numpy as np
 
 # 顔の補正関数
 def correct_face(image):
-    # 画像をグレースケールに変換
-    gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
+    # BGR（OpenCVのデフォルト）からLab色空間に変換
+    lab = cv2.cvtColor(image, cv2.COLOR_BGR2Lab)
     
-    # ヒストグラム均等化で明るさ補正
-    corrected_image = cv2.equalizeHist(gray)
+    # LabのLチャンネル（輝度）にヒストグラム均等化を適用
+    l, a, b = cv2.split(lab)
+    l_eq = cv2.equalizeHist(l)
     
-    # グレースケール画像をカラー画像に変換
-    corrected_image = cv2.cvtColor(corrected_image, cv2.COLOR_GRAY2BGR)
+    # 再度チャンネルを統合し、LabからBGRに戻す
+    lab_eq = cv2.merge((l_eq, a, b))
+    corrected_image = cv2.cvtColor(lab_eq, cv2.COLOR_Lab2BGR)
     
     return corrected_image
 
@@ -21,7 +23,7 @@ iface = gr.Interface(
     inputs=gr.Image(type="numpy"), 
     outputs=gr.Image(type="numpy"),
     title="顔補正アプリ",
-    description="画像をアップロードして顔の明るさ補正を行います。"
+    description="画像をアップロードして顔の色と明るさ補正を行います。"
 )
 
 # アプリを実行