import cv2
import gradio as gr
import numpy as np

# 顔の補正関数
def correct_face(image):
    # 画像をグレースケールに変換
    gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
    
    # ヒストグラム均等化で明るさ補正
    corrected_image = cv2.equalizeHist(gray)
    
    # グレースケール画像をカラー画像に変換
    corrected_image = cv2.cvtColor(corrected_image, cv2.COLOR_GRAY2BGR)
    
    return corrected_image

# Gradioインターフェースの作成
iface = gr.Interface(
    fn=correct_face, 
    inputs=gr.Image(type="numpy"), 
    outputs=gr.Image(type="numpy"),
    title="顔補正アプリ",
    description="画像をアップロードして顔の明るさ補正を行います。"
)

# アプリを実行
iface.launch()