app.py

import gradio as gr

from src.chat import respond
from src.production.flow import play_fn, pause_fn, reset_fn

custom_theme = gr.themes.Base(
    primary_hue="blue",
    secondary_hue="green",
    neutral_hue="gray",
    font=[gr.themes.GoogleFont("Open Sans"), "sans-serif"],
)

with gr.Blocks(theme=custom_theme) as demo:

    gr.Markdown("# IndustryMind AI ⚡️")
    gr.Markdown("## An industrial production intelligence (WIP)")

    with gr.Tab("Demo"):
        gr.Markdown(
            """
            This demo showcases the capabilities of IndustryMind AI.
            You can interact with the chatbot to get insights and assistance on production-related queries.
            """
        )

        with gr.Row():
            with gr.Column(scale=1):
                chatbot = gr.ChatInterface(respond)
                with gr.Row():
                    with gr.Column(scale=2):
                        with gr.Group():
                            play = gr.Button("▶️ Play")
                            pause = gr.Button("⏸️ Pause")
                            reset = gr.Button("🔄 Reset")

            with gr.Column(scale=2):
                display_df = gr.DataFrame(
                    label="Production Data",
                    headers=[
                        "Part ID", "Timestamp", "Position", "Orientation", "Tool ID",
                        "Compliance", "Event", "Error Code", "Error Description",
                        "Downtime Start", "Downtime End"
                    ]
                )
                play.click(
                        fn=play_fn,
                        inputs=None,
                        outputs=display_df,
                    )
                pause.click(
                    fn=pause_fn,
                    inputs=None,
                    outputs=None
                )
                reset.click(
                    fn=reset_fn,
                    inputs=None,
                    outputs=None
                )

    with gr.Tab("Description"):
        gr.Markdown(
            """
            IndustryMind AI is an AI-powered chatbot designed to assist with industrial production processes. 
            It can help you manage production lines, monitor equipment, and optimize workflows.
            """
        )

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