import gradio as gr

def process_depth(image):
    """シンプルな深度マップ生成"""
    if image is None:
        return None, None
    
    return image, image  # とりあえず同じ画像を返す

# Gradioインターフェース
demo = gr.Interface(
    fn=process_depth,
    inputs=gr.Image(type="pil"),
    outputs=[
        gr.Image(label="元画像"),
        gr.Image(label="深度マップ")
    ],
    title="深度推定API",
    description="画像をアップロードして深度マップを生成"
)

if __name__ == "__main__":
    demo.launch()