import gradio as gr
import os
import glob

base_dir = os.path.dirname(__file__)
examples_dir = os.path.join(base_dir, 'examples')


with gr.Blocks() as iface:
    gr.HTML("""
        <div style="text-align: center; font-size: 16px;">
            <img src="file/static/dark_logo.jpg" alt="Fotros Logo" style="width: 200px; margin: 0 auto;">
            <h2>An application that generates 3D meshes from multiple 2D images.</h2>
            Created by <a href="https://abolfazlmohajeri.ir" target="_blank"><strong>Abolfazl Mohajeri</strong></a>
        </div>
        """)

    with gr.Row():
        with gr.Column():
            output_mesh = gr.Model3D(label="3D Mesh", inputs=None, interactive=False)

    gr.HTML("""
        <div style="text-align: center; font-size: 22px;">
            <strong>Examples</strong>
        </div>
        <div style="text-align: center; font-size: 20px;margin-top: 5px;">
            Click the render button below each example to see the output in the box above
        </div>
        """)

    with gr.Row():
        with gr.Column():
            gr.HTML("""
                <div style="text-align: center;">
                    <h3>Example 1</h3>
                </div>
                """)
            gr.Gallery(label="Input Images",
                       columns=3,
                       object_fit="contain",
                       value=glob.glob(os.path.join(examples_dir, 'fox', 'input', '*')))
            gr.Button("Render").click(lambda: os.path.join(examples_dir, 'fox', 'output', "mesh.glb"), inputs=None, outputs=output_mesh)
        with gr.Column():
            gr.HTML("""
                <div style="text-align: center;">
                    <h3>Example 2</h3>
                </div>
                """)
            gr.Gallery(label="Input Images",
                       columns=3,
                       object_fit="contain",
                       value=glob.glob(os.path.join(examples_dir, 'buddha', 'input', '*')))
            gr.Button("Render").click(lambda : os.path.join(examples_dir, 'buddha', 'output', "mesh.glb"), inputs=None, outputs=output_mesh)
        with gr.Column():
            gr.HTML("""
                <div style="text-align: center;">
                    <h3>Example 3</h3>
                </div>
                """)
            gr.Gallery(label="Input Images",
                       columns=3,
                       object_fit="contain",
                       value=glob.glob(os.path.join(examples_dir, 'stone', 'input', '*')))
            gr.Button("Render").click(lambda: os.path.join(examples_dir, 'stone', 'output', "mesh.glb"), inputs=None,outputs=output_mesh)

    gr.HTML("""
        <div style="text-align: center; font-size: 16px;">
            <h1>FOTROS</h1>
            Created by <a href="https://abolfazlmohajeri.ir" target="_blank"><strong>Abolfazl Mohajeri</strong></a>
        </div>
        """)


iface.launch(allowed_paths=["static"])