from src.gradio_interface import generate_anaglyph
import gradio as gr

iface = gr.Interface(
    fn=generate_anaglyph,
    inputs=[
        gr.Image(type="filepath", label="Upload Person Image"),
        gr.Image(type="filepath", label="Upload Left Stereoscopic Image"),
        gr.Image(type="filepath", label="Upload Right Stereoscopic Image"),
        gr.Dropdown(["", "close", "medium", "far"], label="Depth Level"),
    ],
    outputs=gr.Image(label="Anaglyph Image"),
)

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